feat: integrate xxl-job-admin to implements automatic task scheduling

api/service/xxl_job_service.go

@@ -40,12 +40,13 @@ func NewXXLJobExecutor(config *types.AppConfig, db *gorm.DB) *XXLJobExecutor {
 func (e *XXLJobExecutor) Run() error {
 	e.executor.RegTask("ClearOrders", e.ClearOrders)
 	e.executor.RegTask("ResetVipPower", e.ResetVipPower)
+	e.executor.RegTask("ResetUserPower", e.ResetUserPower)
 	return e.executor.Run()
 }
 
 // ClearOrders 清理未支付的订单，如果没有抛出异常则表示执行成功
 func (e *XXLJobExecutor) ClearOrders(cxt context.Context, param *xxl.RunReq) (msg string) {
-	logger.Debug("执行清理未支付订单...")
+	logger.Info("执行清理未支付订单...")
 	var sysConfig model.Config
 	res := e.db.Where("marker", "system").First(&sysConfig)
 	if res.Error != nil {
@@ -64,15 +65,17 @@ func (e *XXLJobExecutor) ClearOrders(cxt context.Context, param *xxl.RunReq) (ms
 	timeout := time.Now().Unix() - int64(config.OrderPayTimeout)
 	start := utils.Stamp2str(timeout)
 	// 这里不是用软删除，而是永久删除订单
-	res = e.db.Unscoped().Where("status != ? AND created_at < ?", types.OrderPaidSuccess, start).Delete(&model.Order{})
-	return fmt.Sprintf("Clear order successfully, affect rows: %d", res.RowsAffected)
+	res = e.db.Unscoped().Where("status IN ? AND created_at < ?", []types.OrderStatus{types.OrderNotPaid, types.OrderScanned}, start).Delete(&model.Order{})
+	logger.Infof("Clear order successfully, affect rows: %d", res.RowsAffected)
+	return "success"
 }
 
-// ResetVipPower 清理过期的 VIP 会员
+// ResetVipPower 重置VIP会员算力
+// 自动将 VIP 会员的算力补充到每月赠送的最大值
 func (e *XXLJobExecutor) ResetVipPower(cxt context.Context, param *xxl.RunReq) (msg string) {
 	logger.Info("开始进行月底账号盘点...")
 	var users []model.User
-	res := e.db.Where("vip = ?", 1).Find(&users)
+	res := e.db.Where("vip", 1).Where("status", 1).Find(&users)
 	if res.Error != nil {
 		return "No vip users found"
 	}
@@ -90,23 +93,27 @@ func (e *XXLJobExecutor) ResetVipPower(cxt context.Context, param *xxl.RunReq) (
 	}
 
 	for _, u := range users {
-		if u.Power <= 0 {
-			u.Power = 0
+		// 处理过期的 VIP
+		if u.ExpiredTime > 0 && u.ExpiredTime <= time.Now().Unix() {
+			u.Vip = false
+			e.db.Model(&model.User{}).Where("id", u.Id).UpdateColumn("vip", false)
+			continue
 		}
-		u.Power += config.VipMonthPower
 		// update user
-		tx := e.db.Updates(&u)
-		// 记录算力充值日志
+		tx := e.db.Model(&model.User{}).Where("id", u.Id).UpdateColumn("power", gorm.Expr("power + ?", config.VipMonthPower))
+		// 记录算力变动日志
 		if tx.Error == nil {
+			var user model.User
+			e.db.Where("id", u.Id).First(&user)
 			e.db.Create(&model.PowerLog{
 				UserId:    u.Id,
 				Username:  u.Username,
 				Type:      types.PowerRecharge,
 				Amount:    config.VipMonthPower,
 				Mark:      types.PowerAdd,
-				Balance:   u.Power + config.VipMonthPower,
-				Model:     "",
-				Remark:    fmt.Sprintf("月底盘点，会员每月赠送算力：%d", config.VipMonthPower),
+				Balance:   user.Power,
+				Model:     "系统盘点",
+				Remark:    fmt.Sprintf("VIP会员每月算力派发，：%d", config.VipMonthPower),
 				CreatedAt: time.Now(),
 			})
 		}
@@ -115,6 +122,58 @@ func (e *XXLJobExecutor) ResetVipPower(cxt context.Context, param *xxl.RunReq) (
 	return "success"
 }
 
+func (e *XXLJobExecutor) ResetUserPower(cxt context.Context, param *xxl.RunReq) (msg string) {
+	logger.Info("今日算力派发开始：", time.Now())
+	var users []model.User
+	res := e.db.Where("status", 1).Find(&users)
+	if res.Error != nil {
+		return "No matching users"
+	}
+
+	var sysConfig model.Config
+	res = e.db.Where("marker", "system").First(&sysConfig)
+	if res.Error != nil {
+		return "error with get system config: " + res.Error.Error()
+	}
+
+	var config types.SystemConfig
+	err := utils.JsonDecode(sysConfig.Config, &config)
+	if err != nil {
+		return "error with decode system config: " + err.Error()
+	}
+
+	var counter = 0
+	var totalPower = 0
+	for _, u := range users {
+		if u.Power >= config.DailyPower {
+			continue
+		}
+		var power = config.DailyPower - u.Power
+		// update user
+		tx := e.db.Model(&model.User{}).Where("id", u.Id).UpdateColumn("power", gorm.Expr("power + ?", power))
+		// 记录算力充值日志
+		if tx.Error == nil {
+			var user model.User
+			e.db.Where("id", u.Id).First(&user)
+			e.db.Create(&model.PowerLog{
+				UserId:    u.Id,
+				Username:  u.Username,
+				Type:      types.PowerGift,
+				Amount:    power,
+				Mark:      types.PowerAdd,
+				Balance:   user.Power,
+				Model:     "系统赠送",
+				Remark:    fmt.Sprintf("系统每日算力派发，今日额度：%d", config.VipMonthPower),
+				CreatedAt: time.Now(),
+			})
+		}
+		counter++
+		totalPower += power
+	}
+	logger.Infof("今日派发算力结束！累计派发 %d 人，累计派发算力：%d", counter, totalPower)
+	return "success"
+}
+
 type customLogger struct{}
 
 func (l *customLogger) Info(format string, a ...interface{}) {