diff --git a/DB/dbConnection.go b/DB/dbConnection.go index b32c40451d99f60bc1258bb0920d8bed78bc3ef6..2a68e2f5143871d8f417a8f879f660d0e641348b 100644 --- a/DB/dbConnection.go +++ b/DB/dbConnection.go @@ -13,6 +13,7 @@ package db import ( "fmt" + dataentity "go-iris-1/DataEntity" "github.com/jinzhu/gorm" _ "github.com/jinzhu/gorm/dialects/mysql" @@ -38,6 +39,8 @@ func NewInstance() *gorm.DB { } db, err := gorm.Open("mysql", fmt.Sprintf("%s:%s@(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", configData.Username, configData.Pwd, configData.DbServer, configData.DBName)) + db.AutoMigrate(&dataentity.Tbl_ac{}) + db.SingularTable(true) if err != nil { panic(err) diff --git a/Data/userProvider.go b/Data/userProvider.go index 2cef93cc1e55648c2e10fbb8efd12b98eb5c2d7f..9e20bff9e697b967107badf5dcc09969cc5c7360 100644 --- a/Data/userProvider.go +++ b/Data/userProvider.go @@ -26,3 +26,13 @@ func GetUserInfo(userId string) *dataentity.Tbl_user { } return &tbl_user } + +func AddUserInfo(userId string, userName string) { + db := db.NewInstance() + defer db.Close() + user := &dataentity.Tbl_user{ + ID: userId, + UserName: userName, + } + db.Create(user) +} diff --git a/DataEntity/tbl_user.go b/DataEntity/tbl_user.go index ef34469322af9204087fcaaf3d49213164b46cf6..b3b63935f1041c3428ada26d4481df7c8512a58e 100644 --- a/DataEntity/tbl_user.go +++ b/DataEntity/tbl_user.go @@ -15,3 +15,8 @@ type Tbl_user struct { ID string `gorm:"column:id;primary_key"` UserName string `gorm:"column:name"` } + +type Tbl_ac struct { + ID string `gorm:"column:id;primary_key"` + UserName string `gorm:"column:name"` +} diff --git a/controller/register.go b/controller/register.go index 91632597eaa2de5e4d1afd793c0ad2f3a6acfd4b..4883f514f92f3ad52a200e6f48f4c0e916e6864f 100644 --- a/controller/register.go +++ b/controller/register.go @@ -19,5 +19,6 @@ func Register(app *iris.Application) { userController := app.Party("/user") { userController.Get("/getUserInfo", GetUserInfo) + userController.Get("/addUser", addUser) } } diff --git a/controller/usercontroller.go b/controller/usercontroller.go index fd3412c47a1680a423197ee9bde7dd8db7e91741..5963fd8dece943f9668cb9e9f4251b7f339afcad 100644 --- a/controller/usercontroller.go +++ b/controller/usercontroller.go @@ -22,3 +22,9 @@ func GetUserInfo(ctx iris.Context) { user := handler.GetUserInfo(userId) ctx.JSON(*user) } + +func addUser(ctx iris.Context) { + userId := ctx.URLParam("userId") + userName := ctx.URLParam("userName") + handler.AddUserInfo(userId, userName) +} diff --git a/handler/userhandler.go b/handler/userhandler.go index dcb8befc49846a2fa1c8094c2af8ad98c132c5ae..b6ed6d09be0e4a4cbe94f93b53d4b94fe84cdb2f 100644 --- a/handler/userhandler.go +++ b/handler/userhandler.go @@ -20,3 +20,7 @@ func GetUserInfo(userId string) *dataentity.Tbl_user { user := data.GetUserInfo(userId) return user } + +func AddUserInfo(userId string, userName string) { + data.AddUserInfo(userId, userName) +}