43 lines
920 B
Go
43 lines
920 B
Go
|
package dao
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"go-common/library/xstr"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
_addAuthoritySQL = "INSERT INTO up_white_list(mid) VALUES %s ON DUPLICATE KEY UPDATE is_deleted=0"
|
||
|
|
_rmAuthoritySQL = "UPDATE up_white_list SET is_deleted = 1 WHERE mid IN (%s)"
|
||
|
|
)
|
||
|
|
|
||
|
|
// AddAuthority .
|
||
|
|
func (d *Dao) AddAuthority(c context.Context, mids []int64) (int64, error) {
|
||
|
|
if len(mids) == 0 {
|
||
|
|
return 0, nil
|
||
|
|
}
|
||
|
|
s := make([]string, 0)
|
||
|
|
for _, v := range mids {
|
||
|
|
s = append(s, fmt.Sprintf("(%d)", v))
|
||
|
|
}
|
||
|
|
res, err := d.db.Exec(c, fmt.Sprintf(_addAuthoritySQL, strings.Join(s, ",")))
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
return res.RowsAffected()
|
||
|
|
}
|
||
|
|
|
||
|
|
// RmAuthority .
|
||
|
|
func (d *Dao) RmAuthority(c context.Context, mids []int64) (int64, error) {
|
||
|
|
if len(mids) == 0 {
|
||
|
|
return 0, nil
|
||
|
|
}
|
||
|
|
res, err := d.db.Exec(c, fmt.Sprintf(_rmAuthoritySQL, xstr.JoinInts(mids)))
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
return res.RowsAffected()
|
||
|
|
}
|