40 lines
811 B
Go
40 lines
811 B
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type config struct {
|
|
Host string `json:"host"`
|
|
User string `json:"user"`
|
|
Password string `json:"password"`
|
|
Database string `json:"database"`
|
|
}
|
|
|
|
var Database *gorm.DB
|
|
|
|
func getDSN() string {
|
|
user := os.Getenv("PSQL_USER")
|
|
password := os.Getenv("PSQL_PASS")
|
|
database := os.Getenv("PSQL_NAME")
|
|
log.Println("user:", user, "database:", database)
|
|
return fmt.Sprintf("postgresql://%s:%s@%s/%s?sslmode=disable", user, password, "postgres", database)
|
|
}
|
|
|
|
func Connect() {
|
|
var err error
|
|
Database, err = gorm.Open(postgres.Open(getDSN()), new(gorm.Config))
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
err = Database.AutoMigrate(&User{}, &Fraction{}, &Group{}, &ShopAuto{})
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}
|