43 lines
921 B
Go
43 lines
921 B
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{}
|
||
|
|
|
||
|
|
// 剩余字符统计
|
||
|
|
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
|
||
|
|
}
|