Init: Create & Init Project...

database/database.go

@@ -0,0 +1,36 @@
+package database
+
+import (
+	"fmt"
+
+	"github.com/jinzhu/gorm"
+	_ "github.com/jinzhu/gorm/dialects/mysql" //MySQL
+)
+
+// Db gorm
+var Db *gorm.DB
+
+//Init database
+func Init() error {
+	config := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=%t&loc=%s",
+		"nvwa",
+		"nvwa",
+		"server.yumc.pw",
+		"nvwa",
+		true,
+		"Local")
+	db, err := gorm.Open("mysql", config)
+	db.Set("gorm:table_options", "ENGINE=InnoDB")
+	Db = db
+	return err
+}
+
+// Create insert the value into database
+func Create(value interface{}) *gorm.DB {
+	return Db.Create(value)
+}
+
+//Close database
+func Close() {
+	Db.Close()
+}