103 lines
3.3 KiB
Go
103 lines
3.3 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"go-common/app/interface/bbq/app-bbq/model"
|
||
"go-common/library/cache/redis"
|
||
"go-common/library/time"
|
||
"go-common/library/xstr"
|
||
"strconv"
|
||
|
||
"go-common/library/log"
|
||
)
|
||
|
||
// FetchAvailableOutboxList 根据提供的up主mids去获取比lastSvID还小的svid
|
||
// @param CursorID 函数并不判断SvID正确性，由调用方保证
|
||
// @param cursorNext 表明fetch的方向，同时会影响排序顺序
|
||
// true： 则有如下条件sql"... and id < {{svid}} order by id desc..."
|
||
// false： 则有如下条件sql"... and id > {{svid}} order by id asc..."
|
||
// @return svIDs 注意：svid的返回有顺序
|
||
func (d *Dao) FetchAvailableOutboxList(c context.Context, fetchStates string, mids []int64, cursorNext bool, cursorSvID int64, cursorPubtime time.Time, size int) (svIDs []int64, err error) {
|
||
if len(mids) == 0 {
|
||
return
|
||
}
|
||
compareSymbol := string(">=")
|
||
orderDirection := "asc"
|
||
if cursorNext {
|
||
compareSymbol = "<="
|
||
orderDirection = "desc"
|
||
}
|
||
midStr := xstr.JoinInts(mids)
|
||
// 多个字段order by，每个字段都要指定desc，否则不带的字段为asc
|
||
querySQL := fmt.Sprintf("select svid, pubtime from video where mid in (%s) and state in (%s) and "+
|
||
"pubtime %s ? order by pubtime %s, svid %s limit %d",
|
||
midStr, fetchStates, compareSymbol, orderDirection, orderDirection, size)
|
||
log.V(1).Infov(c, log.KV("event", "mysql_select"), log.KV("table", "video"),
|
||
log.KV("sql", querySQL))
|
||
rows, err := d.db.Query(c, querySQL, cursorPubtime)
|
||
if err != nil {
|
||
log.Errorv(c, log.KV("event", "mysql_select"), log.KV("table", "video"),
|
||
log.KV("sql", querySQL))
|
||
return
|
||
}
|
||
defer rows.Close()
|
||
var svID int64
|
||
var pubTIme time.Time
|
||
conflict := bool(true)
|
||
for rows.Next() {
|
||
if err = rows.Scan(&svID, &pubTIme); err != nil {
|
||
log.Errorv(c, log.KV("event", "mysql_scan"), log.KV("table", "video"),
|
||
log.KV("sql", querySQL))
|
||
return
|
||
}
|
||
// 为了解决同一个pubtime的冲突问题
|
||
if pubTIme == cursorPubtime && conflict {
|
||
if svID == cursorSvID {
|
||
conflict = false
|
||
}
|
||
continue
|
||
}
|
||
svIDs = append(svIDs, svID)
|
||
}
|
||
log.Infov(c, log.KV("event", "mysql_select"), log.KV("table", "video"),
|
||
log.KV("desc", "fetch outbox"), log.KV("size", len(svIDs)))
|
||
return
|
||
}
|
||
|
||
// SetMIDLastPubtime redis设置
|
||
func (d *Dao) SetMIDLastPubtime(c context.Context, mid int64, pubtime int64) (err error) {
|
||
key := fmt.Sprintf(model.CacheKeyLastPubtime, mid)
|
||
conn := d.redis.Get(c)
|
||
defer conn.Close()
|
||
_, err = conn.Do("set", key, pubtime, "ex", model.CacheExpireLastPubtime)
|
||
if err != nil {
|
||
log.Errorv(c, log.KV("event", "redis_set"), log.KV("key", key), log.KV("value", pubtime))
|
||
}
|
||
return
|
||
}
|
||
|
||
// GetMIDLastPubtime 获取该用户feed中的最新浏览svid
|
||
func (d *Dao) GetMIDLastPubtime(c context.Context, mid int64) (pubtime int64, err error) {
|
||
pubtime = 0
|
||
key := fmt.Sprintf(model.CacheKeyLastPubtime, mid)
|
||
conn := d.redis.Get(c)
|
||
defer conn.Close()
|
||
var data []byte
|
||
if data, err = redis.Bytes(conn.Do("get", key)); err != nil {
|
||
if err == redis.ErrNil {
|
||
err = nil
|
||
log.V(1).Infov(c, log.KV("event", "redis_get"), log.KV("key", key), log.KV("result", "not_found"))
|
||
} else {
|
||
log.Errorv(c, log.KV("event", "redis_get"), log.KV("key", key))
|
||
}
|
||
return
|
||
}
|
||
|
||
pubtime, err = strconv.ParseInt(string(data), 10, 0)
|
||
if err != nil {
|
||
log.Errorv(c, log.KV("event", "redis_get"), log.KV("key", key), log.KV("value", data))
|
||
}
|
||
return
|
||
}
|