From 36da39e5ab0fe02733ed7fb602fbc708b94682a1 Mon Sep 17 00:00:00 2001 From: withchao <993506633@qq.com> Date: Tue, 18 Apr 2023 18:19:25 +0800 Subject: [PATCH] config --- pkg/common/db/cache/rockscache.go | 41 ------------------------------- pkg/common/db/controller/group.go | 2 +- 2 files changed, 1 insertion(+), 42 deletions(-) diff --git a/pkg/common/db/cache/rockscache.go b/pkg/common/db/cache/rockscache.go index 6d81aa456..2a7b0fe7d 100644 --- a/pkg/common/db/cache/rockscache.go +++ b/pkg/common/db/cache/rockscache.go @@ -4,9 +4,6 @@ import ( "context" "encoding/json" "errors" - "fmt" - "runtime/debug" - "strings" "time" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/log" @@ -69,105 +66,67 @@ func GetDefaultOpt() rockscache.Options { func getCache[T any](ctx context.Context, rcClient *rockscache.Client, key string, expire time.Duration, fn func(ctx context.Context) (T, error)) (T, error) { var t T - var arr []string - arr = append(arr, "-------------------getCache---------------------") - arr = append(arr, time.Now().String()) - arr = append(arr, fmt.Sprintf("key: <%s> expire: <%s>", key, expire.String())) - defer func() { - arr = append(arr, fmt.Sprintf("%#v", t)) - arr = append(arr, string(debug.Stack())) - arr = append(arr, "----------------------------------------") - fmt.Println(strings.Join(arr, "\n")) - }() var write bool v, err := rcClient.Fetch2(ctx, key, expire, func() (s string, err error) { - arr = append(arr, "find db") t, err = fn(ctx) if err != nil { - arr = append(arr, fmt.Sprintf("fn error %s", err)) return "", err } - arr = append(arr, fmt.Sprintf("fn value %#v", t)) bs, err := json.Marshal(t) if err != nil { - arr = append(arr, fmt.Sprintf("json marshal %s", err)) return "", utils.Wrap(err, "") } write = true - arr = append(arr, fmt.Sprintf("json value %d", len(bs))) - arr = append(arr, string(bs)) - arr = append(arr, "****************") return string(bs), nil }) if err != nil { - arr = append(arr, "fetch error "+err.Error()) return t, err } if write { - arr = append(arr, "first return") return t, nil } err = json.Unmarshal([]byte(v), &t) if err != nil { - arr = append(arr, "json.Unmarshal error "+err.Error()) log.ZError(ctx, "cache json.Unmarshal failed", err, "key", key, "value", v, "expire", expire) return t, utils.Wrap(err, "") } - arr = append(arr, "success") return t, nil } func batchGetCache[T any](ctx context.Context, rcClient *rockscache.Client, keys []string, expire time.Duration, keyIndexFn func(t T, keys []string) (int, error), fn func(ctx context.Context) ([]T, error)) ([]T, error) { - var arr []string - arr = append(arr, "-------------------batchGetCache---------------------") - arr = append(arr, time.Now().String()) - arr = append(arr, fmt.Sprintf("keys: <%#v> expire: <%s>", keys, expire.String())) - defer func() { - arr = append(arr, string(debug.Stack())) - arr = append(arr, "----------------------------------------") - fmt.Println(strings.Join(arr, "\n")) - }() batchMap, err := rcClient.FetchBatch2(ctx, keys, expire, func(idxs []int) (m map[int]string, err error) { values := make(map[int]string) tArrays, err := fn(ctx) if err != nil { - arr = append(arr, "fn error "+err.Error()) return nil, err } for _, v := range tArrays { index, err := keyIndexFn(v, keys) if err != nil { - arr = append(arr, "keyIndexFn continue "+err.Error()) continue } bs, err := json.Marshal(v) if err != nil { - arr = append(arr, "json.Marshal "+err.Error()) return nil, utils.Wrap(err, "marshal failed") } values[index] = string(bs) } - arr = append(arr, fmt.Sprintf("rcClient.FetchBatch2 %#v", values)) return values, nil }) if err != nil { - arr = append(arr, "rcClient.FetchBatch2 error "+err.Error()) return nil, err } - arr = append(arr, fmt.Sprintf("rcClient.FetchBatch2 %#v", batchMap)) var tArrays []T for _, v := range batchMap { if v != "" { var t T err = json.Unmarshal([]byte(v), &t) if err != nil { - arr = append(arr, "json.Unmarshal error "+err.Error()) return nil, utils.Wrap(err, "unmarshal failed") } tArrays = append(tArrays, t) } } - arr = append(arr, fmt.Sprintf("tArrays %#v", tArrays)) return tArrays, nil } diff --git a/pkg/common/db/controller/group.go b/pkg/common/db/controller/group.go index 58475c30b..0a0170047 100644 --- a/pkg/common/db/controller/group.go +++ b/pkg/common/db/controller/group.go @@ -128,7 +128,7 @@ func (g *groupDatabase) CreateGroup(ctx context.Context, groups []*relationTb.Gr cache = cache.DelGroupMemberIDs(groupMember.GroupID).DelGroupMembersHash(groupMember.GroupID).DelJoinedGroupID(groupMember.UserID).DelGroupsMemberNum(groupMember.GroupID) } } - return g.cache.ExecDel(ctx) + return cache.ExecDel(ctx) }) }