You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.5 KiB
116 lines
3.5 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
|
|
"go-common/app/job/main/thumbup/model" |
|
"go-common/library/cache/redis" |
|
"go-common/library/log" |
|
) |
|
|
|
func userLikesKey(businessID, mid int64, state int8) string { |
|
return fmt.Sprintf("u_m_%d_b_%d_%d", mid, businessID, state) |
|
} |
|
|
|
// ExpireUserLikesCache . |
|
func (d *Dao) ExpireUserLikesCache(c context.Context, mid, businessID int64, state int8) (ok bool, err error) { |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := userLikesKey(businessID, mid, state) |
|
if ok, err = redis.Bool(conn.Do("EXPIRE", key, d.redisUserLikesExpire)); err != nil { |
|
log.Errorv(c, log.KV("log", fmt.Sprintf("conn.Send(EXPIRE, %s) error(%v)", key, err))) |
|
} |
|
return |
|
} |
|
|
|
// AppendCacheUserLikeList . |
|
func (d *Dao) AppendCacheUserLikeList(c context.Context, mid int64, item *model.ItemLikeRecord, businessID int64, state int8, limit int) (err error) { |
|
if item == nil { |
|
return |
|
} |
|
var count int |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := userLikesKey(businessID, mid, state) |
|
id := item.MessageID |
|
score := int64(item.Time) |
|
if err = conn.Send("ZADD", key, "CH", score, id); err != nil { |
|
log.Errorv(c, log.KV("log", fmt.Sprintf("conn.Send(ZADD, %s, %d, %v) error(%v)", key, score, id, err))) |
|
return |
|
} |
|
count++ |
|
if err = conn.Send("ZREMRANGEBYRANK", key, 0, -(limit + 1)); err != nil { |
|
log.Errorv(c, log.KV("log", fmt.Sprintf("conn.Send(ZREMRANGEBYRANK, %s, 0, %d) error(%v)", key, -(limit+1), err))) |
|
return |
|
} |
|
count++ |
|
if err = conn.Send("EXPIRE", key, d.redisUserLikesExpire); err != nil { |
|
log.Errorv(c, log.KV("log", fmt.Sprintf("conn.Send(EXPIRE, %s, %d) error(%v)", key, d.redisUserLikesExpire, err))) |
|
return |
|
} |
|
count++ |
|
if err = conn.Flush(); err != nil { |
|
log.Errorv(c, log.KV("log", fmt.Sprintf("conn.Flush error(%v)", err))) |
|
return |
|
} |
|
for i := 0; i < count; i++ { |
|
if _, err = conn.Receive(); err != nil { |
|
log.Errorv(c, log.KV("log", fmt.Sprintf("conn.Receive error(%v)", err))) |
|
return |
|
} |
|
} |
|
return |
|
} |
|
|
|
// AddUserLikesCache . |
|
func (d *Dao) AddUserLikesCache(c context.Context, mid, businessID int64, items []*model.ItemLikeRecord, typ int8, limit int) (err error) { |
|
if len(items) == 0 { |
|
return |
|
} |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := userLikesKey(businessID, mid, typ) |
|
if err = conn.Send("DEL", key); err != nil { |
|
log.Errorv(c, log.KV("AddCacheUserLikeList", fmt.Sprintf("conn.Send(DEL, %s) error(%+v)", key, err))) |
|
return |
|
} |
|
args := redis.Args{}.Add(key).Add("CH") |
|
for _, item := range items { |
|
args = args.Add(int64(item.Time)).Add(item.MessageID) |
|
} |
|
if err = conn.Send("ZADD", args...); err != nil { |
|
log.Error("zadd key(%s) args(%v) error(%v)", key, args, err) |
|
return |
|
} |
|
if err = conn.Send("ZREMRANGEBYRANK", key, 0, -(limit + 1)); err != nil { |
|
log.Error("zremrangebyrank error(%v)", err) |
|
return |
|
} |
|
if err = conn.Send("EXPIRE", key, d.redisUserLikesExpire); err != nil { |
|
log.Error("expire key(%s) error(%v)", key, err) |
|
return |
|
} |
|
if err = conn.Flush(); err != nil { |
|
log.Error("redis flush error(%v)", err) |
|
return |
|
} |
|
for i := 0; i < 4; i++ { |
|
if _, err = conn.Receive(); err != nil { |
|
log.Error("redis receive error(%v)", err) |
|
return |
|
} |
|
} |
|
return |
|
} |
|
|
|
// DelUserLikeCache . |
|
func (d *Dao) DelUserLikeCache(c context.Context, mid, businessID int64, messageID int64, state int8) (err error) { |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := userLikesKey(businessID, mid, state) |
|
if _, err = conn.Do("ZREM", key, messageID); err != nil { |
|
log.Errorv(c, log.KV("log", fmt.Sprintf("conn.Send(ZREM, %s, %v) error(%v)", key, messageID, err))) |
|
} |
|
return |
|
}
|
|
|