opensource/sub2api
3
0
Fork 0
mirror of https://github.com/Wei-Shaw/sub2api.git synced 2026-09-24 16:05:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: use UTC timezone in quota period SQL and reset period_start on quota reset

Browse Source 
- Use NOW() AT TIME ZONE 'UTC' to ensure consistent timestamps
  regardless of database timezone setting
- Clear quota_period_start when manually resetting quota so a
  fresh period starts from next usage

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
erio
2026-03-06 20:03:16 +08:00
co-authored by Claude Opus 4.6
parent a4ee71d796
commit c087e94413
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/repository/account_repo.go
+7 -8
View File
@@ -1718,8 +1718,8 @@ func (r *accountRepository) IncrementQuotaUsed(ctx context.Context, id int64, am
'{quota_period_start}',
to_jsonb(
CASE WHEN (SELECT needs_reset FROM reset_check)
THEN to_char(NOW(), 'YYYY-MM-DD"T"HH24:MI:SS.US"Z"')
ELSE COALESCE(extra->>'quota_period_start', to_char(NOW(), 'YYYY-MM-DD"T"HH24:MI:SS.US"Z"')) END
THEN to_char(NOW() AT TIME ZONE 'UTC', 'YYYY-MM-DD"T"HH24:MI:SS.US"Z"')
ELSE COALESCE(extra->>'quota_period_start', to_char(NOW() AT TIME ZONE 'UTC', 'YYYY-MM-DD"T"HH24:MI:SS.US"Z"')) END
)
), updated_at = NOW()
WHERE id = $2 AND deleted_at IS NULL
@@ -1764,14 +1764,13 @@ func periodToInterval(period string) string {
}
}
// ResetQuotaUsed 重置账号的 extra.quota_used 为 0
// ResetQuotaUsed 重置账号的 extra.quota_used 为 0，同时清除 quota_period_start 以开始新周期
func (r *accountRepository) ResetQuotaUsed(ctx context.Context, id int64) error {
_, err := r.sql.ExecContext(ctx,
`UPDATE accounts SET extra = jsonb_set(
COALESCE(extra, '{}'::jsonb),
'{quota_used}',
'0'::jsonb
), updated_at = NOW()
`UPDATE accounts SET extra = (
COALESCE(extra, '{}'::jsonb)
|| '{"quota_used": 0}'::jsonb
) - 'quota_period_start', updated_at = NOW()
WHERE id = $1 AND deleted_at IS NULL`,
id)
if err != nil {