初始化项目

dao/stat_report_dao.go

@@ -5,8 +5,9 @@ import (
 
 	"cid/model/entity"
 
-	"github.com/gogf/gf/v2/database/gdb"
-	"github.com/gogf/gf/v2/frame/g"
+	"gitee.com/red-future---jilin-g/common/mongo"
+	"go.mongodb.org/mongo-driver/v2/bson"
+	"go.mongodb.org/mongo-driver/v2/mongo/options"
 )
 
 // statReportDao 统计报表DAO
@@ -14,77 +15,74 @@ type statReportDao struct{}
 
 var StatReport = &statReportDao{}
 
-// Ctx 获取数据库上下文
-func (d *statReportDao) Ctx(ctx context.Context) *gdb.Model {
-	return g.DB().Model("stat_report").Ctx(ctx)
-}
-
 // Create 创建统计报表
-func (d *statReportDao) Create(ctx context.Context, report *entity.StatReport) (int64, error) {
-	result, err := d.Ctx(ctx).Insert(report)
-	if err != nil {
-		return 0, err
-	}
-	id, _ := result.LastInsertId()
-	return id, nil
+func (d *statReportDao) Create(ctx context.Context, report *entity.StatReport) (err error) {
+	_, err = mongo.Insert(ctx, []interface{}{report}, "stat_report")
+	return
 }
 
 // GetByID 根据ID获取统计报表
-func (d *statReportDao) GetByID(ctx context.Context, id int64) (*entity.StatReport, error) {
-	var report *entity.StatReport
-	err := d.Ctx(ctx).Where("id", id).Scan(&report)
-	return report, err
+func (d *statReportDao) GetByID(ctx context.Context, id string) (report *entity.StatReport, err error) {
+	filter := bson.M{"_id": id}
+	report = &entity.StatReport{}
+	err = mongo.FindOne(ctx, filter, report, "stat_report")
+	return
 }
 
 // GetByTenantAndDate 根据租户和日期获取统计报表
-func (d *statReportDao) GetByTenantAndDate(ctx context.Context, tenantID int64, reportType, date string) (*entity.StatReport, error) {
-	var report *entity.StatReport
-	err := d.Ctx(ctx).Where("tenant_id", tenantID).Where("report_type", reportType).Where("report_date", date).Scan(&report)
-	return report, err
+func (d *statReportDao) GetByTenantAndDate(ctx context.Context, tenantID, reportType, date string) (report *entity.StatReport, err error) {
+	filter := bson.M{"tenantId": tenantID, "reportType": reportType, "reportDate": date}
+	report = &entity.StatReport{}
+	err = mongo.FindOne(ctx, filter, report, "stat_report")
+	return
 }
 
 // Update 更新统计报表
-func (d *statReportDao) Update(ctx context.Context, report *entity.StatReport) error {
-	_, err := d.Ctx(ctx).Where("id", report.Id).Update(report)
-	return err
+func (d *statReportDao) Update(ctx context.Context, report *entity.StatReport) (err error) {
+	filter := bson.M{"_id": report.Id}
+	update := bson.M{"$set": report}
+	_, err = mongo.Update(ctx, filter, update, "stat_report")
+	return
 }
 
 // Delete 删除统计报表
-func (d *statReportDao) Delete(ctx context.Context, id int64) error {
-	_, err := d.Ctx(ctx).Where("id", id).Delete()
-	return err
+func (d *statReportDao) Delete(ctx context.Context, id string) (err error) {
+	filter := bson.M{"_id": id}
+	_, err = mongo.Delete(ctx, filter, "stat_report")
+	return
 }
 
 // List 统计报表列表
-func (d *statReportDao) List(ctx context.Context, tenantID, appID int64, reportType, startDate, endDate string, page, pageSize int) ([]*entity.StatReport, int, error) {
-	model := d.Ctx(ctx)
+func (d *statReportDao) List(ctx context.Context, tenantID, appID, reportType, startDate, endDate string, page, pageSize int) (reports []*entity.StatReport, total int, err error) {
+	filter := bson.M{}
 
-	if tenantID > 0 {
-		model = model.Where("tenant_id", tenantID)
+	if tenantID != "" {
+		filter["tenantId"] = tenantID
 	}
-	if appID > 0 {
-		model = model.Where("app_id", appID)
+	if appID != "" {
+		filter["appId"] = appID
 	}
 	if reportType != "" {
-		model = model.Where("report_type", reportType)
+		filter["reportType"] = reportType
 	}
-	if startDate != "" {
-		model = model.WhereGTE("report_date", startDate)
-	}
-	if endDate != "" {
-		model = model.WhereLTE("report_date", endDate)
+	if startDate != "" && endDate != "" {
+		filter["reportDate"] = bson.M{"$gte": startDate, "$lte": endDate}
 	}
 
-	var reports []*entity.StatReport
-	err := model.Page(page, pageSize).OrderDesc("generated_at").Scan(&reports)
+	// 获取总数
+	total64, err := mongo.Count(ctx, filter, "stat_report")
 	if err != nil {
 		return nil, 0, err
 	}
+	total = int(total64)
 
-	total, err := model.Count()
-	if err != nil {
-		return nil, 0, err
-	}
+	// 分页参数处理
+	limit := int64(pageSize)
+	skip := int64((page - 1) * pageSize)
 
-	return reports, total, nil
+	// 排序处理
+	opts := options.Find().SetLimit(limit).SetSkip(skip).SetSort(bson.M{"generatedAt": -1})
+
+	err = mongo.Find(ctx, filter, &reports, "stat_report", opts)
+	return
 }