blob: 1a5b9ab544e2f43b390e3c2ff8d866b5a31ab765 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
package db
import (
"context"
"database/sql"
"log"
"github.com/jackc/pgx/v5/pgxpool"
"github.com/jackc/pgx/v5/stdlib"
"gitlab.com/alexkavon/newsstand/src/conf"
)
type Db struct {
pool *pgxpool.Pool
}
func NewDb(config *conf.Conf) *Db {
pool, err := pgxpool.New(context.Background(), config.Db.Url)
if err != nil {
log.Fatal(err)
}
var testquery string
err = pool.QueryRow(context.Background(), "select 'Hello, PostgreSQL!'").Scan(&testquery)
if err != nil {
log.Fatal(err)
}
log.Println("Database connection pool created.", testquery)
return &Db{pool}
}
func (d *Db) Conn() *pgxpool.Pool {
return d.pool
}
func (d *Db) ToSqlDb() *sql.DB {
return stdlib.OpenDBFromPool(d.pool)
}
func (d *Db) Close() {
d.pool.Close()
}
|
