153 lines
3.7 KiB
Go
153 lines
3.7 KiB
Go
package adminMod
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"stat/common/log"
|
|
"stat/common/mongos"
|
|
"stat/common/utils"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
func Init() {
|
|
// 获取数据库表
|
|
db = mongos.New(mongos.MgoManager, CollAdmin)
|
|
|
|
// 初始化索引
|
|
indexes := []mongo.IndexModel{
|
|
{
|
|
Keys: bson.D{{Key: "account", Value: 1}},
|
|
Options: options.Index().SetUnique(true),
|
|
},
|
|
}
|
|
|
|
if _, err := db.CreateIndex(context.Background(), indexes); err != nil {
|
|
panic(fmt.Sprintf("Model %s InitIndex error %+v", CollAdmin, err))
|
|
}
|
|
}
|
|
|
|
// 根据账号获取管理员信息
|
|
func FindOneByAccount(ctx context.Context, account string) (data Admin, err error) {
|
|
filter := bson.M{"account": account}
|
|
|
|
if err = db.FindOne(ctx, &data, filter); err != nil {
|
|
log.Error("admin FindOneByAccount err", log.Any("account", account), log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// FindOnedmin 查找一个渠道信息
|
|
func FindOnedmin(ctx context.Context, filter bson.M) (*Admin, error) {
|
|
var Admin Admin
|
|
if err := db.FindOne(ctx, &Admin, filter); err != nil {
|
|
log.Warn("admin:FindOnedmin.FindOne failed",
|
|
log.E(err), log.Any("filter", filter))
|
|
return nil, err
|
|
}
|
|
return &Admin, nil
|
|
}
|
|
|
|
// 获取管理员列表
|
|
func FindAdminList(ctx context.Context, blocRole *int, skip int64, limit int64) (count int64, data []Admin, err error) {
|
|
filter := bson.M{}
|
|
if blocRole != nil {
|
|
filter["blocRole"] = blocRole
|
|
}
|
|
opts := options.Find().SetSkip(skip).SetLimit(limit)
|
|
|
|
if err = db.Find(ctx, &data, filter, opts); err != nil {
|
|
log.Error("admin FindAdminList Find err", log.E(err))
|
|
return
|
|
}
|
|
if count, err = db.Count(ctx, filter); err != nil {
|
|
log.Error("admin FindAdminList Count err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// 新增管理员
|
|
func AddAdmin(ctx context.Context, doc *Admin) (err error) {
|
|
if _, err = db.InsertOne(ctx, doc); err != nil {
|
|
log.Error("admin AddAdmin InsertOne err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// 更新管理员
|
|
func UpdateAdmin(ctx context.Context, account string, doc UpdateDoc) (err error) {
|
|
filter := bson.M{"account": account}
|
|
update, _ := utils.ToBsonM(doc)
|
|
|
|
if _, err = db.UpdateOne(ctx, filter, bson.M{"$set": update}); err != nil {
|
|
log.Error("admin UpdateAdmin UpdateOne err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// 删除管理员
|
|
func DelAdmin(ctx context.Context, account string) (err error) {
|
|
filter := bson.M{"account": account}
|
|
|
|
if _, err = db.DeleteOne(ctx, filter); err != nil {
|
|
log.Error("admin DelAdmin DeleteOne err", log.E(err))
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
// 验证token
|
|
func CheckToken(ctx context.Context, token string) (ok bool, err error) {
|
|
booToken := Admin{}
|
|
|
|
if err = db.FindOne(ctx, &booToken, bson.M{"token": token}); err != nil {
|
|
log.Error("admin CheckToken FindOne err", log.E(err))
|
|
return false, err
|
|
}
|
|
|
|
if booToken.Account == "" {
|
|
return false, nil
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
// 刷新token
|
|
func RefreshToken(ctx context.Context, account string, newToken string) (err error) {
|
|
filter := bson.M{"account": account}
|
|
update := bson.M{"token": newToken}
|
|
|
|
if _, err = db.UpdateOne(ctx, filter, bson.M{"$set": update}); err != nil {
|
|
log.Error("admin RefreshToken UpdateOne err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// 绑定两步验证
|
|
func TwoStepBind(ctx context.Context, account string) (err error) {
|
|
filter := bson.M{"account": account}
|
|
update := bson.M{"$set": bson.M{"hasBind": true}}
|
|
_, err = db.UpdateOne(ctx, filter, update)
|
|
return
|
|
}
|
|
|
|
// 设置两步验证密钥
|
|
func TwoStepSecretSet(ctx context.Context, account string, secret string) (err error) {
|
|
filter := bson.M{"account": account}
|
|
update := bson.M{"$set": bson.M{"secret": secret}}
|
|
_, err = db.UpdateOne(ctx, filter, update)
|
|
return
|
|
}
|