You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.4 KiB
93 lines
2.4 KiB
package income |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
"time" |
|
|
|
"go-common/app/job/main/growup/conf" |
|
"go-common/app/job/main/growup/dao/dataplatform" |
|
"go-common/app/job/main/growup/dao/email" |
|
incomeD "go-common/app/job/main/growup/dao/income" |
|
"go-common/app/job/main/growup/service" |
|
"go-common/app/job/main/growup/service/ctrl" |
|
"go-common/library/log" |
|
) |
|
|
|
// Service struct |
|
type Service struct { |
|
conf *conf.Config |
|
dao *incomeD.Dao |
|
avCharge *AvChargeSvr |
|
income *Income |
|
ratio *ChargeRatioSvr |
|
email *email.Dao |
|
dp *dataplatform.Dao |
|
} |
|
|
|
// New fn |
|
func New(c *conf.Config, executor ctrl.Executor) (s *Service) { |
|
s = &Service{ |
|
conf: c, |
|
dao: incomeD.New(c), |
|
email: email.New(c), |
|
dp: dataplatform.New(c), |
|
} |
|
s.avCharge = NewAvChargeSvr(s.dao) |
|
s.income = NewIncome(batchSize, s.dao) |
|
s.ratio = NewChargeRatioSvr(s.dao) |
|
log.Info("income service start") |
|
|
|
executor.Submit( |
|
s.calDailyCreativeIncome, |
|
s.syncBGM, |
|
) |
|
return s |
|
} |
|
|
|
func (s *Service) syncBGM(ctx context.Context) { |
|
for { |
|
time.Sleep(service.NextDay(1, 0, 0)) |
|
msg := "" |
|
log.Info("sync BGM begin:%v", time.Now().Format("2006-01-02 15:04:05")) |
|
err := s.SyncBgmInfo(context.TODO()) |
|
if err != nil { |
|
log.Error("s.SyncBgmInfo error(%v)", err) |
|
} |
|
if err != nil { |
|
msg = fmt.Sprintf("SyncBgmInfo error(%v)", err) |
|
} else { |
|
msg = "Success" |
|
} |
|
err = s.email.SendMail(time.Now(), msg, "创作激励同步bgm%d年%d月%d日", "[email protected]", "[email protected]", "[email protected]") |
|
if err != nil { |
|
log.Error("s.email.SendMail error(%v)", err) |
|
} |
|
log.Info("sync BGM end:%v", time.Now().Format("2006-01-02 15:04:05")) |
|
} |
|
} |
|
|
|
func (s *Service) calDailyCreativeIncome(ctx context.Context) { |
|
for { |
|
time.Sleep(service.NextDay(18, 0, 0)) |
|
log.Info("calDailyCreativeIncome begin:%v", time.Now().Format("2006-01-02 15:04:05")) |
|
now := time.Now() |
|
date := time.Date(now.Year(), now.Month(), now.Day()-1, 0, 0, 0, 0, time.Local) |
|
err := s.RunAndSendMail(context.TODO(), date) |
|
if err != nil { |
|
log.Error("s.RunAndSendMail error(%v)", err) |
|
} |
|
if err == nil { |
|
err = s.RunStatis(context.TODO(), date) |
|
if err != nil { |
|
log.Error("s.RunStatis error(%v)", err) |
|
} |
|
} |
|
log.Info("calDailyCreativeIncome end:%v", time.Now().Format("2006-01-02 15:04:05")) |
|
} |
|
} |
|
|
|
// Ping check dao health. |
|
func (s *Service) Ping(c context.Context) (err error) { |
|
return s.dao.Ping(c) |
|
}
|
|
|