/* 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 migrator import ( "strings" "github.com/cgrates/cgrates/engine" "github.com/cgrates/cgrates/utils" "github.com/mediocregopher/radix.v2/redis" ) type redisMigrator struct { dm *engine.DataManager rds *engine.RedisStorage dataKeys []string qryIdx *int } var ( reverseAliasesPrefix = "rls_" ) func newRedisMigrator(dm *engine.DataManager) (rM *redisMigrator) { return &redisMigrator{ dm: dm, rds: dm.DataDB().(*engine.RedisStorage), } } func (v1rs *redisMigrator) close() { v1rs.rds.Close() } func (v1rs *redisMigrator) DataManager() *engine.DataManager { return v1rs.dm } //Account methods //V1 //get func (v1rs *redisMigrator) getv1Account() (v1Acnt *v1Account, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(v1AccountDBPrefix) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } v1Acnt = &v1Account{Id: v1rs.dataKeys[*v1rs.qryIdx]} if err := v1rs.rds.Marshaler().Unmarshal(strVal, v1Acnt); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1Acnt, nil } //set func (v1rs *redisMigrator) setV1Account(x *v1Account) (err error) { key := v1AccountDBPrefix + x.Id bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //rem func (v1rs *redisMigrator) remV1Account(id string) (err error) { key := v1AccountDBPrefix + id return v1rs.rds.Cmd("DEL", key).Err } //V2 //get func (v1rs *redisMigrator) getv2Account() (v2Acnt *v2Account, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.ACCOUNT_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } v2Acnt = &v2Account{ID: v1rs.dataKeys[*v1rs.qryIdx]} if err := v1rs.rds.Marshaler().Unmarshal(strVal, v2Acnt); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v2Acnt, nil } //set func (v1rs *redisMigrator) setV2Account(x *v2Account) (err error) { key := utils.ACCOUNT_PREFIX + x.ID bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //rem func (v1rs *redisMigrator) remV2Account(id string) (err error) { key := utils.ACCOUNT_PREFIX + id return v1rs.rds.Cmd("DEL", key).Err } //ActionPlans methods //get func (v1rs *redisMigrator) getV1ActionPlans() (v1aps *v1ActionPlans, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.ACTION_PLAN_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1aps); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1aps, nil } //set func (v1rs *redisMigrator) setV1ActionPlans(x *v1ActionPlans) (err error) { key := utils.ACTION_PLAN_PREFIX + (*x)[0].Id bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //Actions methods //get func (v1rs *redisMigrator) getV1Actions() (v1acs *v1Actions, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.ACTION_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1acs); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1acs, nil } //set func (v1rs *redisMigrator) setV1Actions(x *v1Actions) (err error) { key := utils.ACTION_PREFIX + (*x)[0].Id bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //ActionTriggers methods //get func (v1rs *redisMigrator) getV1ActionTriggers() (v1acts *v1ActionTriggers, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.ACTION_TRIGGER_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1acts); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1acts, nil } //set func (v1rs *redisMigrator) setV1ActionTriggers(x *v1ActionTriggers) (err error) { key := utils.ACTION_TRIGGER_PREFIX + (*x)[0].Id bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //SharedGroup methods //get func (v1rs *redisMigrator) getV1SharedGroup() (v1sg *v1SharedGroup, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.SHARED_GROUP_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1sg); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1sg, nil } //set func (v1rs *redisMigrator) setV1SharedGroup(x *v1SharedGroup) (err error) { key := utils.SHARED_GROUP_PREFIX + x.Id bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //Stats methods //get func (v1rs *redisMigrator) getV1Stats() (v1st *v1Stat, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.CDR_STATS_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1st); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1st, nil } //set func (v1rs *redisMigrator) setV1Stats(x *v1Stat) (err error) { key := utils.CDR_STATS_PREFIX + x.Id bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //Action methods //get func (v1rs *redisMigrator) getV2ActionTrigger() (v2at *v2ActionTrigger, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.ACTION_TRIGGER_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v2at); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v2at, nil } //set func (v1rs *redisMigrator) setV2ActionTrigger(x *v2ActionTrigger) (err error) { key := utils.ACTION_TRIGGER_PREFIX + x.ID bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //AttributeProfile methods //get func (v1rs *redisMigrator) getV1AttributeProfile() (v1attrPrf *v1AttributeProfile, err error) { var v1attr *v1AttributeProfile if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.AttributeProfilePrefix) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1attr); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1attr, nil } //set func (v1rs *redisMigrator) setV1AttributeProfile(x *v1AttributeProfile) (err error) { key := utils.AttributeProfilePrefix + utils.ConcatenatedKey(x.Tenant, x.ID) bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //ThresholdProfile methods //get func (v1rs *redisMigrator) getV2ThresholdProfile() (v2T *v2Threshold, err error) { var v2Th *v2Threshold if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.ThresholdProfilePrefix) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v2Th); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v2Th, nil } //set func (v1rs *redisMigrator) setV2ThresholdProfile(x *v2Threshold) (err error) { key := utils.ThresholdProfilePrefix + utils.ConcatenatedKey(x.Tenant, x.ID) bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //rem func (v1rs *redisMigrator) remV2ThresholdProfile(tenant, id string) (err error) { key := utils.ThresholdProfilePrefix + utils.ConcatenatedKey(tenant, id) return v1rs.rds.Cmd("DEL", key).Err } //ThresholdProfile methods //get func (v1rs *redisMigrator) getV1Alias() (v1a *v1Alias, err error) { v1a = &v1Alias{Values: make(v1AliasValues, 0)} if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(ALIASES_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { key := v1rs.dataKeys[*v1rs.qryIdx] strVal, err := v1rs.rds.Cmd("GET", key).Bytes() if err != nil { return nil, err } v1a.SetId(strings.TrimPrefix(key, ALIASES_PREFIX)) if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1a.Values); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v1a, nil } //set func (v1rs *redisMigrator) setV1Alias(al *v1Alias) (err error) { var result []byte result, err = v1rs.rds.Marshaler().Marshal(al.Values) if err != nil { return } key := ALIASES_PREFIX + al.GetId() if err = v1rs.rds.Cmd("SET", key, result).Err; err != nil { return } return } //rem func (v1rs *redisMigrator) remV1Alias(key string) (err error) { // get alias for values list var values []byte if values, err = v1rs.rds.Cmd("GET", ALIASES_PREFIX+key).Bytes(); err != nil { if err == redis.ErrRespNil { // did not find the destination err = utils.ErrNotFound } return } al := &v1Alias{Values: make(v1AliasValues, 0)} al.SetId(key) if err = v1rs.rds.Marshaler().Unmarshal(values, &al.Values); err != nil { return err } err = v1rs.rds.Cmd("DEL", ALIASES_PREFIX+key).Err if err != nil { return err } for _, value := range al.Values { tmpKey := utils.ConcatenatedKey(al.GetId(), value.DestinationId) for target, pairs := range value.Pairs { for _, alias := range pairs { revID := alias + target + al.Context err = v1rs.rds.Cmd("SREM", reverseAliasesPrefix+revID, tmpKey).Err if err != nil { return err } } } } return return v1rs.rds.Cmd("DEL", key).Err } // User methods //get func (v1rs *redisMigrator) getV1User() (v1u *v1UserProfile, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.USERS_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } v1u = new(v1UserProfile) if err := v1rs.rds.Marshaler().Unmarshal(strVal, v1u); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 return v1u, nil } v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } //set func (v1rs *redisMigrator) setV1User(us *v1UserProfile) (err error) { bit, err := v1rs.rds.Marshaler().Marshal(us) if err != nil { return err } return v1rs.rds.Cmd("SET", utils.USERS_PREFIX+us.GetId(), bit).Err } //rem func (v1rs *redisMigrator) remV1User(key string) (err error) { return v1rs.rds.Cmd("DEL", utils.USERS_PREFIX+key).Err } // DerivedChargers methods //get func (v1rs *redisMigrator) getV1DerivedChargers() (v1d *v1DerivedChargersWithKey, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.DERIVEDCHARGERS_PREFIX) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } v1d = new(v1DerivedChargersWithKey) v1d.Key = strings.TrimPrefix(v1rs.dataKeys[*v1rs.qryIdx], utils.DERIVEDCHARGERS_PREFIX) v1d.Value = new(v1DerivedChargers) if err := v1rs.rds.Marshaler().Unmarshal(strVal, v1d.Value); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 return v1d, nil } v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } //set func (v1rs *redisMigrator) setV1DerivedChargers(dc *v1DerivedChargersWithKey) (err error) { if dc == nil || len(dc.Value.Chargers) == 0 { return v1rs.remV1DerivedChargers(dc.Key) } bit, err := v1rs.rds.Marshaler().Marshal(dc.Value) if err != nil { return err } return v1rs.rds.Cmd("SET", utils.DERIVEDCHARGERS_PREFIX+dc.Key, bit).Err } //rem func (v1rs *redisMigrator) remV1DerivedChargers(key string) (err error) { return v1rs.rds.Cmd("DEL", utils.DERIVEDCHARGERS_PREFIX+key).Err } //AttributeProfile methods //get func (v1rs *redisMigrator) getV2AttributeProfile() (v2attrPrf *v2AttributeProfile, err error) { var v2attr *v2AttributeProfile if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.AttributeProfilePrefix) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v2attr); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v2attr, nil } //set func (v1rs *redisMigrator) setV2AttributeProfile(x *v2AttributeProfile) (err error) { key := utils.AttributeProfilePrefix + utils.ConcatenatedKey(x.Tenant, x.ID) bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //rem func (v1rs *redisMigrator) remV2AttributeProfile(tenant, id string) (err error) { key := utils.AttributeProfilePrefix + utils.ConcatenatedKey(tenant, id) return v1rs.rds.Cmd("DEL", key).Err } //AttributeProfile methods //get func (v1rs *redisMigrator) getV3AttributeProfile() (v3attrPrf *v3AttributeProfile, err error) { var v3attr *v3AttributeProfile if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.AttributeProfilePrefix) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v3attr); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v3attr, nil } //set func (v1rs *redisMigrator) setV3AttributeProfile(x *v3AttributeProfile) (err error) { key := utils.AttributeProfilePrefix + utils.ConcatenatedKey(x.Tenant, x.ID) bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //rem func (v1rs *redisMigrator) remV3AttributeProfile(tenant, id string) (err error) { key := utils.AttributeProfilePrefix + utils.ConcatenatedKey(tenant, id) return v1rs.rds.Cmd("DEL", key).Err } //AttributeProfile methods //get func (v1rs *redisMigrator) getV4AttributeProfile() (v3attrPrf *v4AttributeProfile, err error) { var v4attr *v4AttributeProfile if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.AttributeProfilePrefix) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v4attr); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return v4attr, nil } //set func (v1rs *redisMigrator) setV4AttributeProfile(x *v4AttributeProfile) (err error) { key := utils.AttributeProfilePrefix + utils.ConcatenatedKey(x.Tenant, x.ID) bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } if err = v1rs.rds.Cmd("SET", key, bit).Err; err != nil { return err } return } //rem func (v1rs *redisMigrator) remV4AttributeProfile(tenant, id string) (err error) { key := utils.AttributeProfilePrefix + utils.ConcatenatedKey(tenant, id) return v1rs.rds.Cmd("DEL", key).Err } // Filter Methods //get func (v1rs *redisMigrator) getV1Filter() (v1Fltr *v1Filter, err error) { if v1rs.qryIdx == nil { v1rs.dataKeys, err = v1rs.rds.GetKeysForPrefix(utils.FilterPrefix) if err != nil { return } else if len(v1rs.dataKeys) == 0 { return nil, utils.ErrNoMoreData } v1rs.qryIdx = utils.IntPointer(0) } if *v1rs.qryIdx <= len(v1rs.dataKeys)-1 { strVal, err := v1rs.rds.Cmd("GET", v1rs.dataKeys[*v1rs.qryIdx]).Bytes() if err != nil { return nil, err } if err := v1rs.rds.Marshaler().Unmarshal(strVal, &v1Fltr); err != nil { return nil, err } *v1rs.qryIdx = *v1rs.qryIdx + 1 } else { v1rs.qryIdx = nil return nil, utils.ErrNoMoreData } return } //set func (v1rs *redisMigrator) setV1Filter(x *v1Filter) (err error) { key := utils.FilterPrefix + utils.ConcatenatedKey(x.Tenant, x.ID) bit, err := v1rs.rds.Marshaler().Marshal(x) if err != nil { return err } return v1rs.rds.Cmd("SET", key, bit).Err } //rem func (v1rs *redisMigrator) remV1Filter(tenant, id string) (err error) { key := utils.FilterPrefix + utils.ConcatenatedKey(tenant, id) return v1rs.rds.Cmd("DEL", key).Err }