From c60f7557153f5cced0f5c2af30c5e4183ecfb24f Mon Sep 17 00:00:00 2001 From: Martial BE Date: Thu, 21 Dec 2023 10:24:07 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20support=20mysql=20of=20statistic?= =?UTF-8?q?s?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- model/log.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/model/log.go b/model/log.go index 31ed61e7..a8699673 100644 --- a/model/log.go +++ b/model/log.go @@ -194,8 +194,14 @@ func DeleteOldLog(targetTimestamp int64) (int64, error) { } func SearchLogsByDayAndModel(user_id, start, end int) (LogStatistics []*LogStatistic, err error) { + groupSelect := "DATE_FORMAT(FROM_UNIXTIME(created_at), '%Y-%m-%d') as day" + + if common.UsingPostgreSQL { + groupSelect = "TO_CHAR(date_trunc('day', to_timestamp(created_at)), 'YYYY-MM-DD') as day" + } + err = DB.Raw(` - SELECT TO_CHAR(date_trunc('day', to_timestamp(created_at)), 'YYYY-MM-DD') as day, + SELECT `+groupSelect+`, model_name, count(1) as request_count, sum(quota) as quota, sum(prompt_tokens) as prompt_tokens,