new

internal/domains/api/entities/api_user.go

@@ -20,12 +20,20 @@ const (
 
 // ApiUser API用户（聚合根）
 type ApiUser struct {
-	ID        string    `gorm:"primaryKey;type:varchar(64)" json:"id"`
-	UserId    string    `gorm:"type:varchar(36);not null;uniqueIndex" json:"user_id"`
-	AccessId  string    `gorm:"type:varchar(64);not null;uniqueIndex" json:"access_id"`
-	SecretKey string    `gorm:"type:varchar(128);not null" json:"secret_key"`
-	Status    string    `gorm:"type:varchar(20);not null;default:'normal'" json:"status"`
-	WhiteList []string  `gorm:"type:json;serializer:json;default:'[]'" json:"white_list"` // 支持多个白名单
+	ID        string   `gorm:"primaryKey;type:varchar(64)" json:"id"`
+	UserId    string   `gorm:"type:varchar(36);not null;uniqueIndex" json:"user_id"`
+	AccessId  string   `gorm:"type:varchar(64);not null;uniqueIndex" json:"access_id"`
+	SecretKey string   `gorm:"type:varchar(128);not null" json:"secret_key"`
+	Status    string   `gorm:"type:varchar(20);not null;default:'normal'" json:"status"`
+	WhiteList []string `gorm:"type:json;serializer:json;default:'[]'" json:"white_list"` // 支持多个白名单
+
+	// 余额预警配置
+	BalanceAlertEnabled   bool       `gorm:"default:true" json:"balance_alert_enabled" comment:"是否启用余额预警"`
+	BalanceAlertThreshold float64    `gorm:"default:200.00" json:"balance_alert_threshold" comment:"余额预警阈值"`
+	AlertPhone            string     `gorm:"type:varchar(20)" json:"alert_phone" comment:"预警手机号"`
+	LastLowBalanceAlert   *time.Time `json:"last_low_balance_alert" comment:"最后低余额预警时间"`
+	LastArrearsAlert      *time.Time `json:"last_arrears_alert" comment:"最后欠费预警时间"`
+
 	CreatedAt time.Time `gorm:"autoCreateTime" json:"created_at"`
 	UpdatedAt time.Time `gorm:"autoUpdateTime" json:"updated_at"`
 }
@@ -51,7 +59,7 @@ func (u *ApiUser) IsFrozen() bool {
 }
 
 // NewApiUser 工厂方法
-func NewApiUser(userId string) (*ApiUser, error) {
+func NewApiUser(userId string, defaultAlertEnabled bool, defaultAlertThreshold float64) (*ApiUser, error) {
 	if userId == "" {
 		return nil, errors.New("用户ID不能为空")
 	}
@@ -64,12 +72,14 @@ func NewApiUser(userId string) (*ApiUser, error) {
 		return nil, err
 	}
 	return &ApiUser{
-		ID:        uuid.New().String(),
-		UserId:    userId,
-		AccessId:  accessId,
-		SecretKey: secretKey,
-		Status:    ApiUserStatusNormal,
-		WhiteList: []string{},
+		ID:                    uuid.New().String(),
+		UserId:                userId,
+		AccessId:              accessId,
+		SecretKey:             secretKey,
+		Status:                ApiUserStatusNormal,
+		WhiteList:             []string{},
+		BalanceAlertEnabled:   defaultAlertEnabled,
+		BalanceAlertThreshold: defaultAlertThreshold,
 	}, nil
 }
 
@@ -124,6 +134,68 @@ func (u *ApiUser) RemoveFromWhiteList(entry string) error {
 	return nil
 }
 
+// 余额预警相关方法
+
+// UpdateBalanceAlertSettings 更新余额预警设置
+func (u *ApiUser) UpdateBalanceAlertSettings(enabled bool, threshold float64, phone string) error {
+	if threshold < 0 {
+		return errors.New("预警阈值不能为负数")
+	}
+	if phone != "" && len(phone) != 11 {
+		return errors.New("手机号格式不正确")
+	}
+
+	u.BalanceAlertEnabled = enabled
+	u.BalanceAlertThreshold = threshold
+	u.AlertPhone = phone
+	return nil
+}
+
+// ShouldSendLowBalanceAlert 是否应该发送低余额预警（24小时冷却期）
+func (u *ApiUser) ShouldSendLowBalanceAlert(balance float64) bool {
+	if !u.BalanceAlertEnabled || u.AlertPhone == "" {
+		return false
+	}
+
+	// 余额低于阈值
+	if balance < u.BalanceAlertThreshold {
+		// 检查是否已经发送过预警（避免频繁发送）
+		if u.LastLowBalanceAlert != nil {
+			// 如果距离上次预警不足24小时，不发送
+			if time.Since(*u.LastLowBalanceAlert) < 24*time.Hour {
+				return false
+			}
+		}
+		return true
+	}
+	return false
+}
+
+// ShouldSendArrearsAlert 是否应该发送欠费预警（不受冷却期限制）
+func (u *ApiUser) ShouldSendArrearsAlert(balance float64) bool {
+	if !u.BalanceAlertEnabled || u.AlertPhone == "" {
+		return false
+	}
+
+	// 余额为负数（欠费）- 欠费预警不受冷却期限制
+	if balance < 0 {
+		return true
+	}
+	return false
+}
+
+// MarkLowBalanceAlertSent 标记低余额预警已发送
+func (u *ApiUser) MarkLowBalanceAlertSent() {
+	now := time.Now()
+	u.LastLowBalanceAlert = &now
+}
+
+// MarkArrearsAlertSent 标记欠费预警已发送
+func (u *ApiUser) MarkArrearsAlertSent() {
+	now := time.Now()
+	u.LastArrearsAlert = &now
+}
+
 // Validate 校验ApiUser聚合根的业务规则
 func (u *ApiUser) Validate() error {
 	if u.UserId == "" {