37 lines
591 B
Go
37 lines
591 B
Go
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()
|
|
}
|