自动生成账单，自动支付账单

2025-12-15 07:06:35 +08:00 · 2021-11-11 08:30:53 +08:00
parent ed74adebac
commit 3e76bf6e5a
12 changed files with 372 additions and 101 deletions
--- a/internal/db/models/user_bill_dao.go
+++ b/internal/db/models/user_bill_dao.go
@@ -2,12 +2,45 @@ package models

 import (
 	"encoding/json"
+	"github.com/TeaOSLab/EdgeAPI/internal/remotelogs"
 	"github.com/TeaOSLab/EdgeAPI/internal/utils/numberutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/lists"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/rands"
+	"github.com/iwind/TeaGo/types"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"time"
 )

+func init() {
+	dbs.OnReadyDone(func() {
+		var generatedMonths = []string{}
+
+		go func() {
+			// 自动生成账单任务
+			var ticker = time.NewTicker(1 * time.Minute)
+			for range ticker.C {
+				// 是否已经生成了，如果已经生成了就跳过
+				var lastMonth = timeutil.Format("Ym", time.Now().AddDate(0, -1, 0))
+				if lists.ContainsString(generatedMonths, lastMonth) {
+					continue
+				}
+
+				err := SharedUserBillDAO.GenerateBills(nil, lastMonth)
+				if err != nil {
+					remotelogs.Error("UserBillDAO", "generate bills failed: "+err.Error())
+				} else {
+					generatedMonths = append(generatedMonths, lastMonth)
+				}
+			}
+		}()
+	})
+}
+
 type BillType = string

 const (
@@ -35,7 +68,7 @@ func init() {
 	})
 }

