blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'errcode' of github.com:OpenIMSDK/Open-IM-Server into errcode

Browse Source
This commit is contained in:
wangchuxiao 2023-04-18 18:37:55 +08:00
commit ba45e9d50c
1 changed files with 1 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
pkg/common/db/cache/rockscache.go vendored
View File

@ -4,9 +4,6 @@ import (
"context"
"encoding/json"
"errors"
"fmt"
"runtime/debug"
"strings"
"time"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
@ -70,42 +67,23 @@ 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
}
if v == "" {
@ -113,64 +91,46 @@ func getCache[T any](ctx context.Context, rcClient *rockscache.Client, key strin
}
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
}