|
@@ -2,10 +2,14 @@ package api
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
+ "github.com/dgrijalva/jwt-go"
|
|
|
"github.com/gin-gonic/gin"
|
|
|
- "main/controller/support"
|
|
|
+ "main/controller/servers"
|
|
|
+ "main/middleware"
|
|
|
"main/model/dbModel"
|
|
|
+ "main/model/modelInterface"
|
|
|
"mime/multipart"
|
|
|
+ "time"
|
|
|
)
|
|
|
|
|
|
var (
|
|
@@ -18,12 +22,12 @@ type RegistAndLoginStuct struct {
|
|
|
PassWord string `json:"passWord"`
|
|
|
}
|
|
|
|
|
|
-
|
|
|
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
-
|
|
|
+
|
|
|
func Regist(c *gin.Context) {
|
|
|
var R RegistAndLoginStuct
|
|
|
_ = c.BindJSON(&R)
|
|
@@ -31,30 +35,54 @@ func Regist(c *gin.Context) {
|
|
|
U := &dbModel.User{UserName: R.UserName, PassWord: R.PassWord}
|
|
|
err, user := U.Regist()
|
|
|
if err != nil {
|
|
|
- ReportFormat(c, false, fmt.Sprintf("%v", err), gin.H{
|
|
|
+ servers.ReportFormat(c, false, fmt.Sprintf("%v", err), gin.H{
|
|
|
"user": user,
|
|
|
})
|
|
|
} else {
|
|
|
- ReportFormat(c, false, "创建成功", gin.H{
|
|
|
+ servers.ReportFormat(c, false, "创建成功", gin.H{
|
|
|
"user": user,
|
|
|
})
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-
|
|
|
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
-
|
|
|
+
|
|
|
func Login(c *gin.Context) {
|
|
|
var L RegistAndLoginStuct
|
|
|
_ = c.BindJSON(&L)
|
|
|
U := &dbModel.User{UserName: L.UserName, PassWord: L.PassWord}
|
|
|
if err, user := U.Login(); err != nil {
|
|
|
- ReportFormat(c, false, "用户名密码错误", gin.H{"user": user})
|
|
|
+ servers.ReportFormat(c, false, "用户名密码错误", gin.H{"user": user})
|
|
|
+ } else {
|
|
|
+ tokenNext(c, *user)
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+func tokenNext(c *gin.Context, user dbModel.User) {
|
|
|
+ j := &middleware.JWT{
|
|
|
+ []byte("qmPlus"),
|
|
|
+ }
|
|
|
+ clams := middleware.CustomClaims{
|
|
|
+ UUID: user.UUID,
|
|
|
+ ID: user.ID,
|
|
|
+ NickName: user.NickName,
|
|
|
+ AuthorityId: user.AuthorityId,
|
|
|
+ StandardClaims: jwt.StandardClaims{
|
|
|
+ NotBefore: int64(time.Now().Unix() - 1000),
|
|
|
+ ExpiresAt: int64(time.Now().Unix() + 3600*7),
|
|
|
+ Issuer: "qmPlus",
|
|
|
+ },
|
|
|
+ }
|
|
|
+ token, err := j.CreateToken(clams)
|
|
|
+ if err != nil {
|
|
|
+ servers.ReportFormat(c, false, "获取token失败", gin.H{})
|
|
|
} else {
|
|
|
- ReportFormat(c, true, "登录成功", gin.H{"user": user})
|
|
|
+ servers.ReportFormat(c, true, "登录成功", gin.H{"user": user, "token": token})
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -66,6 +94,7 @@ type ChangePassWordStutrc struct {
|
|
|
|
|
|
|
|
|
|
|
|
+
|
|
|
|
|
|
|
|
|
|
|
@@ -75,9 +104,9 @@ func ChangePassWord(c *gin.Context) {
|
|
|
_ = c.BindJSON(¶ms)
|
|
|
U := &dbModel.User{UserName: params.UserName, PassWord: params.PassWord}
|
|
|
if err, _ := U.ChangePassWord(params.NewPassWord); err != nil {
|
|
|
- ReportFormat(c, false, "修改失败,请检查用户名密码", gin.H{})
|
|
|
+ servers.ReportFormat(c, false, "修改失败,请检查用户名密码", gin.H{})
|
|
|
} else {
|
|
|
- ReportFormat(c, true, "修改成功", gin.H{})
|
|
|
+ servers.ReportFormat(c, true, "修改成功", gin.H{})
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -87,6 +116,7 @@ type UserHeaderImg struct {
|
|
|
|
|
|
|
|
|
|
|
|
+
|
|
|
|
|
|
|
|
|
|
|
@@ -94,26 +124,54 @@ type UserHeaderImg struct {
|
|
|
|
|
|
|
|
|
func UploadHeaderImg(c *gin.Context) {
|
|
|
+ claims, _ := c.Get("claims")
|
|
|
|
|
|
+
|
|
|
+ waitUse := claims.(*middleware.CustomClaims)
|
|
|
+ fmt.Println(waitUse.NickName)
|
|
|
_, header, err := c.Request.FormFile("headerImg")
|
|
|
|
|
|
userName := c.PostForm("userName")
|
|
|
if err != nil {
|
|
|
- ReportFormat(c, false, fmt.Sprintf("上传文件失败,%v", err), gin.H{})
|
|
|
+ servers.ReportFormat(c, false, fmt.Sprintf("上传文件失败,%v", err), gin.H{})
|
|
|
} else {
|
|
|
|
|
|
- err, filePath := support.Upload(header, USER_HEADER_BUCKET, USER_HEADER_IMG_PATH)
|
|
|
+ err, filePath := servers.Upload(header, USER_HEADER_BUCKET, USER_HEADER_IMG_PATH)
|
|
|
if err != nil {
|
|
|
- ReportFormat(c, false, fmt.Sprintf("接收返回值失败,%v", err), gin.H{})
|
|
|
+ servers.ReportFormat(c, false, fmt.Sprintf("接收返回值失败,%v", err), gin.H{})
|
|
|
} else {
|
|
|
|
|
|
err, user := new(dbModel.User).UploadHeaderImg(userName, filePath)
|
|
|
|
|
|
if err != nil {
|
|
|
- ReportFormat(c, false, fmt.Sprintf("修改数据库链接失败,%v", err), gin.H{})
|
|
|
+ servers.ReportFormat(c, false, fmt.Sprintf("修改数据库链接失败,%v", err), gin.H{})
|
|
|
} else {
|
|
|
- ReportFormat(c, true, "上传成功", gin.H{"user": user})
|
|
|
+ servers.ReportFormat(c, true, "上传成功", gin.H{"user": user})
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+func GetInfoList(c *gin.Context) {
|
|
|
+ var pageInfo modelInterface.PageInfo
|
|
|
+ _ = c.BindJSON(&pageInfo)
|
|
|
+ err, list, total := new(dbModel.User).GetInfoList(pageInfo)
|
|
|
+ if err != nil {
|
|
|
+ servers.ReportFormat(c, false, fmt.Sprintf("获取数据失败,%v", err), gin.H{})
|
|
|
+ } else {
|
|
|
+ servers.ReportFormat(c, true, "获取数据成功", gin.H{
|
|
|
+ "userList": list,
|
|
|
+ "total": total,
|
|
|
+ "page": pageInfo.Page,
|
|
|
+ "pageSize": pageInfo.PageSize,
|
|
|
+ })
|
|
|
+ }
|
|
|
+}
|