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

// 剩余字符统计
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
}