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 TransferJob struct{}

type TransferStatusJob struct{}

// 定期转账
func (t TransferJob) Exec(arg interface{}) error {
	walletService := service.WmWalletInfo{}
	walletService.Orm = getDefaultDb()
	walletService.Log = logger.NewHelper(logger.DefaultLogger)

	return walletService.ScheduledTask()
}

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

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

// 定时查询交易状态
func (t TransferStatusJob) Exec(arg interface{}) error {
	walletService := service.WmTransfer{}
	walletService.Orm = getDefaultDb()
	walletService.Log = logger.NewHelper(logger.DefaultLogger)

	return walletService.CheckHashStatus()
}