Sillyv2/core/bucket.go

package core

import (
	"errors"
	"fmt"
	"strconv"
	"strings"
	"sync"

	"github.com/cdle/sillyplus/core/logs"
	"github.com/cdle/sillyplus/core/storage"
	"github.com/cdle/sillyplus/core/storage/boltdb"
	"github.com/cdle/sillyplus/core/storage/redis"
	"github.com/cdle/sillyplus/utils"
	"github.com/goccy/go-json"
)

var bkt storage.Bucket
var HttpPort string
var sillyGirl = MakeBucket("sillyGirl")

// var Get = func(key string) string {
// 	return ""
// }
// var Set = func(key, value string, expiration time.Duration) error {
// 	return nil
// }

var MakeBucketlocker sync.Mutex

func MakeBucket(name string) storage.Bucket {
	MakeBucketlocker.Lock()
	defer MakeBucketlocker.Unlock()
	if bkt == nil {
		// utils.ReadYaml(utils.ExecPath+"/conf/", &Config, "https://raw.githubusercontent.com/cdle/sillyplus/main/conf/demo_config.yaml")
		// if !Config.EnableRedis {
		bkt = boltdb.Initsillyplus()
		// Get = boltdb.Get
		// Set = boltdb.Set
		// logs.Info("默认使用boltdb进行数据存储")
		// } else {
		// 	bkt = redis.Initsillyplus(Config.RedisAddr, Config.RedisPassword)
		// 	Get = redis.Get
		// 	Set = redis.Set
		// 	logs.Info("已使用redis进行数据存储")
		// }
		var app = bkt
		isredis := false
		if def := bkt.GetString("storage"); def == "redis" {
			func() {
				defer func() {
					err := recover()
					if err != nil {
						// console.Warn("redis异常，已默认启用boltdb进行数据存储")
						bkt = app
					} else {
						isredis = true
						logs.Info("已使用redis进行数据存储")
					}
				}()
				bkt = redis.Initsillyplus(app.GetString("redis_addr"), app.GetString("redis_password"))
			}()
		} else {
			if def != "boltdb" {
				bkt.Set("storage", "boltdb")
			}
			logs.Info("默认使用boltdb进行数据存储")
		}
		storage.Watch(bkt, "storage", func(old, new, key string) *storage.Final {
			if isredis {
				if new == "boltdb" {
					app.Set2(key, new)
					return &storage.Final{
						Message: "重启生效！",
					}
				}
			} else {
				if new != "redis" {
					return nil
				}
				message := "Redis连接成功，重启生效！"
				err := redis.Try(app.GetString("redis_addr"), app.GetString("redis_password"))
				if err != nil {
					message = "Redis连接失败，操作无效：" + err.Error()
					return &storage.Final{
						Error: errors.New(message),
					}
				} else {
					return &storage.Final{
						Message: message,
					}
				}
			}
			return nil
		})
		if !isredis {
			storage.Watch(app, "redis_addr", func(old, new, _ string) *storage.Final {
				message := "Redis连接成功，重启生效！"
				err := redis.Try(new, app.GetString("redis_password"))
				if err != nil {
					message = "Redis连接失败：" + err.Error()
				}
				return &storage.Final{
					Message: message,
				}
			})
			storage.Watch(app, "redis_password", func(old, new, _ string) *storage.Final {
				message := "Redis连接成功，重启生效！"
				err := redis.Try(app.GetString("redis_addr"), new)
				if err != nil {
					message = "Redis连接失败：" + err.Error()
				}
				return &storage.Final{
					Message: message,
				}
			})
		}
		for _, name := range bkt.Buckets() {
			b := bkt.Copy(name)
			keys, err := b.Keys()
			if len(keys) == 0 && err == nil {
				b.Delete()
			}
		}

	}
	if name == "" {
		name = "sillyGirl"
	}
	if name == "silly" || name == "app" {
		name = "sillyGirl"
	}
	return bkt.Copy(name)
}

func TransformBucketKeyValue(v string) interface{} {
	var result interface{}
	if strings.HasPrefix(v, "f:") {
		result, _ = strconv.ParseFloat(strings.Replace(v, "f:", "", 1), 64)
		return result
	}
	if strings.HasPrefix(v, "d:") {
		result = utils.Int(strings.Replace(v, "d:", "", 1))
		return result
	}
	if strings.HasPrefix(v, "b:") {
		result = strings.Replace(v, "b:", "", 1) == "true"
		return result
	}
	if strings.HasPrefix(v, "o:") {
		json.Unmarshal([]byte(strings.Replace(v, "o:", "", 1)), &result)
		return result
	}
	if v == "" {
		return nil
	}
	return v
}

func GetBucketKeyValue(bucket storage.Bucket, ps ...interface{}) interface{} {
	var key interface{}
	var value interface{}
	if len(ps) == 0 {
		return nil
	}
	if len(ps) > 0 {
		key = ps[0]
	}
	if len(ps) > 1 {
		value = ps[1]
	}
	v := bucket.GetString(key)
	var result = TransformBucketKeyValue(v)
	if result == nil {
		return value
	}
	return result
}

func SetBucketKeyValue(bucket storage.Bucket, key interface{}, value interface{}) (string, error) {
	new := ""
	switch value := value.(type) {
	case int, int64, int32, uint:
		new = fmt.Sprintf("d:%d", value)
	case float32, float64:
		new = fmt.Sprintf("f:%f", value)
	case string, []byte:
		new = fmt.Sprintf("%s", value)
	case bool:
		new = fmt.Sprintf("b:%t", value)
	case nil:
		new = ""
	default:
		new = fmt.Sprintf("o:%s", utils.JsonMarshal(value))
	}
	return bucket.Set(key, new)
}