Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- package models
- import (
- _ "github.com/jinzhu/gorm/dialects/postgres"
- "github.com/jinzhu/gorm"
- "os"
- "github.com/joho/godotenv"
- "fmt"
- )
- var db *gorm.DB //database
- func init() {
- e := godotenv.Load() //Load .env file
- if e != nil {
- fmt.Print(e)
- }
- username := os.Getenv("db_user")
- password := os.Getenv("db_pass")
- dbName := os.Getenv("db_name")
- dbHost := os.Getenv("db_host")
- dbUri := fmt.Sprintf("host=%s user=%s dbname=%s sslmode=disable password=%s", dbHost, username, dbName, password) //Build connection string
- fmt.Println(dbUri)
- conn, err := gorm.Open("postgres", dbUri)
- if err != nil {
- fmt.Print(err)
- }
- db = conn
- db.Debug().AutoMigrate(&Account{}, &Contact{}) //Database migration
- }
- //returns a handle to the DB object
- func GetDB() *gorm.DB {
- return db
- }
Add Comment
Please, Sign In to add comment