summaryrefslogtreecommitdiff
path: root/db/connect.go
blob: 2e7123db41ec60a05b9f5e3e8ae6186770fb0043 (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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package db

import (
	"context"
	"database/sql"
	"fmt"
	"log/slog"
	"strconv"

	dbSql "github.com/Wacky404/lurchers/db"
	"github.com/Wacky404/lurchers/util"
)

type Database struct {
	conn     *sql.DB
	ctx      *context.Context
	host     string
	port     int64
	user     string
	password string
	name     string
	sql      *dbSql.DbStatements
}

// creates an instance of the Database struct and loads in env vars
func LoadConfig(ctx *context.Context) (*Database, error) {
	port, err := strconv.ParseInt(util.GetVar("DB_PORT", ""), 10, 64)
	if err != nil {
		slog.Error("error loading .env var port", slog.Any("error", err))
		return nil, err
	}
	db := &Database{
		ctx:      ctx,
		host:     util.GetVar("DB_HOST", ""),
		port:     port,
		user:     util.GetVar("DB_USER", ""),
		password: util.GetVar("DB_PASSWORD", ""),
		name:     util.GetVar("DB_NAME", ""),
		sql:      dbSql.NewDbStatements(),
	}

	return db, nil
}

// connecting to the precious
func (d *Database) Connect() error {
	psqlInfo := fmt.Sprintf("host=%s port=%d user=%s "+
		"password=%s dbname=%s sslmode=disable",
		d.host, d.port, d.user, d.password, d.name)
	conn, err := sql.Open("postgres", psqlInfo)
	if err != nil {
		slog.Error("error connecting to the database", slog.Any("error", err))
	}

	if err := conn.Ping(); err != nil {
		return err
	}

	d.conn = conn
	return nil
}