feat: 迁移渠道激活列表和绑定期间流水

2026-11-03 20:53:25 +08:00
parent 838c2c496f
commit 8908430a89
28 changed files with 1180 additions and 9 deletions
@@ -0,0 +1,160 @@
+package business
+
+import (
+	"context"
+	"errors"
+	"strings"
+	"time"
+
+	"gorm.io/gorm"
+	"kra/internal/biz/business"
+)
+
+// OA 表只读，不由新商务的系统迁移创建或改写。
+type activationPO struct {
+	ID                                                               uint64 `gorm:"primaryKey"`
+	PromoteID                                                        *uint64
+	PlatformName, PromoteAccount, BusierAccount, ApplicantOA, Status string
+	AccumAmount                                                      *string
+	BindAt, UnbindAt, ReachAt, DeletedAt                             *time.Time
+}
+
+func (activationPO) TableName() string { return "sys_promote_active_pool_bind" }
+func toBizActivation(po activationPO) *business.Activation {
+	amount := "0.00"
+	if po.AccumAmount != nil {
+		amount = *po.AccumAmount
+	}
+	return &business.Activation{ID: po.ID, PromoteID: po.PromoteID, PlatformName: po.PlatformName, PromoteAccount: po.PromoteAccount, BusierAccount: po.BusierAccount, ApplicantOA: po.ApplicantOA, Status: po.Status, AccumAmount: amount, BindAt: po.BindAt, UnbindAt: po.UnbindAt, ReachAt: po.ReachAt}
+}
+
+type activationRepo struct {
+	data      Provider
+	databases PlatformProvider
+}
+
+func NewActivationRepo(data Provider, databases PlatformProvider) business.ActivationRepo {
+	return &activationRepo{data: data, databases: databases}
+}
+
+func (r *activationRepo) OwnAccounts(ctx context.Context, id uint) ([]string, error) {
+	accounts := []string{}
+	err := r.data.DB().WithContext(ctx).Table("sys_user_bind_platform").Where("user_id = ? AND status = ? AND platform_busier_account IS NOT NULL", id, "0").Distinct("platform_busier_account").Pluck("platform_busier_account", &accounts).Error
+	return accounts, storageError(err)
+}
+
+func (r *activationRepo) oa(ctx context.Context) (*gorm.DB, error) {
+	db, err := r.databases.Database("oa")
+	if err != nil || db == nil {
+		return nil, business.ErrActivationSource
+	}
+	db = db.WithContext(ctx)
+	available, err := businessTableAvailable(db, "sys_promote_active_pool_bind")
+	if err != nil || !available {
+		return nil, business.ErrActivationSource
+	}
+	return db, nil
+}
+
+func (r *activationRepo) List(ctx context.Context, scope business.ActivationScope, filter business.ActivationFilter, options ...business.ListOption) (*business.ActivationPage, error) {
+	page := &business.ActivationPage{Rows: []*business.Activation{}, Available: true}
+	if !scope.All && len(scope.Accounts) == 0 {
+		return page, nil
+	}
+	db, err := r.oa(ctx)
+	if err != nil {
+		page.Available = false
+		return page, nil
+	}
+	q := db.Model(&activationPO{}).Where("deleted_at IS NULL")
+	if !scope.All {
+		accounts := make([]string, 0, len(scope.Accounts))
+		for _, account := range scope.Accounts {
+			accounts = append(accounts, strings.ToLower(account))
+		}
+		q = q.Where("LOWER(busier_account) IN ?", accounts)
+	}
+	if len(filter.PlatformNames) > 0 {
+		q = q.Where("platform_name IN ?", filter.PlatformNames)
+	}
+	if len(filter.Statuses) > 0 {
+		q = q.Where("status IN ?", filter.Statuses)
+	}
+	if filter.PromoteAccount != "" {
+		q = q.Where("promote_account LIKE ?", "%"+filter.PromoteAccount+"%")
+	}
+	if filter.BusierAccount != "" {
+		q = q.Where("busier_account LIKE ?", "%"+filter.BusierAccount+"%")
+	}
+	if err := q.Count(&page.Total).Error; err != nil {
+		return nil, business.ErrActivationSource
+	}
+	o := business.NewListOptions(options...)
+	var pos []activationPO
+	if err := q.Order("id DESC").Offset(o.Offset).Limit(o.Limit).Find(&pos).Error; err != nil {
+		return nil, business.ErrActivationSource
+	}
+	for _, po := range pos {
+		page.Rows = append(page.Rows, toBizActivation(po))
+	}
+	return page, nil
+}
+
+func (r *activationRepo) Find(ctx context.Context, id uint64) (*business.Activation, error) {
+	db, err := r.oa(ctx)
+	if err != nil {
+		return nil, err
+	}
+	var po activationPO
+	if err := db.Where("id = ? AND deleted_at IS NULL", id).Take(&po).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, business.ErrActivationNotFound
+		}
+		return nil, business.ErrActivationSource
+	}
+	return toBizActivation(po), nil
+}
+
+func (r *activationRepo) Spend(ctx context.Context, query business.ActivationSpendQuery) ([]*business.ActivationSpendItem, error) {
+	db := r.data.DB().WithContext(ctx)
+	available, err := businessTableAvailable(db, "tab_promote")
+	if err != nil {
+		return nil, err
+	}
+	if !available {
+		return nil, business.ErrActivationMetrics
+	}
+	tree := func() *gorm.DB {
+		return db.Table("tab_promote p").Where("p.source_db = ? AND (p.id = ? OR p.top_promote_id = ? OR p.parent_id = ?)", query.PlatformName, query.PromoteID, query.PromoteID, query.PromoteID)
+	}
+	var count int64
+	if err := tree().Count(&count).Error; err != nil {
+		return nil, business.ErrStorage
+	}
+	items := []*business.ActivationSpendItem{}
+	if count == 0 {
+		return items, nil
+	}
+	available, err = businessTableAvailable(db, "tab_spend")
+	if err != nil {
+		return nil, err
+	}
+	if !available {
+		return nil, business.ErrActivationMetrics
+	}
+	q := db.Table("tab_spend s").Where("s.source_db = ? AND s.pay_status = ? AND s.pay_time >= ?", query.PlatformName, 1, query.Start).Where("EXISTS (?)", tree().Select("1").Where("p.id=s.promote_id"))
+	if query.End != nil {
+		q = q.Where("s.pay_time <= ?", *query.End)
+	}
+	var rows []struct {
+		GameID           uint64
+		GameName, Amount string
+	}
+	if err := q.Select("s.game_id,COALESCE(MAX(s.game_name),'') AS game_name,COALESCE(SUM(s.pay_amount),0) AS amount").Group("s.game_id").Order("amount DESC,s.game_id").Scan(&rows).Error; err != nil {
+		return nil, business.ErrStorage
+	}
+	for _, row := range rows {
+		items = append(items, &business.ActivationSpendItem{GameName: row.GameName, Amount: row.Amount})
+	}
+	return items, nil
+}