mirror of
https://gitee.com/blackfox/geekai.git
synced 2024-12-05 21:58:23 +08:00
155 lines
4.5 KiB
Go
155 lines
4.5 KiB
Go
package service
|
||
|
||
import (
|
||
"chatplus/core/types"
|
||
logger2 "chatplus/logger"
|
||
"chatplus/store/model"
|
||
"chatplus/utils"
|
||
"context"
|
||
"fmt"
|
||
"github.com/xxl-job/xxl-job-executor-go"
|
||
"gorm.io/gorm"
|
||
"time"
|
||
)
|
||
|
||
var logger = logger2.GetLogger()
|
||
|
||
type XXLJobExecutor struct {
|
||
executor xxl.Executor
|
||
db *gorm.DB
|
||
}
|
||
|
||
func NewXXLJobExecutor(config *types.AppConfig, db *gorm.DB) *XXLJobExecutor {
|
||
if !config.XXLConfig.Enabled {
|
||
logger.Info("XXL-JOB service is disabled")
|
||
return nil
|
||
}
|
||
|
||
exec := xxl.NewExecutor(
|
||
xxl.ServerAddr(config.XXLConfig.ServerAddr),
|
||
xxl.AccessToken(config.XXLConfig.AccessToken), //请求令牌(默认为空)
|
||
xxl.ExecutorIp(config.XXLConfig.ExecutorIp), //可自动获取
|
||
xxl.ExecutorPort(config.XXLConfig.ExecutorPort), //默认9999(非必填)
|
||
xxl.RegistryKey(config.XXLConfig.RegistryKey), //执行器名称
|
||
xxl.SetLogger(&customLogger{}), //自定义日志
|
||
)
|
||
exec.Init()
|
||
return &XXLJobExecutor{executor: exec, db: db}
|
||
}
|
||
|
||
func (e *XXLJobExecutor) Run() error {
|
||
e.executor.RegTask("ClearOrders", e.ClearOrders)
|
||
e.executor.RegTask("ResetVipCalls", e.ResetVipCalls)
|
||
return e.executor.Run()
|
||
}
|
||
|
||
// ClearOrders 清理未支付的订单,如果没有抛出异常则表示执行成功
|
||
func (e *XXLJobExecutor) ClearOrders(cxt context.Context, param *xxl.RunReq) (msg string) {
|
||
logger.Debug("执行清理未支付订单...")
|
||
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()
|
||
}
|
||
|
||
if config.OrderPayTimeout == 0 { // 默认未支付订单的生命周期为 30 分钟
|
||
config.OrderPayTimeout = 1800
|
||
}
|
||
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)
|
||
}
|
||
|
||
// ResetVipCalls 清理过期的 VIP 会员
|
||
func (e *XXLJobExecutor) ResetVipCalls(cxt context.Context, param *xxl.RunReq) (msg string) {
|
||
logger.Info("开始进行月底账号盘点...")
|
||
var users []model.User
|
||
res := e.db.Where("vip = ?", 1).Find(&users)
|
||
if res.Error != nil {
|
||
return "No vip users found"
|
||
}
|
||
|
||
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()
|
||
}
|
||
|
||
// 获取本月月初时间
|
||
currentTime := time.Now()
|
||
year, month, _ := currentTime.Date()
|
||
firstOfMonth := time.Date(year, month, 1, 0, 0, 0, 0, currentTime.Location()).Unix()
|
||
for _, u := range users {
|
||
// 账号到期,直接清零
|
||
if u.ExpiredTime <= currentTime.Unix() {
|
||
logger.Info("账号过期:", u.Username)
|
||
u.Calls = 0
|
||
u.Vip = false
|
||
} else {
|
||
if u.Calls <= 0 {
|
||
u.Calls = 0
|
||
}
|
||
if u.ImgCalls <= 0 {
|
||
u.ImgCalls = 0
|
||
}
|
||
// 如果该用户当月有充值点卡,则将点卡中未用完的点数结余到下个月
|
||
var orders []model.Order
|
||
e.db.Debug().Where("user_id = ? AND pay_time > ?", u.Id, firstOfMonth).Find(&orders)
|
||
var calls = 0
|
||
var imgCalls = 0
|
||
for _, o := range orders {
|
||
var remark types.OrderRemark
|
||
err = utils.JsonDecode(o.Remark, &remark)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
if remark.Days > 0 { // 会员续费
|
||
continue
|
||
}
|
||
calls += remark.Calls
|
||
imgCalls += remark.ImgCalls
|
||
}
|
||
if u.Calls > calls { // 本月套餐没有用完
|
||
u.Calls = calls + config.VipMonthCalls
|
||
} else {
|
||
u.Calls = u.Calls + config.VipMonthCalls
|
||
}
|
||
if u.ImgCalls > imgCalls { // 本月套餐没有用完
|
||
u.ImgCalls = imgCalls + config.VipMonthImgCalls
|
||
} else {
|
||
u.ImgCalls = u.ImgCalls + config.VipMonthImgCalls
|
||
}
|
||
logger.Infof("%s 点卡结余:%d", u.Username, calls)
|
||
}
|
||
u.Tokens = 0
|
||
// update user
|
||
e.db.Updates(&u)
|
||
}
|
||
logger.Info("月底盘点完成!")
|
||
return "success"
|
||
}
|
||
|
||
type customLogger struct{}
|
||
|
||
func (l *customLogger) Info(format string, a ...interface{}) {
|
||
logger.Debugf(format, a...)
|
||
}
|
||
|
||
func (l *customLogger) Error(format string, a ...interface{}) {
|
||
logger.Errorf(format, a...)
|
||
}
|