Initial commit

models/appcategory/appcategory.go

@@ -0,0 +1,141 @@
+package appcategory
+
+import (
+	"context"
+	"fmt"
+	"stat/common/log"
+	"stat/common/mongos"
+	"stat/common/stdLog"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+// 表名
+const Collappcategory = "appcategory"
+
+var db *mongos.MongoTool
+
+func Init(ctx context.Context) {
+	// 获取数据库表
+	db = mongos.New(mongos.MgoManager, Collappcategory)
+	// 获取数据库表
+	indexes := []mongo.IndexModel{
+		{
+			Keys: bson.D{
+				{Key: "parentAppId", Value: 1},
+			},
+		},
+	}
+	if _, err := db.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("(media:initCategoryIndex) Create AppCategory Index failed! err:%+v:", err))
+	}
+}
+
+// InsertOneAppCategory 插入一个应用
+func InsertOneAppCategory(ctx context.Context, Category *AppCategory) *stdLog.SrvErr {
+	if _, err := db.InsertOne(ctx, Category); err != nil {
+		log.Warn("media:InsertOneAppCategory db.InsertOne failed", log.E(err))
+		return stdLog.New(stdLog.InsertDataError, "", "")
+	}
+	return nil
+}
+
+// DelOneAppCategory 删除一个应用tag
+func DelOneAppCategory(ctx context.Context, filter bson.M) (
+	int64, *stdLog.SrvErr) {
+	count, err := db.DeleteOne(ctx, filter)
+	if err != nil {
+		log.Warn("media:DelOneAppCategory db.DeleteOne failed",
+			log.E(err), log.Any("filter", filter))
+		return 0, stdLog.New(stdLog.DelDataError, "", "")
+	}
+	return count, nil
+}
+
+// FindOneAppCategory 查找一个应用
+func FindOneAppCategory(ctx context.Context, filter bson.M) (
+	*AppCategory, *stdLog.SrvErr) {
+	var Category AppCategory
+	if err := db.FindOneEx(
+		ctx, &Category, filter); err != nil {
+		if err != mongo.ErrNoDocuments {
+			log.Info("media:FindOneAppCategory db.FindOne failed",
+				log.E(err), log.Any("filter", filter))
+			return nil, stdLog.New(stdLog.QueryDataError, "", "")
+		}
+		return nil, stdLog.New(stdLog.DataNotExist, "", "")
+	}
+	return &Category, nil
+}
+
+// CountAppCategories 统计应用数量
+func CountAppCategories(ctx context.Context, filter bson.M) (
+	int64, *stdLog.SrvErr) {
+	count, err := db.Count(ctx, filter)
+	if err != nil {
+		log.Warn("media:CountAppCategories db.Count failed",
+			log.E(err), log.Any("filter", filter))
+		return 0, stdLog.New(stdLog.QueryDataError, "", "")
+	}
+	return count, nil
+}
+
+// UpdateOneAppCategory 更新一个应用
+func UpdateOneAppCategory(ctx context.Context, filter bson.M, update bson.M) *stdLog.SrvErr {
+	result, err := db.UpdateOne(ctx, filter, update)
+	if err != nil {
+		log.Warn("media:UpdateOneAppCategory db.UpdateOne failed",
+			log.E(err), log.Any("filter", filter))
+		return stdLog.New(stdLog.UpdateDataError, "", "")
+	}
+	if result.MatchedCount != 1 {
+		log.Warn("media:UpdateOneAppCategory db.UpdateOne failed",
+			log.E(err), log.Any("filter", filter))
+		return stdLog.New(stdLog.DataNotExist, "", "")
+	}
+	return nil
+}
+
+// UpdateManyAppCategory 更新一批应用
+func UpdateManyAppCategory(ctx context.Context, filter bson.M, update bson.M) *stdLog.SrvErr {
+	result, err := db.UpdateMMany(ctx, filter, update)
+	if err != nil {
+		log.Warn("media:UpdateManyAppCategory db.UpdateMMany failed",
+			log.E(err), log.Any("filter", filter))
+		return stdLog.New(stdLog.UpdateDataError, "", "")
+	}
+	if result.MatchedCount <= 0 {
+		log.Warn("media:UpdateManyAppCategory db.UpdateMMany failed",
+			log.E(err), log.Any("filter", filter))
+		//return srvdef.New(srvdef.ErrCategoryNotExsit, "", "")
+	}
+	return nil
+}
+
+// FindManyAppCategory 查找一批应用
+func FindManyAppCategory(ctx context.Context, filter bson.M,
+	opts ...*options.FindOptions) (AppCategoryList, *stdLog.SrvErr) {
+	data := make(AppCategoryList, 0)
+	if err := db.Find(ctx, &data, filter, opts...); err != nil {
+		log.Warn("media:FindManyAppCategory db.Find failed",
+			log.E(err), log.Any("filter", filter))
+		return nil, stdLog.New(stdLog.QueryDataError, "", "")
+	}
+	return data, nil
+}
+
+// GetAppCategoryIdNameMap 查找一批应用
+func GetAppCategoryIdNameMap(ctx context.Context) (map[int64]*AppCategory, *stdLog.SrvErr) {
+	list := make(AppCategoryList, 0)
+	if err := db.Find(ctx, &list, bson.M{}); err != nil {
+		log.Warn("media:GetAppCategoryIdNameMap db.Find failed", log.E(err))
+		return nil, stdLog.New(stdLog.QueryDataError, "", "")
+	}
+	dict := make(map[int64]*AppCategory)
+	for _, item := range list {
+		dict[item.Id] = item
+	}
+	return dict, nil
+}

models/appcategory/type.go

@@ -0,0 +1,21 @@
+package appcategory
+
+import (
+	"time"
+)
+
+// 应用中心
+// @table:appcategory
+type AppCategory struct {
+	Id          int64     `json:"id" bson:"_id,omitempty"`
+	ParentAppId int64     `json:"parentAppId" bson:"parentAppId"` // parentAppId
+	Name        string    `json:"name" bson:"name"`               // 名称
+	Desc        string    `json:"desc" bson:"desc"`               // 描述
+	Status      bool      `json:"status" bson:"status"`           // 状态
+	Rank        int64     `json:"rank" bson:"rank"`               // 应用排序
+	CreatedAt   time.Time `json:"createdAt" bson:"createdAt"`     // 创建时间
+	UpdatedAt   time.Time `json:"updatedAt" bson:"updatedAt"`     // 更新时间
+	ImgType     int64     `json:"imgType" bson:"imgType"`         // 图片展示类型 0:应用图标  1.小图标 2.大图标
+}
+
+type AppCategoryList []*AppCategory