Moved new postgres implementation to v2 folder, also added nil check for db object in postgres config

This commit is contained in:
Technerder
2021-07-26 15:34:46 -04:00
parent 868c89fd7e
commit 73a941a879
4 changed files with 16 additions and 10 deletions

55
postgres/v2/config.go Normal file
View File

@@ -0,0 +1,55 @@
package v2
import (
"github.com/jackc/pgx/pgxpool"
"time"
)
// Config defines the config for storage.
type Config struct {
// Db pgxpool.Pool object
//
// Required
Db pgxpool.Pool
// Table name
//
// Optional. Default is "fiber_storage"
Table string
// Reset clears any existing keys in existing Table
//
// Optional. Default is false
Reset bool
// Time before deleting expired keys
//
// Optional. Default is 10 * time.Second
GCInterval time.Duration
}
// ConfigDefault is the default config
var ConfigDefault = Config{
Table: "fiber_storage",
Reset: false,
GCInterval: 10 * time.Second,
}
// Helper function to set default values
func configDefault(config ...Config) Config {
// Return default config if nothing provided
if len(config) < 1 {
return ConfigDefault
}
// Override default config
cfg := config[0]
// Set default values
if cfg.Table == "" {
cfg.Table = ConfigDefault.Table
}
if int(cfg.GCInterval.Seconds()) <= 0 {
cfg.GCInterval = ConfigDefault.GCInterval
}
return cfg
}