112 lines
3.5 KiB
Go
112 lines
3.5 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
xsql "database/sql"
|
||
"encoding/json"
|
||
"fmt"
|
||
"go-common/app/service/bbq/user/api"
|
||
"go-common/app/service/bbq/user/internal/model"
|
||
"go-common/library/database/sql"
|
||
"go-common/library/log"
|
||
"go-common/library/time"
|
||
"go-common/library/xstr"
|
||
)
|
||
|
||
//常量
|
||
const (
|
||
_addUserLike = "insert into %s (`mid`, `opid`) values (?,?) on duplicate key update state=0"
|
||
_cancelUserLike = "update %s set state = 1 where mid=? and opid=?"
|
||
_selectUserLike = "select opid from user_like_%02d where mid = ? and state = 0 and opid in (%s)"
|
||
_spaceUserLike = "select opid, mtime from user_like_%02d where mid=%d and state=0 and mtime %s ? order by mtime %s, opid %s limit %d"
|
||
)
|
||
|
||
//TxAddUserLike .
|
||
func (d *Dao) TxAddUserLike(tx *sql.Tx, mid, svid int64) (num int64, err error) {
|
||
var res xsql.Result
|
||
if res, err = tx.Exec(d.userLikeSQL(mid, _addUserLike), mid, svid); err != nil {
|
||
return
|
||
}
|
||
return res.RowsAffected()
|
||
}
|
||
|
||
//TxCancelUserLike .
|
||
func (d *Dao) TxCancelUserLike(tx *sql.Tx, mid, svid int64) (num int64, err error) {
|
||
var res xsql.Result
|
||
if res, err = tx.Exec(d.userLikeSQL(mid, _cancelUserLike), mid, svid); err != nil {
|
||
return
|
||
}
|
||
return res.RowsAffected()
|
||
}
|
||
|
||
// CheckUserLike 检测用户是否点赞
|
||
func (d *Dao) CheckUserLike(c context.Context, mid int64, svids []int64) (res []int64, err error) {
|
||
log.V(1).Info("user like mid(%d) svids(%v)", mid, svids)
|
||
ls := len(svids)
|
||
if ls == 0 || mid == 0 {
|
||
return
|
||
}
|
||
idStr := xstr.JoinInts(svids)
|
||
querySQL := fmt.Sprintf(_selectUserLike, d.getTableIndex(mid), idStr)
|
||
rows, err := d.db.Query(c, querySQL, mid)
|
||
log.V(1).Infov(c, log.KV("log", fmt.Sprintf("user like rows(%v) err(%v)", rows, err)))
|
||
if err != nil {
|
||
if err == sql.ErrNoRows {
|
||
err = nil
|
||
}
|
||
return
|
||
}
|
||
for rows.Next() {
|
||
opid := int64(0)
|
||
rows.Scan(&opid)
|
||
res = append(res, opid)
|
||
}
|
||
log.V(1).Infov(c, log.KV("log", fmt.Sprintf("user like res(%v)", res)))
|
||
return
|
||
}
|
||
|
||
// GetUserLikeList 返回用户点赞列表
|
||
// 当前cursorID表示opid
|
||
func (d *Dao) GetUserLikeList(c context.Context, mid int64, cursorNext bool, cursor model.CursorValue, size int) (
|
||
likeSvs []*api.LikeSv, err error) {
|
||
|
||
compareSymbol := string(">=")
|
||
orderDirection := "asc"
|
||
if cursorNext {
|
||
compareSymbol = "<="
|
||
orderDirection = "desc"
|
||
}
|
||
querySQL := fmt.Sprintf(_spaceUserLike, d.getTableIndex(mid), mid, compareSymbol, orderDirection, orderDirection, size)
|
||
log.V(1).Infov(c, log.KV("like_list_sql", querySQL))
|
||
rows, err := d.db.Query(c, querySQL, cursor.CursorTime)
|
||
if err != nil {
|
||
log.Errorv(c, log.KV("event", "mysql_select"), log.KV("table", "user_like"),
|
||
log.KV("mid", mid), log.KV("sql", querySQL))
|
||
return
|
||
}
|
||
defer rows.Close()
|
||
var svID int64
|
||
var curMtime time.Time
|
||
conflict := bool(true)
|
||
for rows.Next() {
|
||
if err = rows.Scan(&svID, &curMtime); err != nil {
|
||
log.Errorv(c, log.KV("event", "mysql_scan"), log.KV("table", "user_like"),
|
||
log.KV("sql", querySQL))
|
||
return
|
||
}
|
||
// 为了解决同一个mtime的冲突问题
|
||
if curMtime == cursor.CursorTime && conflict {
|
||
if svID == cursor.CursorID {
|
||
conflict = false
|
||
}
|
||
continue
|
||
}
|
||
cursorValue := model.CursorValue{CursorID: svID, CursorTime: curMtime}
|
||
jsonStr, _ := json.Marshal(cursorValue) // marshal的时候相信库函数，不做err判断
|
||
likeSvs = append(likeSvs, &api.LikeSv{Svid: svID, CursorValue: string(jsonStr)})
|
||
}
|
||
log.Infov(c, log.KV("event", "mysql_select"), log.KV("table", "user_like"),
|
||
log.KV("mid", mid), log.KV("id", cursor.CursorID), log.KV("size", len(likeSvs)))
|
||
return
|
||
}
|