47 lines
937 B
Go
47 lines
937 B
Go
|
/*
|
||
|
|
* SPDX-FileCopyrightText: 2023 jordi fita mas <jfita@peritasoft.com>
|
||
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
*/
|
||
|
|
|
||
|
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/jackc/pgx/v4"
|
||
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
||
|
|
)
|
||
|
|
|
||
|
|
func New(ctx context.Context, connString string) (*DB, error) {
|
||
|
|
config, err := pgxpool.ParseConfig(connString)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
config.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
|
||
|
|
if _, err := conn.Exec(ctx, "set search_path to camper, public"); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
config.AfterRelease = func(conn *pgx.Conn) bool {
|
||
|
|
if _, err := conn.Exec(context.Background(), "reset role"); err != nil {
|
||
|
|
log.Printf("ERROR - failed to reset role: %v", err)
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
pool, err := pgxpool.ConnectConfig(ctx, config)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &DB{pool}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type DB struct {
|
||
|
|
*pgxpool.Pool
|
||
|
|
}
|