-// 计算账单数量
+// CountAllUserBills 计算账单数量
 func (this *UserBillDAO) CountAllUserBills(tx *dbs.Tx, isPaid int32, userId int64, month string) (int64, error) {
 	query := this.Query(tx)
 	if isPaid == 0 {
@@ -52,7 +85,7 @@ func (this *UserBillDAO) CountAllUserBills(tx *dbs.Tx, isPaid int32, userId int6
 	return query.Count()
 }

-// 列出单页账单
+// ListUserBills 列出单页账单
 func (this *UserBillDAO) ListUserBills(tx *dbs.Tx, isPaid int32, userId int64, month string, offset, size int64) (result []*UserBill, err error) {
 	query := this.Query(tx)
 	if isPaid == 0 {
@@ -75,19 +108,42 @@ func (this *UserBillDAO) ListUserBills(tx *dbs.Tx, isPaid int32, userId int64, m
 	return
 }

-// 创建账单
-func (this *UserBillDAO) CreateBill(tx *dbs.Tx, userId int64, billType BillType, description string, amount float32, month string) (int64, error) {
-	op := NewUserBillOperator()
-	op.UserId = userId
-	op.Type = billType
-	op.Description = description
-	op.Amount = amount
-	op.Month = month
-	op.IsPaid = false
-	return this.SaveInt64(tx, op)
+// FindUnpaidBills 查找未支付订单
+func (this *UserBillDAO) FindUnpaidBills(tx *dbs.Tx, size int64) (result []*UserBill, err error) {
+	if size <= 0 {
+		size = 10000
+	}
+	_, err = this.Query(tx).
+		Attr("isPaid", false).
+		Lt("month", timeutil.Format("Ym")). //当月的不能支付，因为当月还没过完
+		Limit(size).
+		DescPk().
+		Slice(&result).
+		FindAll()
+	return
 }

-// 检查是否有当月账单
+// CreateBill 创建账单
+func (this *UserBillDAO) CreateBill(tx *dbs.Tx, userId int64, billType BillType, description string, amount float32, month string) error {
+	code, err := this.GenerateBillCode(tx)
+	if err != nil {
+		return err
+	}
+	return this.Query(tx).
+		InsertOrUpdateQuickly(maps.Map{
+			"userId":      userId,
+			"type":        billType,
+			"description": description,
+			"amount":      amount,
+			"month":       month,
+			"code":        code,
+			"isPaid":      false,
+		}, maps.Map{
+			"amount": amount,
+		})
+}
+
+// ExistBill 检查是否有当月账单
 func (this *UserBillDAO) ExistBill(tx *dbs.Tx, userId int64, billType BillType, month string) (bool, error) {
 	return this.Query(tx).
 		Attr("userId", userId).
@@ -96,47 +152,10 @@ func (this *UserBillDAO) ExistBill(tx *dbs.Tx, userId int64, billType BillType,
 		Exist()
 }

-// 生成账单
+// GenerateBills 生成账单
 // month 格式YYYYMM
 func (this *UserBillDAO) GenerateBills(tx *dbs.Tx, month string) error {
-	// 用户
-	offset := int64(0)
-	size := int64(100) // 每次只查询N次，防止由于执行时间过长而锁表
-	for {
-		userIds, err := SharedUserDAO.ListEnabledUserIds(tx, offset, size)
-		if err != nil {
-			return err
-		}
-		offset += size
-		if len(userIds) == 0 {
-			break
-		}
-
-		for _, userId := range userIds {
-			// CDN流量账单
-			err := this.generateTrafficBill(tx, userId, month)
-			if err != nil {
-				return err
-			}
-		}
-	}
-
-	return nil
-}
-
-// 生成CDN流量账单
-// month 格式YYYYMM
-func (this *UserBillDAO) generateTrafficBill(tx *dbs.Tx, userId int64, month string) error {
-	// 检查是否已经有账单了
-	b, err := this.ExistBill(tx, userId, BillTypeTraffic, month)
-	if err != nil {
-		return err
-	}
-	if b {
-		return nil
-	}
-
-	// TODO 优化使用缓存
+	// 区域价格
 	regions, err := SharedNodeRegionDAO.FindAllEnabledRegionPrices(tx)
 	if err != nil {
 		return err
@@ -153,18 +172,104 @@ func (this *UserBillDAO) generateTrafficBill(tx *dbs.Tx, userId int64, month str
 		return nil
 	}

-	cost := float32(0)
+	// 计算套餐费用
+	plans, err := SharedPlanDAO.FindAllEnabledPlans(tx)
+	if err != nil {
+		return err
+	}
+	var planMap = map[int64]*Plan{}
+	for _, plan := range plans {
+		planMap[int64(plan.Id)] = plan
+	}
+
+	stats, err := SharedServerDailyStatDAO.FindMonthlyStatsWithPlan(tx, month)
+	if err != nil {
+		return err
+	}
+	for _, stat := range stats {
+		plan, ok := planMap[int64(stat.PlanId)]
+		if !ok {
+			continue
+		}
+		if plan.PriceType != serverconfigs.PlanPriceTypeTraffic {
+			continue
+		}
+		if len(plan.TrafficPrice) == 0 {
+			continue
+		}
+		var priceConfig = &serverconfigs.PlanTrafficPrice{}
+		err = json.Unmarshal([]byte(plan.TrafficPrice), priceConfig)
+		if err != nil {
+			return err
+		}
+		if priceConfig.Base > 0 {
+			var fee = priceConfig.Base * (float32(stat.Bytes) / 1024 / 1024 / 1024)
+			err = SharedServerDailyStatDAO.UpdateStatFee(tx, int64(stat.Id), fee)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	// 用户
+	offset := int64(0)
+	size := int64(100) // 每次只查询N次，防止由于执行时间过长而锁表
+	for {
+		userIds, err := SharedUserDAO.ListEnabledUserIds(tx, offset, size)
+		if err != nil {
+			return err
+		}
+		offset += size
+		if len(userIds) == 0 {
+			break
+		}
+
+		for _, userId := range userIds {
+			// CDN流量账单
+			err := this.generateTrafficBill(tx, userId, month, regions, priceItems)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+// UpdateUserBillIsPaid 设置账单已支付
+func (this *UserBillDAO) UpdateUserBillIsPaid(tx *dbs.Tx, billId int64, isPaid bool) error {
+	return this.Query(tx).
+		Pk(billId).
+		Set("isPaid", isPaid).
+		UpdateQuickly()
+}
+
+// 生成CDN流量账单
+// month 格式YYYYMM
+func (this *UserBillDAO) generateTrafficBill(tx *dbs.Tx, userId int64, month string, regions []*NodeRegion, priceItems []*NodePriceItem) error {
+	// 检查是否已经有账单了
+	if month < timeutil.Format("Ym") {
+		b, err := this.ExistBill(tx, userId, BillTypeTraffic, month)
+		if err != nil {
+			return err
+		}
+		if b {
+			return nil
+		}
+	}
+
+	var cost = float32(0)
 	for _, region := range regions {
 		if len(region.Prices) == 0 || region.Prices == "null" {
 			continue
 		}
 		priceMap := map[string]float32{}
-		err = json.Unmarshal([]byte(region.Prices), &priceMap)
+		err := json.Unmarshal([]byte(region.Prices), &priceMap)
 		if err != nil {
 			return err
 		}

-		trafficBytes, err := SharedServerDailyStatDAO.SumUserMonthly(tx, userId, int64(region.Id), month)
+		trafficBytes, err := SharedServerDailyStatDAO.SumUserMonthlyWithoutPlan(tx, userId, int64(region.Id), month)
 		if err != nil {
 			return err
 		}
@@ -187,16 +292,22 @@ func (this *UserBillDAO) generateTrafficBill(tx *dbs.Tx, userId int64, month str
 		cost += (float32(trafficBytes*8) / 1_000_000_000) * price
 	}

+	// 套餐费用
+	planFee, err := SharedServerDailyStatDAO.SumUserMonthlyFee(tx, userId, month)
+	if err != nil {
+		return err
+	}
+	cost += float32(planFee)
+
 	if cost == 0 {
 		return nil
 	}

 	// 创建账单
-	_, err = this.CreateBill(tx, userId, BillTypeTraffic, "按流量计费", cost, month)
-	return err
+	return this.CreateBill(tx, userId, BillTypeTraffic, "按流量计费", cost, month)
 }

-// 获取账单类型名称
+// BillTypeName 获取账单类型名称
 func (this *UserBillDAO) BillTypeName(billType BillType) string {
 	switch billType {
 	case BillTypeTraffic:
@@ -204,3 +315,18 @@ func (this *UserBillDAO) BillTypeName(billType BillType) string {
 	}
 	return ""
 }
+
+// GenerateBillCode 生成账单编号
+func (this *UserBillDAO) GenerateBillCode(tx *dbs.Tx) (string, error) {
+	var code = timeutil.Format("YmdHis") + types.String(rands.Int(100000, 999999))
+	exists, err := this.Query(tx).
+		Attr("code", code).
+		Exist()
+	if err != nil {
+		return "", err
+	}
+	if !exists {
+		return code, nil
+	}
+	return this.GenerateBillCode(tx)
+}