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.
107 lines
3.2 KiB
107 lines
3.2 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"database/sql" |
|
"fmt" |
|
|
|
"go-common/app/admin/main/growup/model" |
|
"go-common/library/log" |
|
xtime "go-common/library/time" |
|
) |
|
|
|
const ( |
|
// insert |
|
_inBlockedSQL = "INSERT INTO up_blocked (mid,nickname,original_archive_count,category_id,fans,apply_at,is_deleted) VALUES (?,?,?,?,?,?,0) ON DUPLICATE KEY UPDATE mid=?,nickname=?,original_archive_count=?,category_id=?,fans=?,apply_at=?,is_deleted=0" |
|
// select |
|
_upInfoVideoSQL = "SELECT apply_at FROM up_info_video WHERE mid=?" |
|
_blockedCountSQL = "SELECT count(*) FROM up_blocked WHERE %s" |
|
_blockedSQL = "SELECT mid,nickname,original_archive_count,category_id,fans,apply_at FROM up_blocked WHERE %s" |
|
_isBlockedSQL = "SELECT mid FROM up_blocked WHERE mid=? AND is_deleted=0" |
|
// update |
|
_upBlockedStateSQL = "UPDATE up_blocked SET is_deleted=? WHERE mid=?" |
|
) |
|
|
|
// ApplyAt query apply_at from up_info_video by mid |
|
func (d *Dao) ApplyAt(c context.Context, mid int64) (applyAt xtime.Time, err error) { |
|
row := d.rddb.QueryRow(c, _upInfoVideoSQL, mid) |
|
if err = row.Scan(&applyAt); err != nil { |
|
if err == sql.ErrNoRows { |
|
err = nil |
|
} else { |
|
log.Error("row.Scan error(%v)", err) |
|
} |
|
} |
|
return |
|
} |
|
|
|
// InsertBlocked insert block up to blacklist |
|
func (d *Dao) InsertBlocked(c context.Context, v *model.Blocked) (rows int64, err error) { |
|
res, err := d.rddb.Exec(c, _inBlockedSQL, v.MID, v.Nickname, v.OriginalArchiveCount, v.MainCategory, v.Fans, v.ApplyAt, v.MID, v.Nickname, v.OriginalArchiveCount, v.MainCategory, v.Fans, v.ApplyAt) |
|
if err != nil { |
|
log.Error("db.inBlockedStmt.Exec(%s) error(%v)", _inBlockedSQL, err) |
|
return |
|
} |
|
return res.RowsAffected() |
|
} |
|
|
|
// UpdateBlockedState update blocked is_deleted |
|
func (d *Dao) UpdateBlockedState(c context.Context, mid int64, del int) (rows int64, err error) { |
|
res, err := d.rddb.Exec(c, _upBlockedStateSQL, del, mid) |
|
if err != nil { |
|
log.Error("db.upBlockedState.Exec(%s) error(%v)", _upBlockedStateSQL, err) |
|
return |
|
} |
|
return res.RowsAffected() |
|
} |
|
|
|
// DelFromBlocked del blocked |
|
func (d *Dao) DelFromBlocked(c context.Context, mid int64) (rows int64, err error) { |
|
return d.UpdateBlockedState(c, mid, 1) |
|
} |
|
|
|
// BlockCount get blocked count |
|
func (d *Dao) BlockCount(c context.Context, query string) (count int, err error) { |
|
row := d.rddb.QueryRow(c, fmt.Sprintf(_blockedCountSQL, query)) |
|
if err = row.Scan(&count); err != nil { |
|
if err == sql.ErrNoRows { |
|
err = nil |
|
} else { |
|
log.Error("rows.Scan error(%v)", err) |
|
} |
|
} |
|
return |
|
} |
|
|
|
// QueryFromBlocked query blocked user in black list |
|
func (d *Dao) QueryFromBlocked(c context.Context, query string) (ups []*model.Blocked, err error) { |
|
rows, err := d.rddb.Query(c, fmt.Sprintf(_blockedSQL, query)) |
|
if err != nil { |
|
return |
|
} |
|
|
|
defer rows.Close() |
|
for rows.Next() { |
|
v := &model.Blocked{} |
|
err = rows.Scan(&v.MID, &v.Nickname, &v.OriginalArchiveCount, &v.MainCategory, &v.Fans, &v.ApplyAt) |
|
if err != nil { |
|
log.Error("rows.Scan error(%v)", err) |
|
return |
|
} |
|
ups = append(ups, v) |
|
} |
|
return |
|
} |
|
|
|
// Blocked check user is blocked |
|
func (d *Dao) Blocked(c context.Context, mid int64) (id int64, err error) { |
|
row := d.rddb.QueryRow(c, _isBlockedSQL, mid) |
|
if err = row.Scan(&id); err != nil { |
|
if err == sql.ErrNoRows { |
|
err = nil |
|
} else { |
|
log.Error("row Scan error(%v)", err) |
|
} |
|
} |
|
return |
|
}
|
|
|