Use ConfigDefault

This commit is contained in:
hi019
2020-10-31 11:09:14 -04:00
parent f2dcfe31fe
commit 5c07f58b7f

View File

@@ -12,6 +12,8 @@ type Config struct {
// ConfigDefault is the default config // ConfigDefault is the default config
var ConfigDefault = Config{ var ConfigDefault = Config{
GCInterval: 10 * time.Second, GCInterval: 10 * time.Second,
FilePath: "./db.sqlite3",
TableName: "fiber",
} }
// Helper function to set default values // Helper function to set default values
@@ -20,10 +22,10 @@ func configDefault(cfg Config) Config {
cfg.GCInterval = ConfigDefault.GCInterval cfg.GCInterval = ConfigDefault.GCInterval
} }
if cfg.FilePath == "" { if cfg.FilePath == "" {
cfg.FilePath = "./db" cfg.FilePath = ConfigDefault.FilePath
} }
if cfg.TableName == "" { if cfg.TableName == "" {
cfg.TableName = "fiber" cfg.TableName = ConfigDefault.TableName
} }
return cfg return cfg
} }