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.
77 lines
2.1 KiB
77 lines
2.1 KiB
package income |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
|
|
"go-common/library/database/sql" |
|
"go-common/library/log" |
|
"go-common/library/xstr" |
|
) |
|
|
|
const ( |
|
// select |
|
_blackListByMIDSQL = "SELECT av_id FROM av_black_list WHERE mid = ? AND ctype = ? AND is_delete = 0" |
|
_blackListByAvIDSQL = "SELECT av_id FROM av_black_list WHERE av_id in (%s) AND ctype = ? AND is_delete = 0" |
|
// insert |
|
_inBlackListSQL = "INSERT INTO av_black_list(av_id,mid,ctype,reason,nickname,has_signed,is_delete) VALUES %s ON DUPLICATE KEY UPDATE reason=VALUES(reason),nickname=VALUES(nickname),has_signed=VALUES(has_signed),is_delete=VALUES(is_delete)" |
|
) |
|
|
|
// ListAvBlackList list av_blakc_list by av_id |
|
func (d *Dao) ListAvBlackList(c context.Context, avID []int64, ctype int) (avb map[int64]struct{}, err error) { |
|
avb = make(map[int64]struct{}) |
|
rows, err := d.db.Query(c, fmt.Sprintf(_blackListByAvIDSQL, xstr.JoinInts(avID)), ctype) |
|
if err != nil { |
|
log.Error("ListAvBlackList d.db.Query error(%v)", err) |
|
return |
|
} |
|
defer rows.Close() |
|
|
|
for rows.Next() { |
|
var id int64 |
|
err = rows.Scan(&id) |
|
if err != nil { |
|
log.Error("ListAvBlackList rows scan error(%v)", err) |
|
return |
|
} |
|
avb[id] = struct{}{} |
|
} |
|
err = rows.Err() |
|
return |
|
} |
|
|
|
// GetAvBlackListByMID list av_blakc_list by av_id |
|
func (d *Dao) GetAvBlackListByMID(c context.Context, mid int64, typ int) (avb map[int64]struct{}, err error) { |
|
avb = make(map[int64]struct{}) |
|
rows, err := d.db.Query(c, _blackListByMIDSQL, mid, typ) |
|
if err != nil { |
|
log.Error("GetAvBlackListByMID d.db.Query error(%v)", err) |
|
return |
|
} |
|
defer rows.Close() |
|
|
|
for rows.Next() { |
|
var avID int64 |
|
err = rows.Scan(&avID) |
|
if err != nil { |
|
log.Error("GetAvBlackListByMID rows scan error(%v)", err) |
|
return |
|
} |
|
avb[avID] = struct{}{} |
|
} |
|
err = rows.Err() |
|
return |
|
} |
|
|
|
// TxInsertAvBlackList insert val into av_black_list |
|
func (d *Dao) TxInsertAvBlackList(tx *sql.Tx, val string) (rows int64, err error) { |
|
if val == "" { |
|
return |
|
} |
|
res, err := tx.Exec(fmt.Sprintf(_inBlackListSQL, val)) |
|
if err != nil { |
|
log.Error("TxInsertAvBlackList tx.Exec error(%v)", err) |
|
return |
|
} |
|
return res.RowsAffected() |
|
}
|
|
|