/
database.go
63 lines (51 loc) · 1.57 KB
/
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package database
import (
"database/sql"
"github.com/hendrorahmat/golang-clean-architecture/src/infrastructure/config"
"github.com/hendrorahmat/golang-clean-architecture/src/infrastructure/constants"
"github.com/hendrorahmat/golang-clean-architecture/src/infrastructure/persistance/database/mysql"
"github.com/hendrorahmat/golang-clean-architecture/src/infrastructure/persistance/database/postgres"
"github.com/sirupsen/logrus"
"gorm.io/gorm"
"sync"
)
type IDB interface {
SqlDB() *sql.DB
DB() *gorm.DB
GetDsn() string
}
type Connection map[string]IDB
type Connections struct {
Connection
}
var connections *Connections
var dbConnOnce sync.Once
func MakeDatabase(databases config.Databases, log *logrus.Logger) *Connections {
dbConnOnce.Do(func() {
listConnections := Connection{}
var dbConnection IDB
for name, databaseConf := range databases {
if databaseConf.SkipCreateConnection {
continue
}
databaseConf.ConnectionName = string(name)
switch databaseConf.Driver {
case constants.POSTGRES:
pgCon := postgres.NewPostgresDB()
dbConnection = pgCon.NewConnection(databaseConf, log)
case constants.MYSQL:
mysqlCon := mysql.NewMysqlDB()
dbConnection = mysqlCon.NewConnection(databaseConf, log)
}
listConnections[string(name)] = dbConnection
}
listConnections[constants.ActiveConnectionDb] = listConnections[constants.DefaultConnectionDB]
connections = &Connections{
listConnections,
}
})
return connections
}
func (c *Connections) GetConnection(connectionName string) *gorm.DB {
return c.Connection[connectionName].DB()
}