89 lines
2.5 KiB
Go
89 lines
2.5 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cid/model/entity"
|
|
|
|
"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
|
|
type statReportDao struct{}
|
|
|
|
var StatReport = &statReportDao{}
|
|
|
|
// Create 创建统计报表
|
|
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 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, 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) (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 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, reportType, startDate, endDate string, page, pageSize int) (reports []*entity.StatReport, total int, err error) {
|
|
filter := bson.M{}
|
|
|
|
if tenantID != "" {
|
|
filter["tenantId"] = tenantID
|
|
}
|
|
if appID != "" {
|
|
filter["appId"] = appID
|
|
}
|
|
if reportType != "" {
|
|
filter["reportType"] = reportType
|
|
}
|
|
if startDate != "" && endDate != "" {
|
|
filter["reportDate"] = bson.M{"$gte": startDate, "$lte": endDate}
|
|
}
|
|
|
|
// 获取总数
|
|
total64, err := mongo.Count(ctx, filter, "stat_report")
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
total = int(total64)
|
|
|
|
// 分页参数处理
|
|
limit := int64(pageSize)
|
|
skip := int64((page - 1) * pageSize)
|
|
|
|
// 排序处理
|
|
opts := options.Find().SetLimit(limit).SetSkip(skip).SetSort(bson.M{"generatedAt": -1})
|
|
|
|
err = mongo.Find(ctx, filter, &reports, "stat_report", opts)
|
|
return
|
|
}
|