From 3dae04afcdc5f43a3b5e44fe8c4004594e91cec9 Mon Sep 17 00:00:00 2001
From: Sakurasan <1173092237@qq.com>
Date: Wed, 26 Apr 2023 03:03:43 +0800
Subject: [PATCH] fix null data

---
 store/usage.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/store/usage.go b/store/usage.go
index 26aedf0..91df4e2 100644
--- a/store/usage.go
+++ b/store/usage.go
@@ -1,9 +1,11 @@
 package store
 
 import (
+	"errors"
 	"time"
 
 	"github.com/Sakurasan/to"
+	"gorm.io/gorm"
 )
 
 type DailyUsage struct {
@@ -95,7 +97,7 @@ func Record(chatlog *Tokens) (err error) {
 func SumDaily(userid int) error {
 	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
-	if err != nil {
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
 		return err
 	}
 	if count == 0 {