Initial commit

2026-05-20 16:13:58 +08:00
commit ceffbe382e
500 changed files with 157469 additions and 0 deletions
@@ -0,0 +1,121 @@
+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
+}