mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix null data

Browse Source
This commit is contained in:
Sakurasan
2023-04-26 03:03:43 +08:00
parent 56b56c703d
commit 3dae04afcd
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
store/usage.go
View File

@@ -1,9 +1,11 @@
package store package store
import ( import (
"errors"
"time" "time"
"github.com/Sakurasan/to" "github.com/Sakurasan/to"
"gorm.io/gorm"
) )
type DailyUsage struct { type DailyUsage struct {
@@ -95,7 +97,7 @@ func Record(chatlog *Tokens) (err error) {
func SumDaily(userid int) error { func SumDaily(userid int) error {
var count int64 var count int64
err := usage.Model(&DailyUsage{}).Where("user_id = ? and date = ?", userid, time.Date(time.Now().Year(), time.Now().Month(), time.Now().Day(), 0, 0, 0, 0, time.UTC)).Count(&count).Error err := usage.Model(&DailyUsage{}).Where("user_id = ? and date = ?", userid, time.Date(time.Now().Year(), time.Now().Month(), time.Now().Day(), 0, 0, 0, 0, time.UTC)).Count(&count).Error
if err != nil { if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
return err return err
} }
if count == 0 { if count == 0 {