128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
package test
|
|
|
|
import (
|
|
//"go-admin/models/tools"
|
|
//"os"
|
|
"fmt"
|
|
"github.com/jinzhu/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/mysql"
|
|
"go-admin/app/admin/models"
|
|
"testing"
|
|
"time"
|
|
//"text/template"
|
|
)
|
|
|
|
func TestGoModelTemplate(t *testing.T) {
|
|
//t1, err := template.ParseFiles("model.go.template")
|
|
//if err != nil {
|
|
// t.Error(err)
|
|
//}
|
|
//table := tools.SysTables{}
|
|
//table.TBName = "sys_tables"
|
|
//tab, err := table.Get()
|
|
//if err != nil {
|
|
// t.Error(err)
|
|
//}
|
|
//file, err := os.Create("models/" + table.PackageName + ".go")
|
|
//if err != nil {
|
|
// t.Error(err)
|
|
//}
|
|
//defer file.Close()
|
|
//
|
|
//_ = t1.Execute(file, tab)
|
|
t.Log("")
|
|
}
|
|
|
|
func TestGoApiTemplate(t *testing.T) {
|
|
//t1, err := template.ParseFiles("api.go.template")
|
|
//if err != nil {
|
|
// t.Error(err)
|
|
//}
|
|
//table := tools.SysTables{}
|
|
//table.TBName = "sys_tables"
|
|
//tab, _ := table.Get()
|
|
//file, err := os.Create("apis/" + table.PackageName + ".go")
|
|
//if err != nil {
|
|
// t.Error(err)
|
|
//}
|
|
//defer file.Close()
|
|
//
|
|
//_ = t1.Execute(file, tab)
|
|
t.Log("")
|
|
}
|
|
|
|
var DBTest *gorm.DB
|
|
|
|
func InitDBTest() {
|
|
var err error
|
|
//dialect := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
|
// "mh_pro", "fLeytnBFCRB553ab", "39.108.188.218", "3306",
|
|
// "mh_pro")
|
|
//site_admin:ZxcZdDjDC3k72xsN@tcp(139.224.216.42:3306)/site_admin?charset=utf8&parseTime=True&loc=Local&timeout=1000ms
|
|
dialect := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
|
"site_admin", "ZxcZdDjDC3k72xsN", "139.224.216.42", "3306",
|
|
"site_admin")
|
|
|
|
DBTest, err = gorm.Open("mysql", dialect)
|
|
if err != nil {
|
|
fmt.Println("Connect Database Error, error = ", err)
|
|
panic(err)
|
|
}
|
|
DBTest.DB().SetMaxOpenConns(1000)
|
|
DBTest.DB().SetMaxIdleConns(500)
|
|
DBTest.DB().SetConnMaxLifetime(time.Second * 10)
|
|
DBTest.LogMode(true)
|
|
DBTest.SingularTable(true)
|
|
DBTest.AutoMigrate(
|
|
//&models.Article{},
|
|
//&models.ArticleCollect{},
|
|
//&models.CannibalizeStockTask{},
|
|
//&models.CannibalizeGameCardGoods{},
|
|
|
|
models.Strategy{},
|
|
)
|
|
|
|
if err := DBTest.DB().Ping(); err != nil {
|
|
fmt.Println("Ping Database Fail, error = ", err)
|
|
panic(err)
|
|
}
|
|
fmt.Println("DB init success")
|
|
}
|
|
|
|
func TestInitDBTest(t *testing.T) {
|
|
InitDBTest()
|
|
}
|
|
|
|
var DBProd *gorm.DB
|
|
|
|
func InitDBProd() {
|
|
var err error
|
|
dialect := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
|
"mh_pro", "fLeytnBFCRB553ab", "39.108.188.218", "3306",
|
|
"mh_pro")
|
|
|
|
DBProd, err = gorm.Open("mysql", dialect)
|
|
if err != nil {
|
|
fmt.Println("Connect Database Error, error = ", err)
|
|
panic(err)
|
|
}
|
|
DBProd.DB().SetMaxOpenConns(1000)
|
|
DBProd.DB().SetMaxIdleConns(500)
|
|
DBProd.DB().SetConnMaxLifetime(time.Second * 10)
|
|
DBProd.LogMode(true)
|
|
DBProd.SingularTable(true)
|
|
DBProd.AutoMigrate(
|
|
models.SysUser{},
|
|
)
|
|
|
|
if err := DBProd.DB().Ping(); err != nil {
|
|
fmt.Println("Ping Database Fail, error = ", err)
|
|
panic(err)
|
|
}
|
|
fmt.Println("DB init success")
|
|
}
|
|
|
|
func TestInitDBProd(t *testing.T) {
|
|
InitDBProd()
|
|
}
|