80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package cron
|
||
|
||
import (
|
||
"SciencesServer/app/common/model"
|
||
"SciencesServer/app/service"
|
||
"SciencesServer/config"
|
||
"SciencesServer/serve/logger"
|
||
"SciencesServer/serve/orm"
|
||
"time"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type Tenant struct{}
|
||
|
||
type TenantHandle func() (string, func())
|
||
|
||
// ContractDeadline 合同处理
|
||
func (this *Tenant) ContractDeadline() TenantHandle {
|
||
return func() (string, func()) {
|
||
return "0 0 0 * * ?", func() {
|
||
now := time.Now()
|
||
|
||
mSysTenant := model.NewSysTenant()
|
||
|
||
out := make([]*model.SysTenant, 0)
|
||
|
||
err := model.Find(mSysTenant, &out, &model.ModelWhereOrder{Where: model.NewWhereIn("status", []model.SysTenantStatus{
|
||
model.SysTenantStatusForNormal, model.SysTenantStatusForWellExpire,
|
||
})}, &model.ModelWhereOrder{Where: model.NewWhereCondition("deadline", ">=", now.AddDate(0, 0, -7))})
|
||
|
||
if err != nil {
|
||
logger.ErrorF("Cron Tenant ContractDeadline Error：%v", err)
|
||
return
|
||
}
|
||
well := make([]uint64, 0)
|
||
expired := make([]uint64, 0)
|
||
expiredKey := make([]string, 0)
|
||
|
||
for _, v := range out {
|
||
// 超过协议时间
|
||
if v.Deadline.After(now) {
|
||
expired = append(expired, v.ID)
|
||
expiredKey = append(expiredKey, v.Key)
|
||
continue
|
||
}
|
||
if v.Status == model.SysTenantStatusForNormal {
|
||
well = append(well, v.ID)
|
||
}
|
||
}
|
||
if err = orm.GetDB().Transaction(func(tx *gorm.DB) error {
|
||
if len(well) > 0 {
|
||
if err = model.UpdatesWhere(mSysTenant, map[string]interface{}{
|
||
"status": model.SysTenantStatusForWellExpire, "updated_at": now,
|
||
}, []*model.ModelWhere{model.NewWhereIn("id", well)}, tx); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
if len(expired) > 0 {
|
||
if err = model.UpdatesWhere(mSysTenant, map[string]interface{}{
|
||
"status": model.SysTenantStatusForExpired, "updated_at": now,
|
||
}, []*model.ModelWhere{model.NewWhereIn("id", expired)}, tx); err != nil {
|
||
return err
|
||
}
|
||
service.Publish(config.EventForRedisListDestroy, config.RedisKeyForTenant, expiredKey)
|
||
}
|
||
return nil
|
||
}); err != nil {
|
||
logger.ErrorF("Cron Tenant ContractDeadline Transaction Error：%v", err)
|
||
return
|
||
}
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
func NewTenant() *Tenant {
|
||
return &Tenant{}
|
||
}
|