优化代码

internal/db/models/node_region_dao.go

@@ -120,8 +120,8 @@ func (this *NodeRegionDAO) FindAllEnabledRegionPrices(tx *dbs.Tx) (result []*Nod
 	return
 }
 
-// FindAllEnabledAndOnRegions 列出所有启用的区域
-func (this *NodeRegionDAO) FindAllEnabledAndOnRegions(tx *dbs.Tx) (result []*NodeRegion, err error) {
+// FindAllAvailableRegions 列出所有启用的区域
+func (this *NodeRegionDAO) FindAllAvailableRegions(tx *dbs.Tx) (result []*NodeRegion, err error) {
 	_, err = this.Query(tx).
 		State(NodeRegionStateEnabled).
 		Attr("isOn", true).
@@ -132,6 +132,25 @@ func (this *NodeRegionDAO) FindAllEnabledAndOnRegions(tx *dbs.Tx) (result []*Nod
 	return
 }
 
+// FindAllRegionIds 查找所有区域ID
+func (this *NodeRegionDAO) FindAllRegionIds(tx *dbs.Tx) ([]int64, error) {
+	ones, err := this.Query(tx).
+		ResultPk().
+		State(NodeRegionStateEnabled).
+		Desc("order").
+		AscPk().
+		FindAll()
+	if err != nil {
+		return nil, err
+	}
+
+	var regionIds = []int64{}
+	for _, one := range ones {
+		regionIds = append(regionIds, int64(one.(*NodeRegion).Id))
+	}
+	return regionIds, nil
+}
+
 // UpdateRegionOrders 排序
 func (this *NodeRegionDAO) UpdateRegionOrders(tx *dbs.Tx, regionIds []int64) error {
 	order := len(regionIds)

internal/db/models/server_daily_stat_dao.go

@@ -86,7 +86,7 @@ func (this *ServerDailyStatDAO) SaveStats(tx *dbs.Tx, stats []*pb.ServerDailySta
 			InsertOrUpdate(maps.Map{
 				"userId":              serverUserId,
 				"serverId":            stat.ServerId,
-				"regionId":            stat.RegionId,
+				"regionId":            stat.NodeRegionId,
 				"bytes":               stat.Bytes,
 				"cachedBytes":         stat.CachedBytes,
 				"countRequests":       stat.CountRequests,

internal/db/models/user_traffic_bill_model.go

@@ -2,27 +2,29 @@ package models
 
 // UserTrafficBill 用户流量/带宽账单
 type UserTrafficBill struct {
-	Id           uint64  `field:"id"`           // ID
-	BillId       uint64  `field:"billId"`       // 主账单ID
-	RegionId     uint32  `field:"regionId"`     // 区域ID
-	Amount       float64 `field:"amount"`       // 金额
-	BandwidthMB  float64 `field:"bandwidthMB"`  // 带宽MB
-	TrafficGB    float64 `field:"trafficGB"`    // 流量GB
-	PricePerUnit float64 `field:"pricePerUnit"` // 单位价格
-	PriceType    string  `field:"priceType"`    // 计费方式：traffic|bandwidth
-	State        uint8   `field:"state"`        // 状态
+	Id                  uint64  `field:"id"`                  // ID
+	BillId              uint64  `field:"billId"`              // 主账单ID
+	RegionId            uint32  `field:"regionId"`            // 区域ID
+	Amount              float64 `field:"amount"`              // 金额
+	BandwidthMB         float64 `field:"bandwidthMB"`         // 带宽MB
+	BandwidthPercentile uint8   `field:"bandwidthPercentile"` // 带宽百分位
+	TrafficGB           float64 `field:"trafficGB"`           // 流量GB
+	PricePerUnit        float64 `field:"pricePerUnit"`        // 单位价格
+	PriceType           string  `field:"priceType"`           // 计费方式：traffic|bandwidth
+	State               uint8   `field:"state"`               // 状态
 }
 
 type UserTrafficBillOperator struct {
-	Id           any // ID
-	BillId       any // 主账单ID
-	RegionId     any // 区域ID
-	Amount       any // 金额
-	BandwidthMB  any // 带宽MB
-	TrafficGB    any // 流量GB
-	PricePerUnit any // 单位价格
-	PriceType    any // 计费方式：traffic|bandwidth
-	State        any // 状态
+	Id                  any // ID
+	BillId              any // 主账单ID
+	RegionId            any // 区域ID
+	Amount              any // 金额
+	BandwidthMB         any // 带宽MB
+	BandwidthPercentile any // 带宽百分位
+	TrafficGB           any // 流量GB
+	PricePerUnit        any // 单位价格
+	PriceType           any // 计费方式：traffic|bandwidth
+	State               any // 状态
 }
 
 func NewUserTrafficBillOperator() *UserTrafficBillOperator {