database/database.go


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

package database

import (
	"dove/messages"
	"dove/utils/logger"

	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

var DB *gorm.DB

func init() {
	databaseDSN := resolveDatabasePath()
	dialector := sqlite.Open(databaseDSN)

	var connectionError error
	DB, connectionError = gorm.Open(dialector, &gorm.Config{
		Logger: resolveGORMLogLevel(),
	})

	if connectionError != nil {
		logger.Fatalf(LOG_PREFIX, messages.DatabaseConnectionFailed, connectionError)
	}

	sqlDB, poolError := DB.DB()
	if poolError != nil {
		logger.Fatalf(LOG_PREFIX, messages.DatabasePoolConfigFailed, poolError)
	}

	sqlDB.SetMaxOpenConns(MAX_OPEN_CONNECTIONS)
	sqlDB.SetMaxIdleConns(MAX_IDLE_CONNECTIONS)
	sqlDB.SetConnMaxLifetime(MAX_CONNECTION_LIFETIME)

	logger.Successf(LOG_PREFIX, messages.DatabaseConnected, databaseDSN)

	migrate()
}