da36506b89
P0 fixes: - ModelError.Is(): use exact matching instead of substring contains() - shouldClearStickySession: add context param for cancellation/tracing P1 fixes: - TODO stubs: return 501 Not Implemented errors - validateInstanceSignature: deduplicate to shared validateCodeSignature() - Error messages: standardize to English only - http.go: remove pseudo if-else with duplicate branches
56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
dbent "github.com/Wei-Shaw/sub2api/ent"
|
|
"github.com/Wei-Shaw/sub2api/ent/setting"
|
|
dbuser "github.com/Wei-Shaw/sub2api/ent/user"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
)
|
|
|
|
const (
|
|
simpleModeAdminConcurrencyUpgradeKey = "simple_mode_admin_concurrency_upgraded_30"
|
|
simpleModeLegacyAdminConcurrency = 5
|
|
simpleModeTargetAdminConcurrency = 30
|
|
)
|
|
|
|
func ensureSimpleModeAdminConcurrency(ctx context.Context, client *dbent.Client) error {
|
|
if client == nil {
|
|
return fmt.Errorf("nil ent client")
|
|
}
|
|
|
|
upgraded, err := client.Setting.Query().Where(setting.KeyEQ(simpleModeAdminConcurrencyUpgradeKey)).Exist(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("check admin concurrency upgrade marker: %w", err)
|
|
}
|
|
if upgraded {
|
|
return nil
|
|
}
|
|
|
|
if _, err := client.User.Update().
|
|
Where(
|
|
dbuser.RoleEQ(service.RoleAdmin),
|
|
dbuser.ConcurrencyEQ(simpleModeLegacyAdminConcurrency),
|
|
).
|
|
SetConcurrency(simpleModeTargetAdminConcurrency).
|
|
Save(ctx); err != nil {
|
|
return fmt.Errorf("upgrade simple mode admin concurrency: %w", err)
|
|
}
|
|
|
|
now := time.Now()
|
|
if err := client.Setting.Create().
|
|
SetKey(simpleModeAdminConcurrencyUpgradeKey).
|
|
SetValue(now.Format(time.RFC3339)).
|
|
SetUpdatedAt(now).
|
|
OnConflictColumns(setting.FieldKey).
|
|
UpdateNewValues().
|
|
Exec(ctx); err != nil {
|
|
return fmt.Errorf("persist admin concurrency upgrade marker: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|