From 0aa83fefcf79a87d93636cee9d3b7c8ca529752c Mon Sep 17 00:00:00 2001
From: Mary Smiley <71322561+bothyouandme@users.noreply.github.com>
Date: Sat, 16 Mar 2024 09:39:10 +0800
Subject: [PATCH] Update relay-audio.go

---
 relay/relay-audio.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/relay/relay-audio.go b/relay/relay-audio.go
index 5d43e79..7c5290d 100644
--- a/relay/relay-audio.go
+++ b/relay/relay-audio.go
@@ -189,6 +189,19 @@ func AudioHelper(c *gin.Context, relayMode int) *dto.OpenAIErrorWithStatusCode {
 				channelId := c.GetInt("channel_id")
 				model.UpdateChannelUsedQuota(channelId, quota)
 			}
+			if common.AutomaticRefreshLinuxDoUserQuotaEnabled {
+				userRefreshTimeStamp , err := model.CacheGetUserRefreshTimeStamp(userId)
+				if err != nil {
+					common.SysError("error get user refresh timestamp: " + err.Error())
+				}
+				TimeStamp := startTime.Unix()
+				if TimeStamp - userRefreshTimeStamp > int64(common.LinuxDoUserQuotaRefreshInterval * 3600) {
+					model.RefreshUserQuotaAndSave(userId, TimeStamp)
+					model.CacheUpdateUserQuota(userId)
+					model.CacheUpdateUserRefreshTimeStamp(userId, TimeStamp)
+
+				}
+			}
 		}()
 	}(c.Request.Context())