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

package database

import (
	"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(LogPrefix, ConnectionFailed, connectionError)
	}

	sqlDB, poolError := DB.DB()
	if poolError != nil {
		logger.Fatalf(LogPrefix, PoolConfigFailed, poolError)
	}

	sqlDB.SetMaxOpenConns(MaxOpenConnections)
	sqlDB.SetMaxIdleConns(MaxIdleConnections)
	sqlDB.SetConnMaxLifetime(MaxConnectionLifetime)

	logger.Successf(LogPrefix, Connected, databaseDSN)

	migrate()
}