|
@@ -9,7 +9,19 @@ import (
|
|
|
"gogs.daxia.dev/huanan/pkg.daxia.dev.git/rdb"
|
|
|
)
|
|
|
|
|
|
+var Prefix = ""
|
|
|
+
|
|
|
+func InitPrefix(prefix string) {
|
|
|
+ Prefix = prefix
|
|
|
+}
|
|
|
+
|
|
|
func GetString(key string, defaultVal string) string {
|
|
|
+ if Prefix == "" {
|
|
|
+ logrus.Fatal("prefix没设置")
|
|
|
+ }
|
|
|
+
|
|
|
+ key = Prefix + key
|
|
|
+
|
|
|
data, err := rdb.GetRDB().Get(context.Background(), key).Result()
|
|
|
if err != nil {
|
|
|
logrus.Error(err)
|
|
@@ -20,6 +32,12 @@ func GetString(key string, defaultVal string) string {
|
|
|
}
|
|
|
|
|
|
func GetInt(key string, defaultVal int) int {
|
|
|
+ if Prefix == "" {
|
|
|
+ logrus.Fatal("prefix没设置")
|
|
|
+ }
|
|
|
+
|
|
|
+ key = Prefix + key
|
|
|
+
|
|
|
data, err := rdb.GetRDB().Get(context.Background(), key).Result()
|
|
|
if err != nil {
|
|
|
logrus.Error(err)
|
|
@@ -30,6 +48,12 @@ func GetInt(key string, defaultVal int) int {
|
|
|
}
|
|
|
|
|
|
func GetInt64(key string, defaultVal int64) int64 {
|
|
|
+ if Prefix == "" {
|
|
|
+ logrus.Fatal("prefix没设置")
|
|
|
+ }
|
|
|
+
|
|
|
+ key = Prefix + key
|
|
|
+
|
|
|
data, err := rdb.GetRDB().Get(context.Background(), key).Result()
|
|
|
if err != nil {
|
|
|
logrus.Error(err)
|
|
@@ -44,6 +68,12 @@ func SetInt(key string, val int) error {
|
|
|
}
|
|
|
|
|
|
func SetString(key string, val string) error {
|
|
|
+ if Prefix == "" {
|
|
|
+ logrus.Fatal("prefix没设置")
|
|
|
+ }
|
|
|
+
|
|
|
+ key = Prefix + key
|
|
|
+
|
|
|
_, err := rdb.GetRDB().Set(context.Background(), key, val, 0).Result()
|
|
|
if err != nil {
|
|
|
logrus.Error(err)
|
|
@@ -52,3 +82,42 @@ func SetString(key string, val string) error {
|
|
|
|
|
|
return nil
|
|
|
}
|
|
|
+
|
|
|
+func ListAll() (map[string]string, error) {
|
|
|
+ if Prefix == "" {
|
|
|
+ logrus.Fatal("prefix没设置")
|
|
|
+ }
|
|
|
+
|
|
|
+ var err error
|
|
|
+
|
|
|
+ resultMap := map[string]string{}
|
|
|
+ var cursor uint64
|
|
|
+ for {
|
|
|
+ var keys []string
|
|
|
+ keys, cursor, err = rdb.GetRDB().Scan(context.Background(), cursor, Prefix+"*", 0).Result()
|
|
|
+ if err != nil {
|
|
|
+ logrus.Error(err)
|
|
|
+
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+ for _, key := range keys {
|
|
|
+ //fmt.Println("key", key)
|
|
|
+ data, err := rdb.GetRDB().Get(context.Background(), key).Result()
|
|
|
+ if err != nil {
|
|
|
+ logrus.Error(err)
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+ resultMap[key] = data
|
|
|
+ }
|
|
|
+
|
|
|
+ // 没有更多key了
|
|
|
+ if cursor == 0 {
|
|
|
+ break
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return resultMap, nil
|
|
|
+
|
|
|
+}
|