From 0f0f05d525bdfad81732dfa9140e68a0ea3ace71 Mon Sep 17 00:00:00 2001 From: porosnicuadrian Date: Tue, 25 May 2021 16:58:37 +0300 Subject: [PATCH] Added back APIS for routes --- apis/routes.go | 142 +++++++++++++++++++++++++ data/tariffplans/tutroutes/Filters.csv | 0 2 files changed, 142 insertions(+) create mode 100644 apis/routes.go create mode 100644 data/tariffplans/tutroutes/Filters.csv diff --git a/apis/routes.go b/apis/routes.go new file mode 100644 index 000000000..e02d116d8 --- /dev/null +++ b/apis/routes.go @@ -0,0 +1,142 @@ +/* +Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments +Copyright (C) ITsysCOM GmbH +This program is free software: you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation, either version 3 of the License, or +(at your option) any later version. +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. +You should have received a copy of the GNU General Public License +along with this program. If not, see +*/ + +package apis + +import ( + "time" + + "github.com/cgrates/birpc/context" + "github.com/cgrates/cgrates/engine" + "github.com/cgrates/cgrates/utils" +) + +// GetRouteProfile returns a Route configuration +func (adms *AdminSv1) GetRouteProfile(ctx *context.Context, arg *utils.TenantIDWithAPIOpts, reply *engine.RouteProfile) error { + if missing := utils.MissingStructFields(arg, []string{utils.ID}); len(missing) != 0 { //Params missing + return utils.NewErrMandatoryIeMissing(missing...) + } + tnt := arg.Tenant + if tnt == utils.EmptyString { + tnt = adms.cfg.GeneralCfg().DefaultTenant + } + if rp, err := adms.dm.GetRouteProfile(ctx, tnt, arg.ID, true, true, utils.NonTransactional); err != nil { + return utils.APIErrorHandler(err) + } else { + *reply = *rp + } + return nil +} + +// GetRouteProfileIDs returns list of routeProfile IDs registered for a tenant +func (adms *AdminSv1) GetRouteProfileIDs(ctx *context.Context, args *utils.PaginatorWithTenant, sppPrfIDs *[]string) error { + tnt := args.Tenant + if tnt == utils.EmptyString { + tnt = adms.cfg.GeneralCfg().DefaultTenant + } + prfx := utils.RouteProfilePrefix + tnt + utils.ConcatenatedKeySep + keys, err := adms.dm.DataDB().GetKeysForPrefix(ctx, prfx) + if err != nil { + return err + } + if len(keys) == 0 { + return utils.ErrNotFound + } + retIDs := make([]string, len(keys)) + for i, key := range keys { + retIDs[i] = key[len(prfx):] + } + *sppPrfIDs = args.PaginateStringSlice(retIDs) + return nil +} + +type RouteWithAPIOpts struct { + *engine.RouteProfile + APIOpts map[string]interface{} +} + +//SetRouteProfile add a new Route configuration +func (adms *AdminSv1) SetRouteProfile(ctx *context.Context, args *RouteWithAPIOpts, reply *string) error { + if missing := utils.MissingStructFields(args.RouteProfile, []string{utils.ID}); len(missing) != 0 { + return utils.NewErrMandatoryIeMissing(missing...) + } + if args.Tenant == utils.EmptyString { + args.Tenant = adms.cfg.GeneralCfg().DefaultTenant + } + if err := adms.dm.SetRouteProfile(ctx, args.RouteProfile, true); err != nil { + return utils.APIErrorHandler(err) + } + //generate a loadID for CacheRouteProfiles and store it in database + if err := adms.dm.SetLoadIDs(ctx, map[string]int64{utils.CacheRouteProfiles: time.Now().UnixNano()}); err != nil { + return utils.APIErrorHandler(err) + } + //handle caching for SupplierProfile + if err := adms.CallCache(ctx, utils.IfaceAsString(args.APIOpts[utils.CacheOpt]), args.Tenant, utils.CacheRouteProfiles, + args.TenantID(), &args.FilterIDs, nil, args.APIOpts); err != nil { + return utils.APIErrorHandler(err) + } + *reply = utils.OK + return nil +} + +//RemoveRouteProfile remove a specific Route configuration +func (adms *AdminSv1) RemoveRouteProfile(ctx *context.Context, args *utils.TenantIDWithAPIOpts, reply *string) error { + if missing := utils.MissingStructFields(args, []string{utils.ID}); len(missing) != 0 { //Params missing + return utils.NewErrMandatoryIeMissing(missing...) + } + tnt := args.Tenant + if tnt == utils.EmptyString { + tnt = adms.cfg.GeneralCfg().DefaultTenant + } + if err := adms.dm.RemoveRouteProfile(ctx, tnt, args.ID, utils.NonTransactional, true); err != nil { + return utils.APIErrorHandler(err) + } + //generate a loadID for CacheRouteProfiles and store it in database + if err := adms.dm.SetLoadIDs(ctx, map[string]int64{utils.CacheRouteProfiles: time.Now().UnixNano()}); err != nil { + return utils.APIErrorHandler(err) + } + //handle caching for SupplierProfile + if err := adms.CallCache(ctx, utils.IfaceAsString(args.APIOpts[utils.CacheOpt]), tnt, utils.CacheRouteProfiles, + utils.ConcatenatedKey(tnt, args.ID), nil, nil, args.APIOpts); err != nil { + return utils.APIErrorHandler(err) + } + *reply = utils.OK + return nil +} + +func NewRouteSv1(rS *engine.RouteService) *RouteSv1 { + return &RouteSv1{rS: rS} +} + +// RouteSv1 exports RPC from RouteS +type RouteSv1 struct { + ping + rS *engine.RouteService +} + +// GetRoutes returns sorted list of routes for Event +func (rS *RouteSv1) GetRoutes(ctx *context.Context, args *engine.ArgsGetRoutes, reply *engine.SortedRoutesList) error { + return rS.rS.V1GetRoutes(ctx, args, reply) +} + +// GetRouteProfilesForEvent returns a list of route profiles that match for Event +func (rS *RouteSv1) GetRouteProfilesForEvent(ctx *context.Context, args *utils.CGREvent, reply *[]*engine.RouteProfile) error { + return rS.rS.V1GetRouteProfilesForEvent(ctx, args, reply) +} + +// GetRoutesList returns sorted list of routes for Event as a string slice +func (rS *RouteSv1) GetRoutesList(ctx *context.Context, args *engine.ArgsGetRoutes, reply *[]string) error { + return rS.rS.V1GetRoutesList(ctx, args, reply) +} diff --git a/data/tariffplans/tutroutes/Filters.csv b/data/tariffplans/tutroutes/Filters.csv new file mode 100644 index 000000000..e69de29bb