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.
111 lines
3.5 KiB
111 lines
3.5 KiB
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 |
|
}
|
|
|