mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-22 23:54:45 +08:00
fix: 简易模式仅提升管理员默认并发到30
This commit is contained in:
@@ -12,6 +12,7 @@ import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/Wei-Shaw/sub2api/internal/config"
|
||||
"github.com/Wei-Shaw/sub2api/internal/pkg/logger"
|
||||
"github.com/Wei-Shaw/sub2api/internal/repository"
|
||||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||||
@@ -23,10 +24,19 @@ import (
|
||||
|
||||
// Config paths
|
||||
const (
|
||||
ConfigFileName = "config.yaml"
|
||||
InstallLockFile = ".installed"
|
||||
ConfigFileName = "config.yaml"
|
||||
InstallLockFile = ".installed"
|
||||
defaultUserConcurrency = 5
|
||||
simpleModeAdminConcurrency = 30
|
||||
)
|
||||
|
||||
func setupDefaultAdminConcurrency() int {
|
||||
if strings.EqualFold(strings.TrimSpace(os.Getenv("RUN_MODE")), config.RunModeSimple) {
|
||||
return simpleModeAdminConcurrency
|
||||
}
|
||||
return defaultUserConcurrency
|
||||
}
|
||||
|
||||
// GetDataDir returns the data directory for storing config and lock files.
|
||||
// Priority: DATA_DIR env > /app/data (if exists and writable) > current directory
|
||||
func GetDataDir() string {
|
||||
@@ -390,7 +400,7 @@ func createAdminUser(cfg *SetupConfig) (bool, string, error) {
|
||||
Role: service.RoleAdmin,
|
||||
Status: service.StatusActive,
|
||||
Balance: 0,
|
||||
Concurrency: 5,
|
||||
Concurrency: setupDefaultAdminConcurrency(),
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
}
|
||||
@@ -462,7 +472,7 @@ func writeConfigFile(cfg *SetupConfig) error {
|
||||
APIKeyPrefix string `yaml:"api_key_prefix"`
|
||||
RateMultiplier float64 `yaml:"rate_multiplier"`
|
||||
}{
|
||||
UserConcurrency: 5,
|
||||
UserConcurrency: defaultUserConcurrency,
|
||||
UserBalance: 0,
|
||||
APIKeyPrefix: "sk-",
|
||||
RateMultiplier: 1.0,
|
||||
|
||||
Reference in New Issue
Block a user