diff --git a/controller/auth.go b/controller/auth.go index 4f0b02e..39474e1 100644 --- a/controller/auth.go +++ b/controller/auth.go @@ -2,11 +2,12 @@ package controller import ( "github.com/taoshihan1991/imaptool/config" + "github.com/taoshihan1991/imaptool/models" "github.com/taoshihan1991/imaptool/tools" ) func CheckPass(username string, password string) string { - //models.FindUsers() + models.FindUsers() account := config.GetAccount() if account == nil { diff --git a/models/models.go b/models/models.go index a317b20..516e882 100644 --- a/models/models.go +++ b/models/models.go @@ -10,6 +10,12 @@ var DB *gorm.DB func init(){ mysqlInfo:=config.GetMysql() dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=True&loc=Local", mysqlInfo["Username"], mysqlInfo["Password"], mysqlInfo["Server"], mysqlInfo["Port"], mysqlInfo["Database"]) - DB,_:=gorm.Open("mysql",dsn) + DB,_=gorm.Open("mysql",dsn) + DB.SingularTable(true) + DB.LogMode(true) + DB.DB().SetMaxIdleConns(10) + DB.DB().SetMaxOpenConns(100) +} +func CloseDB() { defer DB.Close() } \ No newline at end of file diff --git a/models/users.go b/models/users.go index 04cb80f..db1fff0 100644 --- a/models/users.go +++ b/models/users.go @@ -1,4 +1,5 @@ package models + import ( "github.com/jinzhu/gorm" _ "github.com/jinzhu/gorm/dialects/mysql" @@ -9,7 +10,6 @@ type User struct { Name string Password string } - func CreateUser(name string,password string){ user:=&User{ Name:name,