|
|
|
@ -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
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|