Reimplementation with gorilla toolkit.

1 files changed, 40 insertions, 0 deletions

diff --git a/db.go b/db.go
new file mode 100644
index 0000000..dc9a15d
--- /dev/null
+++ b/db.go
@@ -0,0 +1,40 @@
+package main
+
+import (
+	"github.com/jinzhu/gorm"
+	_ "github.com/mattn/go-sqlite3"
+	"log"
+	"os"
+)
+
+//var Db gorm.DB
+var dbdriver = os.Getenv("STATUS_DB_DRIVER")
+var dbcred = os.Getenv("STATUS_DB_CREDENTIALS")
+var Db, dberr = gorm.Open(dbdriver, dbcred)
+
+func InitDB() {
+	/*
+		dbdriver := os.Getenv("STATUS_DB_DRIVER")
+		dbcred := os.Getenv("STATUS_DB_CREDENTIALS")
+		Db, err := gorm.Open(dbdriver, dbcred)
+	*/
+	if dberr != nil {
+		log.Panic(dberr)
+	}
+	Db.LogMode(true)
+	if err := Db.DB().Ping(); err != nil {
+		log.Panic(err)
+	}
+
+	//	u := User{}
+	h := Host{}
+	//	Db.Debug().AutoMigrate(&u)
+	db := Db
+	log.Println(db)
+	db.Debug().AutoMigrate(&h)
+
+	/*
+		Db.Model(&u).AddUniqueIndex("idx_user_name", "name")
+		Db.Model(&u).AddUniqueIndex("idx_user_email", "email")
+	*/
+}