db.go

package provider

import (
	"github.com/jinzhu/gorm"
	_ "github.com/lib/pq"

	"sample-app/model"
	"sample-app/util"
)

var models = []interface{}{
	&model.Building{},
        &model.BuildingLevel{},
	&model.Quest{},
}

func InitDB() (db *gorm.DB, err error) {
	db, err = gorm.Open("postgres", "user=test password=test dbname=sample_app sslmode=disable")
	if err != nil {
		return
	}

	db.AutoMigrate(models...)
	
	util.Register(models...)

	return
}