Marco Pedone
1ac2d4b3ab
- Updated `fs.go` and `minio.go` to replace hardcoded storage paths with values from the new config package. - Introduced a new `config` package to manage application configuration, including database and storage settings. - Removed the `models.go` file from `typedefs` as it was no longer needed. - Added a new `config.go` file to handle loading environment variables and application settings. - Created a new `types.go` file in `typedefs` to define storage strategies. - Updated `storage_handler.go` to utilize the new configuration structure.
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
pgxuuid "github.com/jackc/pgx-gofrs-uuid"
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
type DbHandler struct {
|
|
Dbpool *pgxpool.Pool
|
|
Ctx context.Context
|
|
}
|
|
|
|
type Configs struct {
|
|
Host string
|
|
Dbname string
|
|
Port string
|
|
User string
|
|
Password string
|
|
}
|
|
|
|
func NewDbHandler(configs Configs) *DbHandler {
|
|
fmt.Println("Connecting to db")
|
|
ctx := context.Background()
|
|
|
|
if configs.Host == "" || configs.Dbname == "" || configs.Port == "" || configs.User == "" || configs.Password == "" {
|
|
fmt.Println("Environment variables for database connection are not set")
|
|
return nil
|
|
}
|
|
|
|
connStr := fmt.Sprintf("postgresql://%s:%s@%s:%s/%s?sslmode=disable", configs.User, configs.Password, configs.Host, configs.Port, configs.Dbname)
|
|
fmt.Printf("Db connection string: %s\n", connStr)
|
|
config, err := pgxpool.ParseConfig(connStr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
config.ConnConfig.Tracer = NewMultiQueryTracer(NewLoggingQueryTracer(slog.Default()))
|
|
|
|
config.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
|
|
pgxuuid.Register(conn.TypeMap())
|
|
return nil
|
|
}
|
|
dbpool, err := pgxpool.NewWithConfig(ctx, config)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Println("Connected to db")
|
|
|
|
return &DbHandler{Dbpool: dbpool, Ctx: ctx}
|
|
}
|