first commit

2025-08-16 18:46:29 +08:00
commit 5f008b3228
511 changed files with 53667 additions and 0 deletions
--- a/app/main/model/agentClosureModel.go
+++ b/app/main/model/agentClosureModel.go
@@ -0,0 +1,99 @@
+package model
+
+import (
+	"context"
+	"fmt"
+	"znc-server/common/globalkey"
+
+	"github.com/pkg/errors"
+	"github.com/zeromicro/go-zero/core/stores/cache"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
+
+var _ AgentClosureModel = (*customAgentClosureModel)(nil)
+
+type (
+	// AgentClosureModel is an interface to be customized, add more methods here,
+	// and implement the added methods in customAgentClosureModel.
+	AgentClosureModel interface {
+		agentClosureModel
+		FindUnionPageListByPageWithTotal(ctx context.Context, agentId, subordinateId int64, page, pageSize int64) ([]*UnionDetail, int64, error)
+	}
+
+	customAgentClosureModel struct {
+		*defaultAgentClosureModel
+	}
+
+	// UnionDetail 合并后的详情结构
+	UnionDetail struct {
+		Id         int64   `db:"id"`
+		CreateTime string  `db:"create_time"`
+		Amount     float64 `db:"amount"`
+		Type       string  `db:"type"`
+	}
+)
+
+// NewAgentClosureModel returns a model for the database table.
+func NewAgentClosureModel(conn sqlx.SqlConn, c cache.CacheConf) AgentClosureModel {
+	return &customAgentClosureModel{
+		defaultAgentClosureModel: newAgentClosureModel(conn, c),
+	}
+}
+
+// FindUnionPageListByPageWithTotal 获取合并后的分页列表
+func (m *customAgentClosureModel) FindUnionPageListByPageWithTotal(ctx context.Context, agentId, subordinateId int64, page, pageSize int64) ([]*UnionDetail, int64, error) {
+	// 构建UNION ALL查询
+	deductionQuery := fmt.Sprintf(`
+		SELECT id, create_time, amount, type 
+		FROM agent_commission_deduction 
+		WHERE agent_id = ? AND deducted_agent_id = ? AND del_state = ?
+	`)
+
+	rewardsQuery := fmt.Sprintf(`
+		SELECT id, create_time, amount, type 
+		FROM agent_rewards 
+		WHERE agent_id = ? AND relation_agent_id = ? AND del_state = ?
+	`)
+
+	// 计算总记录数
+	countQuery := fmt.Sprintf(`
+		SELECT COUNT(*) FROM (
+			%s
+			UNION ALL
+			%s
+		) AS union_table
+	`, deductionQuery, rewardsQuery)
+
+	var total int64
+	err := m.QueryRowNoCacheCtx(ctx, &total, countQuery, agentId, subordinateId, globalkey.DelStateNo, agentId, subordinateId, globalkey.DelStateNo)
+	if err != nil {
+		return nil, 0, errors.Wrapf(err, "查询总记录数失败")
+	}
+
+	// 构建分页查询
+	if page < 1 {
+		page = 1
+	}
+	offset := (page - 1) * pageSize
+
+	unionQuery := fmt.Sprintf(`
+		SELECT * FROM (
+			%s
+			UNION ALL
+			%s
+		) AS union_table
+		ORDER BY create_time DESC
+		LIMIT ? OFFSET ?
+	`, deductionQuery, rewardsQuery)
+
+	var resp []*UnionDetail
+	err = m.QueryRowsNoCacheCtx(ctx, &resp, unionQuery,
+		agentId, subordinateId, globalkey.DelStateNo,
+		agentId, subordinateId, globalkey.DelStateNo,
+		pageSize, offset)
+	if err != nil {
+		return nil, 0, errors.Wrapf(err, "查询分页数据失败")
+	}
+
+	return resp, total, nil
+}