This commit is contained in:
xxjwxc
2020-09-29 00:02:57 +08:00
parent 56390a763e
commit 3db1045537
2 changed files with 5 additions and 2 deletions

View File

@@ -6,6 +6,7 @@ import (
"github.com/xxjwxc/public/dev"
myerrors "github.com/xxjwxc/public/errors"
"gorm.io/gorm/logger"
"gorm.io/gorm/schema"
"github.com/xxjwxc/public/mylog"
@@ -30,7 +31,8 @@ func (i *MySqlDB) OnGetDBOrm(dataSourceName string) *gorm.DB {
if i.DB == nil {
var err error
i.DB, err = gorm.Open(mysql.Open(dataSourceName), &gorm.Config{PrepareStmt: false,
Logger: logger.Default})
NamingStrategy: schema.NamingStrategy{SingularTable: true}, // 全局禁用表名复数
Logger: logger.Default})
if err != nil {
mylog.Error(myerrors.Wrap(err, "Got error when connect database:"+dataSourceName))
return nil

View File

@@ -57,14 +57,15 @@ func OnPeMonth(day int, hour, min, sec int, callback func()) {
*/
func OnPeDay(hour, min, sec int, callback func()) {
go func() {
next := time.Now()
for {
next := time.Now().AddDate(0, 1, 0)
next = time.Date(next.Year(), next.Month(), next.Day(), hour, min, sec, 0, next.Location())
mylog.Infof("next pe day on:%v", tools.GetTimeStr(next))
t := time.NewTimer(next.Sub(time.Now()))
log.Println("next time callback:", next)
<-t.C
callback()
next = time.Now().AddDate(0, 0, 1) // 下一天
}
}()
}