bagas/filekeeper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Separate run command from main function

Browse Source
This commit is contained in:
Bagas Aulia Rezki
2024-04-28 22:26:53 +07:00
parent 5ca6cdd129
commit e6344743d5
12 changed files with 74 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
db/database.go
View File

@ -14,6 +14,7 @@ import (
)
var log *logger.AggregatedLogger
var DB Database
type mySQLdb struct {
*gorm.DB
@ -23,6 +24,13 @@ type postgresDB struct {
*gorm.DB
}
type SSLMode string
const (
DisableSSL SSLMode = "disable"
EnableSSL SSLMode = "enable"
)
type Database interface {
IsUserRegistered(email string, username string) bool
@ -79,9 +87,9 @@ func NewMYSQLdb(username, password, host, port, dbName string) Database {
return &mySQLdb{DB}
}
func NewPostgresDB(username, password, host, port, dbName string) Database {
func NewPostgresDB(username, password, host, port, dbName string, mode SSLMode) Database {
var err error
connection := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=Asia/Jakarta", host, username, password, dbName, port, "disable")
connection := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=Asia/Jakarta", host, username, password, dbName, port, mode)
DB, err := gorm.Open(postgres.New(postgres.Config{
DSN: connection,
}), &gorm.Config{