1
0
mirror of https://github.com/pocketbase/pocketbase.git synced 2025-03-17 21:27:52 +02:00

initialize a default queryTimeout

This commit is contained in:
Gani Georgiev 2024-10-05 22:00:24 +03:00
parent 3a579d16ca
commit bc67835de9

View File

@ -49,13 +49,13 @@ type Config struct {
DefaultDev bool
DefaultDataDir string // if not set, it will fallback to "./pb_data"
DefaultEncryptionEnv string
DefaultQueryTimeout time.Duration // default to core.DefaultQueryTimeout (in seconds)
// optional DB configurations
DataMaxOpenConns int // default to core.DefaultDataMaxOpenConns
DataMaxIdleConns int // default to core.DefaultDataMaxIdleConns
AuxMaxOpenConns int // default to core.DefaultAuxMaxOpenConns
AuxMaxIdleConns int // default to core.DefaultAuxMaxIdleConns
QueryTimeout int // default to core.DefaultQueryTimeout (in seconds)
DBConnect core.DBConnectFunc // default to core.dbConnect
}
@ -89,6 +89,10 @@ func NewWithConfig(config Config) *PocketBase {
config.DefaultDataDir = filepath.Join(baseDir, "pb_data")
}
if config.DefaultQueryTimeout == 0 {
config.DefaultQueryTimeout = core.DefaultQueryTimeout
}
executableName := filepath.Base(os.Args[0])
pb := &PocketBase{
@ -122,11 +126,11 @@ func NewWithConfig(config Config) *PocketBase {
IsDev: pb.devFlag,
DataDir: pb.dataDirFlag,
EncryptionEnv: pb.encryptionEnvFlag,
QueryTimeout: time.Duration(pb.queryTimeout) * time.Second,
DataMaxOpenConns: config.DataMaxOpenConns,
DataMaxIdleConns: config.DataMaxIdleConns,
AuxMaxOpenConns: config.AuxMaxOpenConns,
AuxMaxIdleConns: config.AuxMaxIdleConns,
QueryTimeout: time.Duration(config.QueryTimeout) * time.Second,
DBConnect: config.DBConnect,
})
@ -214,7 +218,7 @@ func (pb *PocketBase) eagerParseFlags(config *Config) error {
pb.RootCmd.PersistentFlags().IntVar(
&pb.queryTimeout,
"queryTimeout",
int(core.DefaultQueryTimeout.Seconds()),
int(config.DefaultQueryTimeout.Seconds()),
"the default SELECT queries timeout in seconds",
)