forked from all_user/stat_center
122 lines
2.6 KiB
Go
122 lines
2.6 KiB
Go
package announInfoMod
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"stat/common/log"
|
|
"stat/common/mongos"
|
|
"stat/common/utils"
|
|
"time"
|
|
|
|
"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, CollAnnounInfo)
|
|
|
|
// 初始化索引
|
|
indexes := []mongo.IndexModel{
|
|
{
|
|
Keys: bson.D{{Key: "appId", Value: 1}},
|
|
},
|
|
}
|
|
|
|
if _, err := db.CreateIndex(context.Background(), indexes); err != nil {
|
|
panic(fmt.Sprintf("Model %s InitIndex error %+v", CollAnnounInfo, err))
|
|
}
|
|
}
|
|
|
|
// 获取公告列表
|
|
func FindAnnounceList(ctx context.Context, appId *int32, skip int64, limit int64) (count int64, data []AnnounInfo, err error) {
|
|
filter := bson.M{}
|
|
|
|
if appId != nil {
|
|
filter["appId"] = appId
|
|
}
|
|
|
|
opts := options.Find().SetSkip(skip).SetLimit(limit)
|
|
|
|
if err = db.Find(ctx, &data, filter, opts); err != nil {
|
|
log.Error("announce FindAnnounceList Find err", log.E(err))
|
|
return
|
|
}
|
|
if count, err = db.Count(ctx, filter); err != nil {
|
|
log.Error("announce FindAnnounceList Count err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// 新增管理员
|
|
func AddAnnounce(ctx context.Context, doc *AnnounInfo) (err error) {
|
|
if _, err = db.InsertOne(ctx, doc); err != nil {
|
|
log.Error("announce AddAnnounce InsertOne err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// 更新管理员
|
|
func UpdateAnnounce(ctx context.Context, id string, doc UpdateDoc) (err error) {
|
|
if id == "" {
|
|
return
|
|
}
|
|
|
|
objId, err := primitive.ObjectIDFromHex(id)
|
|
if err != nil {
|
|
log.Error("announce UpdateAnnounce ObjectIDFromHex err", log.E(err))
|
|
return
|
|
}
|
|
|
|
filter := bson.M{"_id": objId}
|
|
update, _ := utils.ToBsonM(doc)
|
|
|
|
if _, err = db.UpdateOne(ctx, filter, bson.M{"$set": update}); err != nil {
|
|
log.Error("announce UpdateAdmin UpdateOne err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// 删除管理员
|
|
func DelAnnounce(ctx context.Context, id string) (err error) {
|
|
if id == "" {
|
|
return
|
|
}
|
|
|
|
objId, err := primitive.ObjectIDFromHex(id)
|
|
if err != nil {
|
|
log.Error("announce DelAnnounce ObjectIDFromHex err", log.E(err))
|
|
return
|
|
}
|
|
|
|
filter := bson.M{"_id": objId}
|
|
|
|
if _, err = db.DeleteOne(ctx, filter); err != nil {
|
|
log.Error("announce DelAnnounce DeleteOne err", log.E(err))
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
// 根据appId获取公告
|
|
func FindAnnounceByAppId(ctx context.Context, appId int32) (data []AnnounInfo, err error) {
|
|
now := time.Now()
|
|
|
|
filer := bson.M{"appId": appId, "startTime": bson.M{"$lt": now}, "endTime": bson.M{"$gt": now}}
|
|
|
|
if err = db.Find(ctx, &data, filer); err != nil {
|
|
log.Error("announce DelAnnounce DeleteOne err", log.E(err))
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|