76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"go-common/app/service/bbq/topic/internal/model"
|
||
"go-common/library/cache/redis"
|
||
"go-common/library/ecode"
|
||
"go-common/library/log"
|
||
"go-common/library/xstr"
|
||
)
|
||
|
||
/*各种情况下，是否需要查询db，前三列表示情景条件
|
||
NEXT offset rank 是否需要db操作
|
||
1 1
|
||
0 1 0 0
|
||
0 n>1 0 1
|
||
0 0 n>0 0
|
||
*/
|
||
func parseCursor(ctx context.Context, cursorPrev, cursorNext string) (cursor model.CursorValue, directionNext bool, err error) {
|
||
// 判断是向前还是向后查询
|
||
directionNext = true
|
||
cursorStr := cursorNext
|
||
if len(cursorNext) == 0 && len(cursorPrev) > 0 {
|
||
directionNext = false
|
||
cursorStr = cursorPrev
|
||
}
|
||
// 解析cursor中的cursor_id
|
||
if len(cursorStr) != 0 {
|
||
var cursorData = []byte(cursorStr)
|
||
err = json.Unmarshal(cursorData, &cursor)
|
||
if err != nil {
|
||
err = ecode.ReqParamErr
|
||
return
|
||
}
|
||
}
|
||
|
||
// 最后做一次校验，保证cursor的值是对的
|
||
if (cursor.StickRank > 0 && cursor.Offset > 0) || (!directionNext && cursor.Offset == 0 && cursor.StickRank == 0) {
|
||
err = ecode.TopicReqParamErr
|
||
log.Errorw(ctx, "log", "cursor value error", "prev", cursorPrev, "next", cursorNext)
|
||
return
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (d *Dao) getRedisList(ctx context.Context, key string) (list []int64, err error) {
|
||
conn := d.redis.Get(ctx)
|
||
defer conn.Close()
|
||
str, err := redis.String(conn.Do("GET", key))
|
||
if err == redis.ErrNil {
|
||
err = nil
|
||
return
|
||
}
|
||
if err != nil {
|
||
log.Errorw(ctx, "log", "get redis list fail", "key", key)
|
||
return
|
||
}
|
||
list, err = xstr.SplitInts(str)
|
||
if err != nil {
|
||
log.Errorw(ctx, "log", "split list_str fail", "key", key, "str", str)
|
||
return
|
||
}
|
||
return
|
||
}
|
||
func (d *Dao) setRedisList(ctx context.Context, key string, list []int64) (err error) {
|
||
conn := d.redis.Get(ctx)
|
||
defer conn.Close()
|
||
if _, err = conn.Do("SET", key, xstr.JoinInts(list)); err != nil {
|
||
log.Errorw(ctx, "log", "set redis list fail", "key", key, "list", list)
|
||
return
|
||
}
|
||
return
|
||
}
|