hpds_jkw_web/internal/service/index.go

622 lines
17 KiB
Go

package service
import (
"context"
"encoding/json"
"fmt"
"git.hpds.cc/Component/logging"
"hpds-iot-web/config"
"hpds-iot-web/internal/middleware"
"hpds-iot-web/internal/proto"
"hpds-iot-web/model"
"hpds-iot-web/pkg/utils"
"net/http"
"time"
"xorm.io/xorm"
)
type PagingStruct struct {
List interface{} `json:"list"`
Total int64 `json:"total"`
}
// FillPaging 填充分页数据
func FillPaging(count, pageNum, pageSize int64, list interface{}, data *proto.BaseResponse) *proto.BaseResponse {
//var tp int64
//if count%pageSize > 0 {
// tp = count/pageSize + 1
//} else {
// tp = count / pageSize
//}
_ = fmt.Sprintf("%d, %d", pageNum, pageSize)
ps := new(PagingStruct)
ps.List = list
ps.Total = count
data.Data = ps
//data.PageSize = pageSize
//data.Data = list
//data.Page = pageNum
//data.PageCount = tp
//data.TotalSize = count
return data
}
type repo struct {
AppConfig *config.WebConfig
engine *xorm.Engine
logger *logging.Logger
}
type UserService interface {
Login(ctx context.Context, userName, pass string) (rsp *proto.BaseResponse, err error)
GetUserInfo(ctx context.Context, userId int64) (rsp *proto.BaseResponse, err error)
MenuList(ctx context.Context, userId int64) (rsp *proto.BaseResponse, err error)
GetUserList(ctx context.Context, req proto.UserRequest) (rsp *proto.BaseResponse, err error)
AddUser(ctx context.Context, req proto.UserRequestItem) (rsp *proto.BaseResponse, err error)
EditUser(ctx context.Context, req proto.UserRequestItem) (rsp *proto.BaseResponse, err error)
GetRoleList(ctx context.Context, req proto.RoleRequest) (rsp *proto.BaseResponse, err error)
AddRole(ctx context.Context, req proto.RoleItemRequest) (rsp *proto.BaseResponse, err error)
EditRole(ctx context.Context, req proto.RoleItemRequest) (rsp *proto.BaseResponse, err error)
}
func NewUserService(engine *xorm.Engine, logger *logging.Logger) UserService {
return &repo{
engine: engine,
logger: logger,
}
}
func (rp *repo) Login(ctx context.Context, userName, pass string) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
//获取用户
us := new(model.SystemUser)
h, err := rp.engine.Where("phone=?", userName).Get(us)
if err != nil {
goto ReturnPoint
}
if !h {
err = fmt.Errorf("未能找到对应的用户")
goto ReturnPoint
}
vPass := utils.GetUserSha1Pass(pass, us.Salt)
if vPass != us.Pass {
err = fmt.Errorf("用户名或者密码不正确")
goto ReturnPoint
}
data := new(proto.UserLoginResponse)
data.UserName = us.Phone
data.RealName = us.RealName
data.UserId = us.UserId
data.Avatar = us.Avatar
data.Desc = us.Desc
data.HomePath = us.HomePath
//生成JWT token
data.Token, err = middleware.Sign(us, 86400)
if err != nil {
goto ReturnPoint
}
body, _ := json.Marshal(us)
err = model.Redis.Set(data.Token, string(body), time.Duration(24)*time.Hour).Err()
if err != nil {
goto ReturnPoint
}
roleList := model.GetRolesByUserId(us.UserId)
data.Roles = make([]proto.RoleItem, len(roleList))
for k, v := range roleList {
data.Roles[k] = proto.RoleItem{
RoleId: v.RoleId,
RoleName: v.RoleName,
Value: v.RoleValue,
}
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp.Data = data
rsp.Err = err
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) GetUserInfo(ctx context.Context, userId int64) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
//获取用户
us := new(model.SystemUser)
h, err := rp.engine.ID(userId).Get(us)
if err != nil {
goto ReturnPoint
}
if !h {
err = fmt.Errorf("未能找到对应的用户")
goto ReturnPoint
}
data := new(proto.UserLoginResponse)
data.UserName = us.Phone
data.RealName = us.RealName
data.UserId = us.UserId
data.Avatar = us.Avatar
data.Desc = us.Desc
data.HomePath = us.HomePath
//生成JWT token
data.Token, err = middleware.Sign(us, 86400)
if err != nil {
goto ReturnPoint
}
body, _ := json.Marshal(us)
err = model.Redis.Set(data.Token, string(body), time.Duration(24)*time.Hour).Err()
if err != nil {
goto ReturnPoint
}
roleList := model.GetRolesByUserId(us.UserId)
data.Roles = make([]proto.RoleItem, len(roleList))
for k, v := range roleList {
data.Roles[k] = proto.RoleItem{
RoleId: v.RoleId,
RoleName: v.RoleName,
Value: v.RoleValue,
}
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp.Data = data
rsp.Err = err
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) MenuList(ctx context.Context, userId int64) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
//获取用户
roleList := make([]model.SystemUserRole, 0)
err = rp.engine.Where("user_id = ? and status = 1", userId).Find(&roleList)
if err != nil {
goto ReturnPoint
}
roleIdList := make([]int64, len(roleList))
for k, v := range roleList {
roleIdList[k] = v.RoleId
}
menuList := make([]model.SystemRoleMenu, 0)
err = rp.engine.In("role_id", roleIdList).And("status = 1").Find(&menuList)
if err != nil {
goto ReturnPoint
}
menuIdList := make([]int64, len(menuList))
for k, v := range menuList {
menuIdList[k] = v.RoleId
}
list := make([]model.SystemMenu, 0)
err = rp.engine.In("menu_id", menuIdList).And("status = 1").Find(&list)
if err != nil {
goto ReturnPoint
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp.Data = list
rsp.Err = err
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) GetUserList(ctx context.Context, req proto.UserRequest) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
userList := make([]model.SystemUser, 0)
count, err := rp.engine.Where("(? = '' or phone like ?)", req.Phone, "%"+req.Phone+"%").
And("(? = '' or real_name like ?)", req.RealName, req.RealName).
And("status = 1").Limit(int(req.Size), int(((req.Page)-1)*req.Size)).
FindAndCount(&userList)
if err != nil {
goto ReturnPoint
}
data := make([]proto.UserLoginResponse, len(userList))
for k, v := range userList {
roleList := model.GetRolesByUserId(v.UserId)
roles := make([]proto.RoleItem, len(roleList))
for key, val := range roleList {
roles[key] = proto.RoleItem{
RoleId: val.RoleId,
RoleName: val.RoleName,
Value: val.RoleValue,
}
}
data[k] = proto.UserLoginResponse{
UserId: v.UserId,
RealName: v.RealName,
UserName: v.Phone,
Roles: roles,
Avatar: v.Avatar,
Desc: v.Desc,
HomePath: v.HomePath,
}
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp = FillPaging(count, req.Page, req.Size, data, rsp)
rsp.Err = err
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) AddUser(ctx context.Context, req proto.UserRequestItem) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
item := new(model.SystemUser)
b, err := rp.engine.Where("phone = ?)", req.Phone).
Get(item)
if err != nil {
goto ReturnPoint
}
if !b {
sess := rp.engine.NewSession()
item.Status = req.Status
item.Phone = req.Phone
item.RealName = req.RealName
//item.RoleId = req.RoleId
item.Desc = req.Desc
item.CreateAt = time.Now().Unix()
item.UpdateAt = time.Now().Unix()
item.Salt = utils.RandomString(16, 3)
item.Pass = utils.GetUserSha1Pass(req.Password, item.Salt)
_, err = sess.Insert(item)
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
if len(req.RoleId) > 0 {
for _, v := range req.RoleId {
role := new(model.SystemUserRole)
role.RoleId = v
role.UserId = item.UserId
role.Status = 1
_, err = sess.Insert(role)
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
}
}
err = sess.Commit()
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
} else {
err = fmt.Errorf("该手机号已经被注册")
goto ReturnPoint
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp.Err = ctx.Err()
rsp.Data = item
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) EditUser(ctx context.Context, req proto.UserRequestItem) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
item := new(model.SystemUser)
b, err := rp.engine.Where("phone = ?", req.Phone).
Get(item)
if err != nil {
goto ReturnPoint
}
if !b {
err = fmt.Errorf("未能找到对应的用户")
goto ReturnPoint
} else {
sess := rp.engine.NewSession()
item.Status = req.Status
if len(item.Phone) > 0 {
item.Phone = req.Phone
}
if len(req.RealName) > 0 {
item.RealName = req.RealName
}
if len(req.Desc) > 0 {
item.Desc = req.Desc
}
if len(req.Password) > 0 {
item.Salt = utils.RandomString(16, 3)
item.Pass = utils.GetUserSha1Pass(req.Password, item.Salt)
}
item.UpdateAt = time.Now().Unix()
_, err = sess.AllCols().Update(item)
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
role := new(model.SystemUserRole)
role.UserId = item.UserId
role.Status = 0
_, err = sess.Where("user_id = ?", item.UserId).AllCols().Update(role)
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
if len(req.RoleId) > 0 {
for _, v := range req.RoleId {
role := new(model.SystemUserRole)
b, err = sess.Where("user_id = ?", item.UserId).And("role_id=?", v).Get(role)
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
if b {
role.Status = 1
_, err = sess.ID(role.UserRoleId).Update(role)
} else {
role.RoleId = v
role.UserId = item.UserId
role.Status = 1
_, err = sess.Insert(role)
}
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
}
}
err = sess.Commit()
if err != nil {
_ = sess.Rollback()
goto ReturnPoint
}
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp.Err = ctx.Err()
rsp.Data = item
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) GetRoleList(ctx context.Context, req proto.RoleRequest) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
data := make([]model.SystemRole, 0)
count, err := rp.engine.Where("(? = '' or role_name like ?)", req.RoleName, "%"+req.RoleName+"%").
And("(? = 0 or status = ?)", req.Status, req.Status).Limit(int(req.Size), int(((req.Page)-1)*req.Size)).
FindAndCount(&data)
if err != nil {
goto ReturnPoint
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp = FillPaging(count, req.Page, req.Size, data, rsp)
rsp.Err = err
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) AddRole(ctx context.Context, req proto.RoleItemRequest) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
item := new(model.SystemRole)
b, err := rp.engine.Where("role_value = ?", req.RoleValue).
Get(item)
if err != nil {
goto ReturnPoint
}
if b {
err = fmt.Errorf("已经存在相同值的角色")
goto ReturnPoint
} else {
item.AliasName = req.AliasName
item.Description = req.Description
item.RoleName = req.RoleName
item.RoleValue = req.RoleValue
item.Status = req.Status
item.CreateAt = time.Now().Unix()
item.UpdateAt = time.Now().Unix()
_, err = rp.engine.Insert(item)
if err != nil {
goto ReturnPoint
}
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp.Err = ctx.Err()
rsp.Data = item
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}
func (rp *repo) EditRole(ctx context.Context, req proto.RoleItemRequest) (rsp *proto.BaseResponse, err error) {
rsp = new(proto.BaseResponse)
select {
case <-ctx.Done():
err = fmt.Errorf("超时/取消")
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Message = "超时/取消"
rsp.Err = ctx.Err()
return rsp, ctx.Err()
default:
item := new(model.SystemRole)
b, err := rp.engine.Where("role_id = ?", req.RoleId).
Get(item)
if err != nil {
goto ReturnPoint
}
if !b {
err = fmt.Errorf("未能找到对应的用户")
goto ReturnPoint
} else {
if len(req.AliasName) > 0 && item.AliasName != req.AliasName {
item.AliasName = req.AliasName
}
if len(req.Description) > 0 && item.Description != req.Description {
item.Description = req.Description
}
if len(req.RoleName) > 0 && item.RoleName != req.RoleName {
item.RoleName = req.RoleName
}
if len(req.RoleValue) > 0 && item.RoleValue != req.RoleValue {
item.RoleValue = req.RoleValue
}
item.Status = req.Status
item.UpdateAt = time.Now().Unix()
_, err = rp.engine.AllCols().Update(item)
if err != nil {
goto ReturnPoint
}
}
rsp.Code = http.StatusOK
rsp.Status = http.StatusText(http.StatusOK)
rsp.Message = "成功"
rsp.Err = ctx.Err()
rsp.Data = item
return rsp, err
}
ReturnPoint:
if err != nil {
rsp.Code = http.StatusInternalServerError
rsp.Status = http.StatusText(http.StatusInternalServerError)
rsp.Err = err
rsp.Message = "失败"
}
return rsp, err
}