39 lines
799 B
Go
39 lines
799 B
Go
|
package model
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gorm.io/driver/postgres"
|
||
|
|
"gorm.io/driver/sqlite"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
type dbHandler struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func newSqliteHandler(path string) (DBHandler, error) {
|
||
|
|
db, err := gorm.Open(sqlite.Open(path), &gorm.Config{})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
db.AutoMigrate(&Device{})
|
||
|
|
db.AutoMigrate(&Controller{})
|
||
|
|
db.AutoMigrate(&Service{})
|
||
|
|
|
||
|
|
return &dbHandler{db}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func newPostgresqlHandler(path string) (DBHandler, error) {
|
||
|
|
dsn := "host=localhost user=user password=user_password dbname=godopudb port=5432 sslmode=disable TimeZone=Asia/Seoul"
|
||
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
db.AutoMigrate(&Device{})
|
||
|
|
db.AutoMigrate(&Controller{})
|
||
|
|
db.AutoMigrate(&Service{})
|
||
|
|
|
||
|
|
return &dbHandler{db}, nil
|
||
|
|
}
|