Initial commit: web_hghl_server

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

pub/models/questionnairemod/questionnaire.go

@@ -0,0 +1,88 @@
+package questionnairemod
+
+import (
+	"context"
+	"fmt"
+
+	"videoweb/pub/common/log"
+	"videoweb/pub/common/srvdef"
+	"videoweb/pub/mongos"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+var db *mongos.Collection
+
+func Init(ctx context.Context) {
+	db = mongos.Shop.Coll(TN.Questionnaire)
+	indexes := []mongo.IndexModel{
+		{
+			Keys: bson.D{
+				{Key: FN.Questionnaire.Status, Value: 1},
+				{Key: FN.Questionnaire.CreatedAt, Value: -1},
+			},
+		},
+		{
+			Keys: bson.D{
+				{Key: FN.Questionnaire.CreatedAt, Value: -1},
+			},
+		},
+	}
+	if _, err := db.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("(questionnaire:Init) Create Index failed! err:%+v", err))
+	}
+}
+
+// InsertOne 插入一条问卷
+func InsertOne(ctx context.Context, doc *QuestionnaireModel) *srvdef.SrvErr {
+	if _, err := db.InsertOne(ctx, doc); err != nil {
+		log.Warn("questionnairemod:InsertOne failed", log.E(err))
+		return srvdef.New(srvdef.ErrDbInsertError, "", "")
+	}
+	return nil
+}
+
+// FindOne 查找一条问卷
+func FindOne(ctx context.Context, filter bson.M, opts ...*options.FindOneOptions) (*QuestionnaireModel, *srvdef.SrvErr) {
+	var doc QuestionnaireModel
+	if err := db.FindOneEx(ctx, &doc, filter, opts...); err != nil {
+		if err == mongo.ErrNoDocuments {
+			return nil, srvdef.New(srvdef.ErrNoData, "", "")
+		}
+		log.Warn("questionnairemod:FindOne failed", log.E(err), log.Any("filter", filter))
+		return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return &doc, nil
+}
+
+// FindMany 查找多条问卷
+func FindMany(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (QuestionnaireModelList, *srvdef.SrvErr) {
+	data := make(QuestionnaireModelList, 0)
+	if err := db.Find(ctx, &data, filter, opts...); err != nil {
+		log.Warn("questionnairemod:FindMany failed", log.E(err), log.Any("filter", filter))
+		return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return data, nil
+}
+
+// Count 统计问卷数量
+func Count(ctx context.Context, filter bson.M) (int64, *srvdef.SrvErr) {
+	count, err := db.Count(ctx, filter)
+	if err != nil {
+		log.Warn("questionnairemod:Count failed", log.E(err), log.Any("filter", filter))
+		return 0, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return count, nil
+}
+
+// UpdateOne 更新一条问卷，返回 MatchedCount
+func UpdateOne(ctx context.Context, filter bson.M, update bson.M) (int64, *srvdef.SrvErr) {
+	result, err := db.UpdateOne(ctx, filter, update)
+	if err != nil {
+		log.Warn("questionnairemod:UpdateOne failed", log.E(err), log.Any("filter", filter))
+		return 0, srvdef.New(srvdef.ErrDbUpdateError, "", "")
+	}
+	return result.MatchedCount, nil
+}