aggregate_translate_server/app/jobs/translate.go

package jobs

import (
	"go-admin/app/admin/service"

	"github.com/go-admin-team/go-admin-core/logger"
	"github.com/go-admin-team/go-admin-core/sdk"
	"gorm.io/gorm"
)

type DailyJob struct{}

type RemainCharJob struct{}

type CleanExpiredOrderJob struct{}

// 同步剩余字符
type SyncRemainCharJob struct{}

// 定时同步第三方用量
func (t SyncRemainCharJob) Exec(arg interface{}) error {
	platformAccountService := service.TmPlatformAccount{}
	platformAccountService.Orm = GetDb()
	platformAccountService.Log = logger.NewHelper(logger.DefaultLogger)

	return platformAccountService.SyncAll()
}

// 清理过期订单
func (t CleanExpiredOrderJob) Exec(arg interface{}) error {
	// expireTime := time.Now().Add(5 * time.Minute)
	rechargeLogService := service.TmRechargeLog{}
	rechargeLogService.Orm = GetDb()
	rechargeLogService.Log = logger.NewHelper(logger.DefaultLogger)

	return rechargeLogService.CleanExpiredOrder()
}

// 剩余字符统计
func (t RemainCharJob) Exec(arg interface{}) error {
	memberService := service.TmMember{}
	memberService.Orm = GetDb()
	memberService.Log = logger.NewHelper(logger.DefaultLogger)

	return memberService.SyncMemberRemain()
}

// 每日使用字符量统计
func (t DailyJob) Exec(arg interface{}) error {
	//TODO: 实现每日使用字符量统计逻辑
	memberService := service.TmMember{}
	memberService.Orm = GetDb()
	memberService.Log = logger.NewHelper(logger.DefaultLogger)

	return memberService.SyncMemberDailyUsage()
}

func GetDb() *gorm.DB {
	dbs := sdk.Runtime.GetDb()

	for _, db := range dbs {
		return db
	}

	return nil
}