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.
106 lines
2.9 KiB
106 lines
2.9 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
|
|
"go-common/app/service/main/thumbup/model" |
|
"go-common/library/cache/redis" |
|
"go-common/library/log" |
|
"go-common/library/xstr" |
|
) |
|
|
|
func hashStatsKey(businessID, originID int64) string { |
|
return fmt.Sprintf("stats_o_%d_b_%d", originID, businessID) |
|
} |
|
|
|
// ExpireHashStatsCache . |
|
func (d *Dao) ExpireHashStatsCache(c context.Context, businessID, originID int64) (ok bool, err error) { |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := hashStatsKey(businessID, originID) |
|
if ok, err = redis.Bool(conn.Do("EXPIRE", key, d.redisStatsExpire)); err != nil { |
|
PromError("redis:计数缓存设定过期") |
|
log.Error("conn.Do(EXPIRE, %s, %d) error(%v)", key, d.redisStatsExpire, err) |
|
} |
|
return |
|
} |
|
|
|
// DelHashStatsCache del hash cache |
|
func (d *Dao) DelHashStatsCache(c context.Context, businessID, originID int64) (err error) { |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := hashStatsKey(businessID, originID) |
|
if _, err = conn.Do("del", key); err != nil { |
|
PromError("redis:计数缓存删除") |
|
log.Error("conn.Do(DEL, %s) error(%v)", key, err) |
|
} |
|
return |
|
} |
|
|
|
// HashStatsCache . |
|
func (d *Dao) HashStatsCache(c context.Context, businessID, originID int64, messageIDs []int64) (res map[int64]*model.Stats, err error) { |
|
if len(messageIDs) == 0 { |
|
return |
|
} |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := hashStatsKey(businessID, originID) |
|
var ss []string |
|
var commonds []interface{} |
|
commonds = append(commonds, key) |
|
for _, m := range messageIDs { |
|
commonds = append(commonds, m) |
|
} |
|
if ss, err = redis.Strings(conn.Do("HMGET", commonds...)); err != nil { |
|
if err == redis.ErrNil { |
|
err = nil |
|
} else { |
|
log.Error("conn.Do(HMGET, %s, %v) error(%v)", key, messageIDs, err) |
|
PromError("redis:获取统计信息") |
|
} |
|
return |
|
} |
|
res = make(map[int64]*model.Stats) |
|
for i, id := range messageIDs { |
|
if ss[i] == "" { |
|
continue |
|
} |
|
stat := &model.Stats{ID: id, OriginID: originID} |
|
num, _ := xstr.SplitInts(ss[i]) |
|
if len(num) > 1 { |
|
stat.Likes = num[0] |
|
stat.Dislikes = num[1] |
|
} |
|
res[id] = stat |
|
} |
|
return |
|
} |
|
|
|
// AddHashStatsCache . |
|
func (d *Dao) AddHashStatsCache(c context.Context, businessID, originID int64, stats ...*model.Stats) (err error) { |
|
if len(stats) == 0 { |
|
return |
|
} |
|
conn := d.redis.Get(c) |
|
defer conn.Close() |
|
key := hashStatsKey(businessID, originID) |
|
var commonds = []interface{}{key} |
|
for _, stat := range stats { |
|
commonds = append(commonds, stat.ID, xstr.JoinInts([]int64{stat.Likes, stat.Dislikes})) |
|
} |
|
if _, err = conn.Do("HMSET", commonds...); err != nil { |
|
PromError("redis:增加统计信息") |
|
log.Error("conn.DO(HMSET, %s, %v) error(%v)", key, commonds, err) |
|
} |
|
return |
|
} |
|
|
|
// AddHashStatsCacheMap . |
|
func (d *Dao) AddHashStatsCacheMap(c context.Context, businessID, originID int64, stats map[int64]*model.Stats) (err error) { |
|
var s []*model.Stats |
|
for _, v := range stats { |
|
s = append(s, v) |
|
} |
|
return d.AddHashStatsCache(c, businessID, originID, s...) |
|
}
|
|
|