opensource/cloudpods
3
0
Fork 0
mirror of https://github.com/yunionio/cloudpods.git synced 2026-08-30 17:13:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(aiproxy): remove 30-day limit on custom usage range (#25356)

Browse Source 
Allow querying usage statistics over custom date ranges longer than 30 days.
This commit is contained in:
Zexi Li
2026-08-17 17:00:14 +08:00
committed by GitHub
parent 21ec4f7f5c
commit d4876a18bd
2 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/aiproxy/handlers/usage.go
-3
View File
@@ -622,9 +622,6 @@ func usageCustomRange(rawStart, rawEnd string, loc *time.Location) (time.Time, t
if !end.After(start) {
return time.Time{}, time.Time{}, errors.New("end must be after start")
}
if end.After(start.AddDate(0, 0, 30)) {
return time.Time{}, time.Time{}, errors.New("range too long")
}
return start, end, nil
}
pkg/aiproxy/handlers/usage_range_test.go
+8 -7
View File
@@ -103,15 +103,16 @@ func TestUsageRangeEndNotAfterStart(t *testing.T) {
}
}
func TestUsageRangeTooLong(t *testing.T) {
func TestUsageRangeLongWindowAllowed(t *testing.T) {
now := time.Date(2026, 8, 13, 16, 0, 0, 0, time.UTC)
_, _, err := usageRange("custom", "2026-01-01T00:00:00Z", "2026-02-02T00:00:00Z", now)
if err == nil {
t.Fatal("expected error for range longer than 30 days")
}
_, _, err = usageRange("custom", "2026-01-01T00:00:00Z", "2026-01-31T00:00:00Z", now)
start, end, err := usageRange("custom", "2026-01-01T00:00:00Z", "2026-02-02T00:00:00Z", now)
if err != nil {
t.Fatalf("30-day window should be allowed: %v", err)
t.Fatalf("long window should be allowed: %v", err)
}
wantStart := time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC)
wantEnd := time.Date(2026, 2, 2, 0, 0, 0, 0, time.UTC)
if !start.Equal(wantStart) || !end.Equal(wantEnd) {
t.Fatalf("got %v %v, want %v %v", start, end, wantStart, wantEnd)
}
}