fix: adjustment directory structure
This commit is contained in:
zhangsz
115
proxy/Nredis/acctount.go
Normal file
115
proxy/Nredis/acctount.go
Normal file
@@ -0,0 +1,115 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
. "proxy/MsgDef"
|
||||
l4g "proxy/logger"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
// "Acct:"+acctData.ServiceNbr
|
||||
func RdbDelAcctRecord(serviceNbr string) {
|
||||
var cursor uint64= 0
|
||||
for {
|
||||
keys, cursorNew, err := rdb.SScan(ctx, "Acct:"+serviceNbr, cursor, "", 0).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("SScan Acct:%s, err: %v", serviceNbr, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
acctKey := "Acct:"+serviceNbr+":"+key
|
||||
err := rdb.Del(ctx, acctKey).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", acctKey, err)
|
||||
} else {
|
||||
//subFields := strings.Split(key, ":")
|
||||
//if len(subFields) == 3 && subFields[2] != "" {
|
||||
_ = rdb.SRem(ctx, "Acct:"+serviceNbr, key)
|
||||
//return srem.Err()
|
||||
//}
|
||||
}
|
||||
}
|
||||
if cursorNew == 0 {
|
||||
break
|
||||
} else {
|
||||
cursor = cursorNew
|
||||
}
|
||||
}
|
||||
_ = rdb.Del(ctx, "Acct:"+serviceNbr).Err()
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func RdbDelAcctOfr(serviceNbr string, ofrId int) error {
|
||||
acctKey := "Acct:"+serviceNbr+":"+strconv.Itoa(ofrId)
|
||||
err := rdb.Del(ctx, acctKey).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", acctKey, err)
|
||||
} else {
|
||||
_ = rdb.SRem(ctx, "Acct:"+serviceNbr, strconv.Itoa(ofrId))
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbSetAcctRecord(acctData *AcctData) error {
|
||||
//&prd_inst_id, &service_nbr, &prd_inst_stas_id, &acct_id, &cust_id
|
||||
key := "Acct:"+acctData.ServiceNbr+":"+strconv.Itoa(acctData.OfrId)
|
||||
err := rdb.HMSet(ctx, key, "acctType", acctData.AcctType, "prdInstId", acctData.PrdInstId, "serviceNbr", acctData.ServiceNbr,
|
||||
"prdInstStasId", acctData.PrdInstStasId, "acctId", acctData.AcctId, "custId", acctData.CustId, "ofrId", acctData.OfrId, "ofrInstId", acctData.OfrInstId,
|
||||
"effTime", acctData.EffTime, "expTime", acctData.ExpTime).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet Acct:%s, err: %v", acctData.ServiceNbr, err)
|
||||
} else {
|
||||
sadd := rdb.SAdd(ctx, "Acct:"+acctData.ServiceNbr, strconv.Itoa(acctData.OfrId))
|
||||
if sadd.Err() != nil {
|
||||
l4g.RedisLog.Errorf("SAdd Acct:%s %d, err: %v", acctData.ServiceNbr, acctData.OfrId, sadd.Err())
|
||||
return sadd.Err()
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("SAdd Acct:%s %d", acctData.ServiceNbr, acctData.OfrId)
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbGetAcctRecordByServiceNbr(serviceNbr string) (rsp string, err error) {
|
||||
var cursor uint64= 0
|
||||
for {
|
||||
keys, cursorNew, err := rdb.SScan(ctx, "Acct:"+serviceNbr, cursor, "", 0).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("SScan Acct:%s, err: %v", serviceNbr, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
acctKey := "Acct:"+serviceNbr+":"+key
|
||||
res := rdb.HGetAll(ctx, acctKey)
|
||||
if res.Err() != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll %s, err: %v", acctKey, err)
|
||||
} else {
|
||||
var acctData AcctData
|
||||
if err = res.Scan(&acctData); err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll Scan %s, err: %v", acctKey, err)
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Get %s, [%v]", acctKey, acctData)
|
||||
rsp += fmt.Sprintf("%s:***************************************\r\nPrdInstId: %d\r\nPrdInstStasId: %d\r\nCustId: %d\r\nAcctId: %d\r\n",
|
||||
acctKey, acctData.PrdInstId, acctData.PrdInstStasId, acctData.CustId, acctData.AcctId)
|
||||
rsp += fmt.Sprintf("OfrId: %d\r\nOfrInstId: %d\r\nEffTime: %s\r\nExpTime: %s\r\n",
|
||||
acctData.OfrId, acctData.OfrInstId, acctData.EffTime, acctData.ExpTime)
|
||||
return rsp, err
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursorNew == 0 {
|
||||
break
|
||||
} else {
|
||||
cursor = cursorNew
|
||||
}
|
||||
}
|
||||
|
||||
return "", err
|
||||
}
|
||||
63
proxy/Nredis/alert_sms.go
Normal file
63
proxy/Nredis/alert_sms.go
Normal file
@@ -0,0 +1,63 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
l4g "proxy/logger"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func RdbSetAlertSmsRecord(alertId int, serviceNbr, smsContent string) error {
|
||||
key := "AlertSms:"+serviceNbr+":"+strconv.Itoa(alertId)
|
||||
err := rdb.Set(ctx, key, smsContent, 0).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Set %s, err: %v", key, err)
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Set %s, err: %s", key, smsContent)
|
||||
sadd := rdb.SAdd(ctx, "Msg2OcsSet", key)
|
||||
if sadd.Err() != nil {
|
||||
l4g.RedisLog.Errorf("SAdd Msg2OcsSet %s, err: %v", key, sadd.Err())
|
||||
return sadd.Err()
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("SAdd Msg2OcsSet %s", key)
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbGetAlertSmsContent(key string) string {
|
||||
var sms string
|
||||
var err error
|
||||
sms, err = rdb.Get(ctx, key).Result()
|
||||
if err != nil {
|
||||
return ""
|
||||
} else {
|
||||
return sms
|
||||
}
|
||||
}
|
||||
|
||||
func RdbScanAlertSmsRecord() []string {
|
||||
var cursor uint64= 0
|
||||
var keysFound []string
|
||||
for {
|
||||
keys, cursor, err := rdb.SScan(ctx, "Msg2OcsSet", cursor, "", 50).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("SScan Msg2OcsSet, err: %v", err)
|
||||
break
|
||||
}
|
||||
|
||||
keysFound = append(keysFound, keys...)
|
||||
|
||||
if cursor == 0 {
|
||||
break
|
||||
} else {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return keysFound
|
||||
}
|
||||
|
||||
func RdsDelMsg2OcsKey(key string) {
|
||||
_ = rdb.SRem(ctx, "Msg2OcsSet", key)
|
||||
_ = rdb.Del(ctx, key).Err()
|
||||
}
|
||||
169
proxy/Nredis/bundleclr.go
Normal file
169
proxy/Nredis/bundleclr.go
Normal file
@@ -0,0 +1,169 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
"proxy/MsgDef"
|
||||
. "proxy/MsgDef"
|
||||
l4g "proxy/logger"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
func CheckIfRrExpired(rr *MsgDef.RrData) bool {
|
||||
if rr == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
if rr.MeasureDomain == "01" || rr.MeasureDomain == "03" {
|
||||
if rr.UsedValue >= rr.FreeValue {
|
||||
return true
|
||||
}
|
||||
nowStr := time.Now().Format("20060102150405")
|
||||
if strings.Compare(nowStr, rr.EndTime) > 0 {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
if rr.MeasureDomain == "02" {
|
||||
nowStr := time.Now().Format("20060102150405")
|
||||
if strings.Compare(nowStr, rr.EndTime) > 0 {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func RdbGetRrByKey(key string) (*RrData, error) {
|
||||
res := rdb.HGetAll(ctx, key)
|
||||
err := res.Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll %s, err: %v", key, err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var rr RrData
|
||||
if err = res.Scan(&rr); err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll Scan %s, err: %v", key, err)
|
||||
return nil, err
|
||||
} else {
|
||||
//l4g.RedisLog.Debugf("Get %s, [%v]", key, rr)
|
||||
}
|
||||
|
||||
return &rr, err
|
||||
}
|
||||
|
||||
func RdbClrExpireRrRecord() error {
|
||||
l4g.RedisLog.Warnf("cron clear expire rr!")
|
||||
var cursor uint64= 0
|
||||
var err error
|
||||
for {
|
||||
var keys []string
|
||||
keys, cursor, err = rdb.Scan(ctx, cursor, "Rr:*", 30).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Scan Rr:*, err: %v", err)
|
||||
break
|
||||
}
|
||||
|
||||
for _, key := range keys {
|
||||
subFields := strings.Split(key, ":")
|
||||
if len(subFields) == 4 {
|
||||
rr, err1 := RdbGetRrByKey(key)
|
||||
if rr != nil && err1 == nil && CheckIfRrExpired(rr) {
|
||||
err1 = rdb.Del(ctx, key).Err()
|
||||
if err1 != nil {
|
||||
l4g.RedisLog.Errorf("Del Rr[%s], err: %v", key, err1)
|
||||
} else {
|
||||
l4g.RedisLog.Warnf("Del Rr[%s] succ", key)
|
||||
|
||||
l4g.RedisLog.Warnf("SRem RrSet:%s:%s %s", subFields[1], subFields[2], subFields[3])
|
||||
_ = rdb.SRem(ctx, "RrSet:"+subFields[1]+":"+subFields[2], subFields[3])
|
||||
l4g.RedisLog.Warnf("Del RrId:%s", subFields[3])
|
||||
_ = rdb.Del(ctx, "RrId:"+subFields[3]).Err()
|
||||
//return srem.Err()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursor == 0 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbGetAcctPlanByKey(key string) (*AcctData, error) {
|
||||
res := rdb.HGetAll(ctx, key)
|
||||
err := res.Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll %s, err: %v", key, err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var acct AcctData
|
||||
if err = res.Scan(&acct); err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll Scan %s, err: %v", key, err)
|
||||
return nil, err
|
||||
} else {
|
||||
//l4g.RedisLog.Debugf("Get %s, [%v]", key, acct)
|
||||
}
|
||||
|
||||
return &acct, err
|
||||
}
|
||||
|
||||
func CheckIfAcctExpired(acct *MsgDef.AcctData) bool {
|
||||
if acct == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
nowStr := time.Now().Format("2006-01-02 15:04:05")
|
||||
if strings.Compare(nowStr, acct.ExpTime[:len(nowStr)]) > 0 {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func RdbClrExpirePlanRecord() error {
|
||||
l4g.RedisLog.Warnf("cron clear expire plan!")
|
||||
var cursor uint64= 0
|
||||
var err error
|
||||
for {
|
||||
var keys []string
|
||||
keys, cursor, err = rdb.Scan(ctx, cursor, "Acct:*", 30).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Scan Acct:*, err: %v", err)
|
||||
break
|
||||
}
|
||||
|
||||
for _, key := range keys {
|
||||
subFields := strings.Split(key, ":")
|
||||
if len(subFields) == 3 {
|
||||
acct, err1 := RdbGetAcctPlanByKey(key)
|
||||
if acct != nil && err1 == nil && CheckIfAcctExpired(acct) {
|
||||
err1 = rdb.Del(ctx, key).Err()
|
||||
if err1 != nil {
|
||||
l4g.RedisLog.Errorf("Del Acct Plan[%s], err: %v", key, err1)
|
||||
} else {
|
||||
l4g.RedisLog.Warnf("Del Acct Plan[%s] succ", key)
|
||||
|
||||
l4g.RedisLog.Warnf("SRem Acct:%s %s", subFields[1], subFields[2])
|
||||
_ = rdb.SRem(ctx, "Acct:"+subFields[1], subFields[2])
|
||||
//return srem.Err()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursor == 0 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func ClrExpRes() {
|
||||
_ = RdbClrExpireRrRecord()
|
||||
_ = RdbClrExpirePlanRecord()
|
||||
}
|
||||
40
proxy/Nredis/canal_info.go
Normal file
40
proxy/Nredis/canal_info.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
l4g "proxy/logger"
|
||||
)
|
||||
|
||||
type BinLogPos struct {
|
||||
File string `redis:"file"`
|
||||
Position uint32 `redis:"position"`
|
||||
}
|
||||
|
||||
func RdbSetBinLogPos(file string, pos uint32) error {
|
||||
err := rdb.HMSet(ctx, "binlogpos", "file", file, "position", pos).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet binlogpos %s %d, err: %v", file, pos, err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbGetBinLogPos() *BinLogPos {
|
||||
// Get the map. The same approach works for HmGet().
|
||||
res := rdb.HGetAll(ctx, "binlogpos")
|
||||
err := res.Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll binlogpos, err: %v", err)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Scan the results into the struct.
|
||||
var pos BinLogPos
|
||||
if err = res.Scan(&pos); err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll Scan binlogpos, err: %v", err)
|
||||
return nil
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Get binlogpos, [%v]", pos)
|
||||
return &pos
|
||||
}
|
||||
}
|
||||
|
||||
47
proxy/Nredis/create_acct.go
Normal file
47
proxy/Nredis/create_acct.go
Normal file
@@ -0,0 +1,47 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
"proxy/MsgDef"
|
||||
l4g "proxy/logger"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func RdbSetCreateAcctRecord(r *MsgDef.ChgSyncMobile) error {
|
||||
key := "CreateAcct:"+r.ServiceNbr+":"+strconv.Itoa(r.PreId)
|
||||
err := rdb.HMSet(ctx, key, "prdInstId", r.PrdInstId, "serviceNbr", r.ServiceNbr,
|
||||
"operType", r.OperType, "state", r.State, "ofrId", r.OfrId,
|
||||
"custId", r.CustId, "acctId", r.AcctId, "mobileType", r.MobileType, "balance", r.Balance, "birthDate", r.BirthDate, "balanceExpDate", r.BalanceExpDate).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet %s, err: %v", key, err)
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("HMSet %s, [%#v]", key, r)
|
||||
sadd := rdb.SAdd(ctx, "Msg2OcsSet", key)
|
||||
if sadd.Err() != nil {
|
||||
l4g.RedisLog.Errorf("SAdd Msg2OcsSet %s, err: %v", key, sadd.Err())
|
||||
return sadd.Err()
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("SAdd Msg2OcsSet %s", key)
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbGetCreateAcct(key string) *MsgDef.ChgSyncMobile {
|
||||
res := rdb.HGetAll(ctx, key)
|
||||
err := res.Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll %s, err: %v", key, err)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Scan the results into the struct.
|
||||
var acctInfo MsgDef.ChgSyncMobile
|
||||
if err = res.Scan(&acctInfo); err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll Scan %s, err: %v", key, err)
|
||||
return nil
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Get %s, [%v]", key, acctInfo)
|
||||
return &acctInfo
|
||||
}
|
||||
}
|
||||
30
proxy/Nredis/holiday.go
Normal file
30
proxy/Nredis/holiday.go
Normal file
@@ -0,0 +1,30 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
l4g "proxy/logger"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func SetHdId2TariffId(id int, tariffId int) bool {
|
||||
err := rdb.Set(ctx, "HdId:"+strconv.Itoa(id), strconv.Itoa(tariffId), 0).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Set HdId:%d %s, err: %v", id, strconv.Itoa(tariffId), err)
|
||||
return false
|
||||
} else {
|
||||
l4g.RedisLog.Infof("Set HdId:%d %s", id, tariffId)
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
func GetTariffIdByHdId(id int) string {
|
||||
var tariffIdStr string
|
||||
var err error
|
||||
tariffIdStr, err = rdb.Get(ctx, "HdId:"+strconv.Itoa(id)).Result()
|
||||
if err != nil {
|
||||
return ""
|
||||
} else {
|
||||
return tariffIdStr
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
33
proxy/Nredis/plan_expiry.go
Normal file
33
proxy/Nredis/plan_expiry.go
Normal file
@@ -0,0 +1,33 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
l4g "proxy/logger"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
func CheckIfBundleLimitSent(planId uint32, rate int) bool {
|
||||
n, err := rdb.Exists(ctx, "BundleLimit:"+strconv.Itoa(int(planId))+":"+strconv.Itoa(rate)).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Check Exists PlanExpiry:%d, err: %v", planId, err)
|
||||
return false
|
||||
} else {
|
||||
if n >= 1 {
|
||||
return true
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func SetPlanExpire(planId uint32, rate int) error {
|
||||
key := "BundleLimit:"+strconv.Itoa(int(planId))+":"+strconv.Itoa(rate)
|
||||
err := rdb.Set(ctx, key, "1", time.Hour * 24 * 15).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Set BundleLimit:%d:%d, err: %v", planId, rate, err)
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Set BundleLimit:%d:%d, succ", planId, rate)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
601
proxy/Nredis/rds_client.go
Normal file
601
proxy/Nredis/rds_client.go
Normal file
@@ -0,0 +1,601 @@
|
||||
package Nredis
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
rds "github.com/go-redis/redis/v8"
|
||||
"os"
|
||||
l4g "proxy/logger" //"github.com/sirupsen/logrus"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
. "proxy/MsgDef"
|
||||
)
|
||||
|
||||
var ctx = context.Background()
|
||||
|
||||
var rdb *rds.Client
|
||||
var locRdb *rds.Client
|
||||
func NewRedisClient(netType, addr, passwd string, sentinels []string) {
|
||||
if sentinels != nil {
|
||||
rdb = rds.NewFailoverClient(&rds.FailoverOptions{
|
||||
MasterName: "mymaster",
|
||||
SentinelAddrs: sentinels,
|
||||
Password: passwd,
|
||||
DB: 0,
|
||||
PoolSize: 5,
|
||||
})
|
||||
locRdb = rds.NewClient(&rds.Options{
|
||||
Network: netType,// The network type, either tcp or unix. Default is tcp.
|
||||
Addr: addr,// can be: /var/run/redis/redis-server.sock???
|
||||
Password: passwd, // no password set
|
||||
DB: 0, // use default DB
|
||||
PoolSize: 5,
|
||||
})
|
||||
} else {
|
||||
rdb = rds.NewClient(&rds.Options{
|
||||
Network: netType,// The network type, either tcp or unix. Default is tcp.
|
||||
Addr: addr,// can be: /var/run/redis/redis-server.sock???
|
||||
Password: passwd, // no password set
|
||||
DB: 0, // use default DB
|
||||
PoolSize: 5,
|
||||
})
|
||||
locRdb = rdb
|
||||
}
|
||||
|
||||
pong, err := rdb.Ping(ctx).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("%v, err: %v", pong, err)// Output: PONG <nil>
|
||||
os.Exit(1)
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("%v", pong)// Output: PONG <nil>
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func CheckIfNeedToSyncDb() (bool, error) {
|
||||
/*if CheckIfRdbMaster() == false {
|
||||
return false, nil
|
||||
}*/
|
||||
|
||||
size, err := rdb.DBSize(ctx).Result()
|
||||
if err == nil {
|
||||
if size < 10 {
|
||||
return true, nil
|
||||
} else {
|
||||
return false, nil
|
||||
}
|
||||
} else {
|
||||
return false, err
|
||||
}
|
||||
}
|
||||
|
||||
func ClrTariffAndBundle() {
|
||||
rdb.FlushDB(ctx)
|
||||
/*n, err := rdb.Del(ctx, "Acct:*").Result()
|
||||
if err != nil {
|
||||
l4g.Errorf("Del Acct, err: %v", err)
|
||||
} else {
|
||||
l4g.Debugf("Del Acct, num[%d]", n)
|
||||
}
|
||||
|
||||
n, err = rdb.Del(ctx, "Rr:*").Result()
|
||||
if err != nil {
|
||||
l4g.Errorf("Del Rr, err: %v", err)
|
||||
} else {
|
||||
l4g.Debugf("Del Rr, num[%d]", n)
|
||||
}
|
||||
|
||||
n, err = rdb.Del(ctx, "Prefix:*").Result()
|
||||
if err != nil {
|
||||
l4g.Errorf("Del Prefix, err: %v", err)
|
||||
} else {
|
||||
l4g.Debugf("Del Prefix, num[%d]", n)
|
||||
}
|
||||
|
||||
n, err = rdb.Del(ctx, "Tariff:*").Result()
|
||||
if err != nil {
|
||||
l4g.Errorf("Del Tariff, err: %v", err)
|
||||
} else {
|
||||
l4g.Debugf("Del Tariff, num[%d]", n)
|
||||
}*/
|
||||
}
|
||||
|
||||
func CheckIfRdbMaster() bool {
|
||||
info := locRdb.Info(ctx, "Replication")
|
||||
if strings.Contains(info.Val(), "role:master") || strings.Contains(info.Val(), "role:active-replica") {
|
||||
return true
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func SetRrId2RrKey(id int64, rrKey string) bool {
|
||||
err := rdb.Set(ctx, "RrId:"+strconv.FormatInt(id, 10), rrKey, 0).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Set RrId:%d %s, err: %v", id, rrKey, err)
|
||||
return false
|
||||
} else {
|
||||
l4g.RedisLog.Infof("Set RrId:%d %s", id, rrKey)
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
func GetRrKeyByRrId(id int64) string {
|
||||
var rrKey string
|
||||
var err error
|
||||
rrKey, err = rdb.Get(ctx, "RrId:"+strconv.FormatInt(id, 10)).Result()
|
||||
if err != nil {
|
||||
return ""
|
||||
} else {
|
||||
return rrKey
|
||||
}
|
||||
}
|
||||
|
||||
func RdbSetRrRecord(rr *RrData) error {
|
||||
key := "Rr:"+rr.ServiceNbr+":"+rr.MeasureDomain+":"+strconv.FormatInt(rr.RrId, 10)
|
||||
err := rdb.HMSet(ctx, key, "prdInstId", rr.PrdInstId, "serviceNbr", rr.ServiceNbr,
|
||||
"pricingSectionId", rr.PricingSectionId, "ofrInstId", rr.OfrInstId, "ofrId", rr.OfrId,
|
||||
"ofrName", rr.OfrName, "beginTime", rr.BeginTime, "endTime", rr.EndTime, "freeValue", rr.FreeValue, "rateUnit", rr.RateUnit, "unitFee", rr.UnitFee,
|
||||
"usedValue", rr.UsedValue, "measureDomain", rr.MeasureDomain, "tariffId", rr.TariffId, "strategyId", rr.StrategyId,
|
||||
"startRefTime", rr.StartRefTime, "endRefTime", rr.EndRefTime, "calcPriority", rr.CalcPriority, "eventPriority", rr.EventPriority).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet Rr:%s, err: %v", rr.ServiceNbr, err)
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("HMSet Rr:%s, [%#v]", rr.ServiceNbr, rr)
|
||||
sadd := rdb.SAdd(ctx, "RrSet:"+rr.ServiceNbr+":"+rr.MeasureDomain, strconv.FormatInt(rr.RrId, 10))
|
||||
if sadd.Err() != nil {
|
||||
l4g.RedisLog.Errorf("SAdd RrSet:%s:%s %d, err: %v", rr.ServiceNbr, rr.MeasureDomain, rr.RrId, sadd.Err())
|
||||
return sadd.Err()
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("SAdd RrSet:%s:%s %d", rr.ServiceNbr, rr.MeasureDomain, rr.RrId)
|
||||
}
|
||||
SetRrId2RrKey(rr.RrId, key)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbSetStrategyEventPriority(rr *RrData) error {
|
||||
err := rdb.HMSet(ctx, "Rr:"+rr.ServiceNbr+":"+rr.MeasureDomain+":"+strconv.FormatInt(rr.RrId, 10), "eventPriority", rr.EventPriority).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet Rr:%s, err: %v", rr.ServiceNbr, err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbSetOfrCalcPriority(rr *RrData) error {
|
||||
err := rdb.HMSet(ctx, "Rr:"+rr.ServiceNbr+":"+rr.MeasureDomain+":"+strconv.FormatInt(rr.RrId, 10), "calcPriority", rr.CalcPriority).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet Rr:%s, err: %v", rr.ServiceNbr, err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
// "Rr:"+rr.ServiceNbr+":"+rr.MeasureDomain+":"+strconv.Itoa(rr.RrId)
|
||||
func RdbUpdateRrRecordOnly(rrId int64, ratableVal, usedVal int64, beginTime, endTime string) (int, error) {
|
||||
key := GetRrKeyByRrId(rrId)
|
||||
if key != "" {
|
||||
err := rdb.HMSet(ctx, key, "beginTime", beginTime, "endTime", endTime, "freeValue", ratableVal,
|
||||
"usedValue", usedVal).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet %s, err: %v", key, err)
|
||||
return -1, err
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("HMSet %s, beginTime[%s], endTime[%s], free/used:[%d/%d]", key, beginTime, endTime, ratableVal, usedVal)
|
||||
return 1, nil
|
||||
}
|
||||
} else {
|
||||
l4g.RedisLog.Errorf("HMSet rrId[%d], err not found, beginTime[%s], endTime[%s], free/used:[%d/%d]", rrId, beginTime, endTime, ratableVal, usedVal)
|
||||
return 0, nil
|
||||
}
|
||||
/*var cursor uint64
|
||||
var err error
|
||||
for {
|
||||
var keys []string
|
||||
// 20 keys each scan
|
||||
keys, cursor, err = rdb.Scan(ctx, cursor, "Rr:*:"+strconv.FormatInt(rrId, 10), 2).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Scan Rr:*:%d, err: %v", rrId, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
err := rdb.HMSet(ctx, key, "beginTime", beginTime, "endTime", endTime, "freeValue", ratableVal,
|
||||
"usedValue", usedVal).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet %s, err: %v", key, err)
|
||||
}
|
||||
}
|
||||
if cursor == 0 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return err*/
|
||||
}
|
||||
|
||||
// "Rr:"+rr.ServiceNbr+":"+rr.MeasureDomain+":"+strconv.Itoa(rr.RrId)
|
||||
func RdbDelRrRecordByRrId(rrId int64) error {
|
||||
key := GetRrKeyByRrId(rrId)
|
||||
if key != "" {
|
||||
err := rdb.Del(ctx, key).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", key, err)
|
||||
} else {
|
||||
l4g.RedisLog.Infof("Del %s succ", key)
|
||||
subFields := strings.Split(key, ":")
|
||||
if len(subFields) == 4 {
|
||||
_ = rdb.SRem(ctx, "RrSet:"+subFields[1]+":"+subFields[2], strconv.FormatInt(rrId, 10))
|
||||
l4g.RedisLog.Infof("SRem %d from RrSet:%s:%s", rrId, subFields[1], subFields[2])
|
||||
//return srem.Err()
|
||||
}
|
||||
|
||||
}
|
||||
rdb.Del(ctx, "RrId:"+strconv.FormatInt(rrId, 10))
|
||||
l4g.RedisLog.Infof("Del RrId:%d", rrId)
|
||||
return nil
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// "Rr:"+rr.ServiceNbr+":"+rr.MeasureDomain+":"+strconv.Itoa(rr.RrId)
|
||||
func RdbDelRrRecordByServiceNbr(serviceNbr string) error {
|
||||
var cursor uint64
|
||||
var err error
|
||||
for {
|
||||
var keys []string
|
||||
// 20 keys each scan
|
||||
keys, cursor, err = rdb.Scan(ctx, cursor, "Rr:"+serviceNbr+":*", 30).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Scan Rr:%s:*, err: %v", serviceNbr, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
err = rdb.Del(ctx, key).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", key, err)
|
||||
} else {
|
||||
subFields := strings.Split(key, ":")
|
||||
if len(subFields) == 4 {
|
||||
_ = rdb.SRem(ctx, "RrSet:"+serviceNbr+":"+subFields[2], subFields[3])
|
||||
//return srem.Err()
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursor == 0 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbDelRrRecord(rr *RrData) error {
|
||||
err := rdb.Del(ctx, "Rr:"+rr.ServiceNbr+":"+rr.MeasureDomain+":"+strconv.FormatInt(rr.RrId, 10)).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del Rr:%s:%d, err: %v", rr.ServiceNbr, rr.RrId, err)
|
||||
} else {
|
||||
_ = rdb.SRem(ctx, "RrSet:"+rr.ServiceNbr+":"+rr.MeasureDomain, strconv.FormatInt(rr.RrId, 10))
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbSetOfrRecord(ofrId int) error {
|
||||
err := rdb.Set(ctx, "Ofr:"+strconv.Itoa(ofrId), ofrId, 0).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Set Ofr:%d, err: %v", ofrId, err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbCheckIfOfrExist(ofrId int) bool {
|
||||
n, err := rdb.Exists(ctx, "Ofr:"+strconv.Itoa(ofrId)).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Check Exists Ofr:%d, err: %v", ofrId, err)
|
||||
return false
|
||||
} else {
|
||||
if n >= 1 {
|
||||
return true
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func RdbSetPrefixRecord(prefix *PrefixData) error {
|
||||
sadd := rdb.SAdd(ctx, "Prefix:"+strconv.Itoa(prefix.StrategyId), prefix.AreaCode)
|
||||
return sadd.Err()
|
||||
}
|
||||
|
||||
// Add holiday discount table
|
||||
// Hd:tariff_id:tariff_seq:hdId, group, name, discount, state;;; type, subState, priority, begin, end;
|
||||
// triggers: 1, create, modify(state), delete of tb_bil_holiday_rel;
|
||||
// 2, modify(STATE), delete of tb_bil_holiday;
|
||||
/* holiday discount
|
||||
SELECT hd.HOLIDAY_GROUP, hd.HOLIDAY_PRIORITY, COUNT(1) AS HOLIDAY_COUNT FROM tb_bil_holiday hd
|
||||
WHERE (((hd.HOLIDAY_TYPE = '02' AND hd.HOLIDAY_BEGIN_DATE <= (WEEKDAY(SYSDATE()) + 1) AND hd.holiday_end_date >= (WEEKDAY(SYSDATE()) + 1))
|
||||
OR (hd.holiday_type = '03' AND DATE_FORMAT(SYSDATE(),'%m%d') >= DATE_FORMAT(CONCAT('2020', hd.HOLIDAY_BEGIN_DATE), '%m%d') AND DATE_FORMAT(SYSDATE(),'%m%d') <= DATE_FORMAT(CONCAT('2020', hd.holiday_end_date), '%m%d'))
|
||||
OR (hd.holiday_type = '04' AND DATE_FORMAT(SYSDATE(), '%H%i%s') >= DATE_FORMAT(STR_TO_DATE(CONCAT('20200101', hd.HOLIDAY_BEGIN_DATE), '%Y%m%d%H%i%s'),'%H%i%s')
|
||||
AND DATE_FORMAT(SYSDATE(), '%H%i%s') <= DATE_FORMAT(STR_TO_DATE(CONCAT('20200101', hd.holiday_end_date), '%Y%m%d%H%i%s'),'%H%i%s')))
|
||||
and hd.STATE='L0R')
|
||||
GROUP BY hd.HOLIDAY_PRIORITY, hd.HOLIDAY_GROUP;
|
||||
*/
|
||||
func RdbSetHolidayDiscountRecord(hd *HolidayDiscountData) error {
|
||||
err := rdb.HMSet(ctx, "Hd:"+strconv.Itoa(hd.TariffId)+":"+strconv.Itoa(hd.HdId),
|
||||
"ofrId", hd.OfrId, "tariffId", hd.TariffId, "tariffSeq", hd.TariffSeq, "hdGroup", hd.HdGroup,
|
||||
"hrState", hd.HrState, "hrDiscount", hd.HrDiscount, "hdId", hd.HdId, "hdName", hd.HdName,
|
||||
"hdType", hd.HdType, "hdState", hd.HdState, "hdPriority", hd.HdPriority,
|
||||
"hdBeginDate", hd.HdBeginDate, "hdEndDate", hd.HdEndDate).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet Hd:%d:%d, err: %v", hd.TariffId, hd.HdId, err)
|
||||
} else {
|
||||
sadd := rdb.SAdd(ctx, "HdSet:"+strconv.Itoa(hd.TariffId), strconv.Itoa(hd.HdId))
|
||||
if sadd.Err() != nil {
|
||||
l4g.RedisLog.Errorf("SAdd HdSet:%d %d, err: %v", hd.TariffId, hd.HdId, sadd.Err())
|
||||
return sadd.Err()
|
||||
}
|
||||
SetHdId2TariffId(hd.HdId, hd.TariffId)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbDelHolidayDiscountRecord(tariffId int) {
|
||||
var cursor uint64= 0
|
||||
for {
|
||||
// 20 keys each scan
|
||||
keys, cursorNew, err := rdb.SScan(ctx, "HdSet:"+strconv.Itoa(tariffId), cursor, "", 0).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("SScan HdSet:%d, err: %v", tariffId, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
hdKey := "Hd:"+strconv.Itoa(tariffId) + ":" + key
|
||||
err := rdb.Del(ctx, hdKey).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", key, err)
|
||||
} else {
|
||||
//subFields := strings.Split(key, ":")
|
||||
//if len(subFields) == 3 && subFields[2] != "" {
|
||||
_ = rdb.SRem(ctx, "HdSet:"+strconv.Itoa(tariffId), key)
|
||||
//return srem.Err()
|
||||
//}
|
||||
}
|
||||
}
|
||||
if cursorNew == 0 {
|
||||
break
|
||||
} else {
|
||||
cursor = cursorNew
|
||||
}
|
||||
}
|
||||
_ = rdb.Del(ctx, "HdSet:"+strconv.Itoa(tariffId)).Err()
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func RdbDelHolidayDiscountRecordByHd(holidayId int) error {
|
||||
tariffIdStr := GetTariffIdByHdId(holidayId)
|
||||
tariffId, err := strconv.Atoi(tariffIdStr)
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Get tariffId by HdId %d, err: %v", holidayId, err)
|
||||
} else {
|
||||
key := "Hd:" + strconv.Itoa(tariffId) + ":"+strconv.Itoa(holidayId)
|
||||
err := rdb.Del(ctx, key).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", key, err)
|
||||
} else {
|
||||
_ = rdb.SRem(ctx, "HdSet:"+tariffIdStr, strconv.Itoa(holidayId))
|
||||
}
|
||||
}
|
||||
/*var cursor uint64
|
||||
for {
|
||||
var keys []string
|
||||
// 20 keys each scan
|
||||
keys, cursor, err = rdb.Scan(ctx, cursor, "Hd:*:"+strconv.Itoa(holidayId), 200).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Scan Hd:*:%d, err: %v", holidayId, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
err := rdb.Del(ctx, key).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", key, err)
|
||||
} else {
|
||||
subFields := strings.Split(key, ":")
|
||||
if len(subFields) == 3 && subFields[1] != "" {
|
||||
_ = rdb.SRem(ctx, "HdSet:"+subFields[1], strconv.Itoa(holidayId))
|
||||
//return srem.Err()
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursor == 0 {
|
||||
break
|
||||
}
|
||||
}*/
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
// "Tariff:"+strconv.Itoa(tariff.OfrId)+":"+tariff.MeasureDomain+":"+strconv.Itoa(tariff.TariffId)
|
||||
func RdbDelTariffRecord(tariffId int) error {
|
||||
var cursor uint64
|
||||
var err error
|
||||
for {
|
||||
var keys []string
|
||||
// 20 keys each scan
|
||||
keys, cursor, err = rdb.Scan(ctx, cursor, "Tariff:*:"+strconv.Itoa(tariffId), 2).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Scan Tariff:*:%d, err: %v", tariffId, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
err := rdb.Del(ctx, key).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", key, err)
|
||||
} else {
|
||||
subFields := strings.Split(key, ":")
|
||||
if len(subFields) == 4 {
|
||||
_ = rdb.SRem(ctx, "TariffSet:"+subFields[1]+":"+subFields[2], strconv.Itoa(tariffId))
|
||||
//return srem.Err()
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursor == 0 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
// "Tariff:"+strconv.Itoa(tariff.OfrId)+":"+tariff.MeasureDomain+":"+strconv.Itoa(tariff.TariffId)
|
||||
func RdbDelTariffRecordByOfr(ofrId int) error {
|
||||
var cursor uint64
|
||||
var err error
|
||||
for {
|
||||
var keys []string
|
||||
// 20 keys each scan
|
||||
keys, cursor, err = rdb.Scan(ctx, cursor, "Tariff:"+strconv.Itoa(ofrId)+":*", 2).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Scan Tariff:%d:*, err: %v", ofrId, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
err := rdb.Del(ctx, key).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Del %s, err: %v", key, err)
|
||||
} else {
|
||||
subFields := strings.Split(key, ":")
|
||||
if len(subFields) == 4 {
|
||||
_ = rdb.SRem(ctx, "TariffSet:"+subFields[1]+":"+subFields[2], subFields[3])
|
||||
//return srem.Err()
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursor == 0 {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbDelPrefixRecord(strategyId int, areaCode string) error {
|
||||
srem := rdb.SRem(ctx, "Prefix:"+strconv.Itoa(strategyId), areaCode)
|
||||
return srem.Err()
|
||||
}
|
||||
|
||||
func RdbSetTariffRecord(tariff *TariffData) error {
|
||||
//&prd_inst_id, &service_nbr, &prd_inst_stas_id, &acct_id, &cust_id
|
||||
err := rdb.HMSet(ctx, "Tariff:"+strconv.Itoa(tariff.OfrId)+":"+tariff.MeasureDomain+":"+strconv.Itoa(tariff.TariffId), "tariffId", tariff.TariffId,
|
||||
"tariffSeq", tariff.TariffSeq, "measureDomain", tariff.MeasureDomain,
|
||||
"startTime", tariff.StartTime, "endTime", tariff.EndTime, "feeUnit", tariff.FeeUnit,
|
||||
"rateUnit", tariff.RateUnit, "calcPriority", tariff.CalcPriority, "eventPriority", tariff.EventPriority, "ofrId", tariff.OfrId,
|
||||
"ofrName", tariff.OfrName, "strategyId", tariff.StrategyId, "strategyName", tariff.StrategyName).Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HMSet Tariff:%d, err: %v", tariff.TariffId, err)
|
||||
} else {
|
||||
sadd := rdb.SAdd(ctx, "TariffSet:"+strconv.Itoa(tariff.OfrId)+":"+tariff.MeasureDomain, strconv.Itoa(tariff.TariffId))
|
||||
if sadd.Err() != nil {
|
||||
l4g.RedisLog.Errorf("SAdd TariffSet:%d:%s %d, err: %v", tariff.OfrId, tariff.MeasureDomain, tariff.TariffId, sadd.Err())
|
||||
return sadd.Err()
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func RdbGetRrByServiceNbrAndDomain(serviceNbrAndDomain string) (rsp string, err error) {
|
||||
var cursor uint64= 0
|
||||
for {
|
||||
keys, cursorNew, err := rdb.SScan(ctx, "RrSet:"+serviceNbrAndDomain, cursor, "", 0).Result()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("SScan RrSet:%s, err: %v", serviceNbrAndDomain, err)
|
||||
break
|
||||
}
|
||||
|
||||
//fmt.Printf("\nfound %d keys\n", len(keys))
|
||||
for _, key := range keys {
|
||||
rrKey := "Rr:"+serviceNbrAndDomain+":"+key
|
||||
res := rdb.HGetAll(ctx, rrKey)
|
||||
if res.Err() != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll %s, err: %v", rrKey, err)
|
||||
return "err: HGetAll "+rrKey, err
|
||||
} else {
|
||||
var rrData RrData
|
||||
if err = res.Scan(&rrData); err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll Scan %s, err: %v", rrKey, err)
|
||||
return "err: Scan "+rrKey, err
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Get %s, [%v]", rrKey, rrData)
|
||||
rsp += fmt.Sprintf("%s:***************************************\r\nPrdInstId: %d\r\nOfrInstId: %d\r\nOfrId: %d\r\nOfrName: %s\r\nBeginTime: %s\r\nEndTime: %s\r\n",
|
||||
rrKey, rrData.PrdInstId, rrData.OfrInstId, rrData.OfrId, rrData.OfrName, rrData.BeginTime, rrData.EndTime)
|
||||
rsp += fmt.Sprintf("FreeValue: %d\r\nUsedValue: %d\r\nTariffId: %d\r\nStrategyId: %d\r\nCalcPriority: %d\r\nEventPriority: %d\r\nRateUnit: %d\r\nUnitFee: %d\r\n",
|
||||
rrData.FreeValue, rrData.UsedValue, rrData.TariffId, rrData.StrategyId, rrData.CalcPriority, rrData.EventPriority, rrData.RateUnit, rrData.UnitFee)
|
||||
}
|
||||
}
|
||||
}
|
||||
if cursorNew == 0 {
|
||||
break
|
||||
} else {
|
||||
cursor = cursorNew
|
||||
}
|
||||
}
|
||||
|
||||
return rsp, err
|
||||
}
|
||||
|
||||
func RdbGetTariffByOfrId(ofrId string) (*TariffData, error) {
|
||||
keys := rdb.Keys(ctx, "Tariff:"+ofrId+":*")
|
||||
err := keys.Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("Get Keys of Tariff:%s:*, err: %v", ofrId, err)
|
||||
return nil, err
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Get Keys of Tariff:%s:*, keys:%v", ofrId, keys.Val())
|
||||
}
|
||||
|
||||
// Scan the results into the struct.
|
||||
var ta TariffData
|
||||
for _, value := range keys.Val() {
|
||||
// Get the map. The same approach works for HmGet().
|
||||
res := rdb.HGetAll(ctx, value)
|
||||
err := res.Err()
|
||||
if err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll %s, err: %v", value, err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err = res.Scan(&ta); err != nil {
|
||||
l4g.RedisLog.Errorf("HGetAll Scan %s, err: %v", value, err)
|
||||
} else {
|
||||
l4g.RedisLog.Debugf("Get %s, [%v]", value, ta)
|
||||
|
||||
// Get prefix related tariff
|
||||
sMembers := rdb.SMembers(ctx, "Prefix:"+strconv.Itoa(ta.StrategyId))
|
||||
if sMembers.Err() == nil {
|
||||
l4g.RedisLog.Debugf("Prefix, [%v]", sMembers.Val())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return &ta, err
|
||||
}
|
||||
Reference in New Issue
Block a user