From 45cf408e8cf8d93128467242ccf0a6594596efbd Mon Sep 17 00:00:00 2001
From: wangwei <wangweinj@sfere-elec.com>
Date: Sun, 25 Jul 2021 23:42:42 +0800
Subject: [PATCH] add user and migrate tables

---
 DB/dbConnection.go           |  3 +++
 Data/userProvider.go         | 10 ++++++++++
 DataEntity/tbl_user.go       |  5 +++++
 controller/register.go       |  1 +
 controller/usercontroller.go |  6 ++++++
 handler/userhandler.go       |  4 ++++
 6 files changed, 29 insertions(+)

diff --git a/DB/dbConnection.go b/DB/dbConnection.go
index b32c404..2a68e2f 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 2cef93c..9e20bff 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 ef34469..b3b6393 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 9163259..4883f51 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 fd3412c..5963fd8 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 dcb8bef..b6ed6d0 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)
+}
-- 
Gitee