作者 yangfu

用户角色修改

... ... @@ -389,6 +389,9 @@ type(
}
SystemUserAccountUpdateRequest struct{
Id int64 `path:"id"`
Name string `json:"name"` // 姓名
Roles []int64 `json:"roles"` // 权限角色
Status int `json:"enable"` // 状态 1:启用 2:禁用
}
SystemUserAccountUpdateResponse struct{}
... ...
... ... @@ -2,7 +2,9 @@ package role
import (
"context"
"github.com/samber/lo"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/domain"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/pkg/contextdata"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/pkg/xerr"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/api/internal/svc"
... ... @@ -29,13 +31,24 @@ func (l *SystemGetRoleLogic) SystemGetRole(req *types.RoleGetRequest) (resp *typ
var (
conn = l.svcCtx.DefaultDBConn()
role *domain.Role
users []*domain.User
userToken = contextdata.GetUserTokenFromCtx(l.ctx)
)
if role, err = l.svcCtx.RoleRepository.FindOne(l.ctx, conn, req.Id); err != nil {
return nil, xerr.NewErrMsgErr("角色不存在", err)
}
_, users, _ = l.svcCtx.UserRepository.FindByCompanyRoles(l.ctx, conn, userToken.CompanyId, []int64{role.Id}, domain.NewQueryOptions().WithFindOnly())
typesRole := NewTypesRole(role)
lo.ForEach(users, func(item *domain.User, index int) {
typesRole.Users = append(typesRole.Users, types.RoleUser{
Id: item.Id,
Name: item.Name,
})
})
resp = &types.RoleGetResponse{
Role: NewTypesRole(role),
Role: typesRole,
}
for _, auth := range domain.Auths {
resp.AuthList = append(resp.AuthList, types.Auth{
Id: auth.Id,
... ...
... ... @@ -92,12 +92,6 @@ func NewTypesRole(item *domain.Role) types.RoleItem {
role types.RoleItem
users []types.RoleUser = make([]types.RoleUser, 0)
)
lo.ForEach(item.Users, func(item domain.User, index int) {
users = append(users, types.RoleUser{
Id: item.Id,
Name: item.Name,
})
})
role = types.RoleItem{
Id: item.Id,
CompanyId: item.CompanyId,
... ...
... ... @@ -35,6 +35,7 @@ func (l *SystemSearchRoleLogic) SystemSearchRole(req *types.RoleSearchRequest) (
userToken = contextdata.GetUserTokenFromCtx(l.ctx)
roles []*domain.Role
total int64
users []*domain.User
)
queryOptions := domain.IndexCompanyId(userToken.CompanyId)()
if req.Page != 0 && req.Size != 0 {
... ... @@ -57,17 +58,27 @@ func (l *SystemSearchRoleLogic) SystemSearchRole(req *types.RoleSearchRequest) (
})
return
}
roleIds := domain.Values(roles, func(item *domain.Role) int64 {
return item.Id
})
_, users, _ = l.svcCtx.UserRepository.FindByCompanyRoles(l.ctx, conn, userToken.CompanyId, roleIds, domain.NewQueryOptions().WithFindOnly())
lo.ForEach(roles, func(item *domain.Role, index int) {
role := NewTypesRole(item)
role.UsersDesc = usersDesc(item)
usersAll := make([]domain.User, 0)
for _, u := range users {
if lo.Contains(u.Roles, item.Id) {
usersAll = append(usersAll, *u)
}
}
role.UsersDesc = usersDesc(usersAll)
role.AuthsDesc = authsDesc(item)
resp.List = append(resp.List, role)
})
return
}
func usersDesc(item *domain.Role) string {
users := lo.Slice(item.Users, 0, 3)
func usersDesc(usersAll []domain.User) string {
users := lo.Slice(usersAll, 0, 3)
var (
nameList = make([]string, 0)
desc string = " "
... ... @@ -79,7 +90,7 @@ func usersDesc(item *domain.Role) string {
return desc
}
desc = fmt.Sprintf("%s%s", strings.Join(nameList, "、"),
lo.Ternary(len(item.Users) > 3, fmt.Sprintf("...共%d人", len(item.Users)), ""),
lo.Ternary(len(usersAll) > 3, fmt.Sprintf("...共%d人", len(usersAll)), ""),
)
return desc
}
... ...
... ... @@ -67,10 +67,20 @@ func (l *SystemUserAccountSaveLogic) SystemUserAccountSave(req *types.SystemUser
AuditStatus: domain.UserAuditStatusPassed,
}
user.WithName(name)
if user, err = l.svcCtx.UserRepository.Insert(ctx, conn, user); err != nil {
return err
}
for _, roleId := range req.Roles {
if _, err = l.svcCtx.UserRoleRepository.Insert(l.ctx, conn, &domain.UserRole{
CompanyId: user.CompanyId,
UserId: user.Id,
RoleId: roleId,
}); err != nil {
return err
}
user.AddRole(roleId)
}
if user, err = l.svcCtx.UserRepository.Insert(ctx, conn, user); err != nil {
if user, err = l.svcCtx.UserRepository.UpdateWithVersion(ctx, conn, user); err != nil {
return err
}
return nil
... ...
... ... @@ -2,6 +2,11 @@ package user
import (
"context"
"github.com/samber/lo"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/db/transaction"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/domain"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/pkg/contextdata"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/pkg/xerr"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/api/internal/svc"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/api/internal/types"
... ... @@ -24,7 +29,49 @@ func NewSystemUserAccountUpdateLogic(ctx context.Context, svcCtx *svc.ServiceCon
}
func (l *SystemUserAccountUpdateLogic) SystemUserAccountUpdate(req *types.SystemUserAccountUpdateRequest) (resp *types.SystemUserAccountUpdateResponse, err error) {
// todo: add your logic here and delete this line
var (
userToken = contextdata.GetUserTokenFromCtx(l.ctx)
user *domain.User
conn = l.svcCtx.DefaultDBConn()
)
if user, err = l.svcCtx.UserRepository.FindOne(l.ctx, conn, req.Id); err != nil {
return nil, xerr.NewErrMsgErr("用户不存在", err)
}
if user.CompanyId != userToken.CompanyId {
return nil, xerr.NewErrMsgErr("无权限更改", err)
}
if req.Name != user.Name {
user.WithName(req.Name)
}
removeList, addList := lo.Difference(user.Roles, req.Roles)
if len(removeList) > 0 || len(addList) > 0 {
user.Roles = req.Roles
}
user.Enable = req.Status
if err = transaction.UseTrans(l.ctx, l.svcCtx.DB, func(ctx context.Context, conn transaction.Conn) error {
if user, err = l.svcCtx.UserRepository.UpdateWithVersion(ctx, conn, user); err != nil {
return err
}
if len(removeList) > 0 || len(addList) > 0 {
if err = l.svcCtx.UserRoleRepository.DeleteByUser(ctx, conn, user.Id); err != nil {
return err
}
user.Roles = []int64{}
for _, roleId := range req.Roles {
if _, err = l.svcCtx.UserRoleRepository.Insert(l.ctx, conn, &domain.UserRole{
CompanyId: user.CompanyId,
UserId: user.Id,
RoleId: roleId,
}); err != nil {
return err
}
user.AddRole(roleId)
}
}
return nil
}, true); err != nil {
return nil, xerr.NewErrMsgErr("更新用户信息失败", err)
}
resp = &types.SystemUserAccountUpdateResponse{}
return
}
... ...
... ... @@ -39,6 +39,7 @@ type ServiceContext struct {
UserLoveFlagRepository domain.UserLoveFlagRepository
UserReadArticleRepository domain.UserReadArticleRepository
UserRepository domain.UserRepository
UserRoleRepository domain.UserRoleRepository
ApiAuthService authlib.ApiAuthService
SmsService smslib.SMSService
... ... @@ -80,6 +81,7 @@ func NewServiceContext(c config.Config) *ServiceContext {
UserRepository: repository.NewUserRepository(cache.NewCachedRepository(mlCache)),
UserReadArticleRepository: repository.NewUserReadArticleRepository(cache.NewCachedRepository(mlCache)),
ArticleTagRepository: repository.NewArticleTagRepository(cache.NewCachedRepository(mlCache)),
UserRoleRepository: repository.NewUserRoleRepository(cache.NewCachedRepository(mlCache)),
}
}
... ...
... ... @@ -690,6 +690,9 @@ type SystemUserAccountEnableResponse struct {
type SystemUserAccountUpdateRequest struct {
Id int64 `path:"id"`
Name string `json:"name"` // 姓名
Roles []int64 `json:"roles"` // 权限角色
Status int `json:"enable"` // 状态 1:启用 2:禁用
}
type SystemUserAccountUpdateResponse struct {
... ...
syntax = "v1"
info(
title: "xx实例"
desc: "xx实例"
author: "author"
email: "email"
version: "v1"
)
@server(
prefix: user_role/v1
group: user_role
jwt: JwtAuth
)
service Core {
@handler getUserRole
post /user_role/:id (UserRoleGetRequest) returns (UserRoleGetResponse)
@handler saveUserRole
post /user_role (UserRoleSaveRequest) returns (UserRoleSaveResponse)
@handler deleteUserRole
delete /user_role/:id (UserRoleDeleteRequest) returns (UserRoleDeleteResponse)
@handler updateUserRole
put /user_role/:id (UserRoleUpdateRequest) returns (UserRoleUpdateResponse)
@handler searchUserRole
post /user_role/search (UserRoleSearchRequest) returns (UserRoleSearchResponse)
}
type (
UserRoleGetRequest {
Id int64 `path:"id"`
}
UserRoleGetResponse struct{
UserRole UserRoleItem `json:"user_role"`
}
UserRoleSaveRequest struct{
UserRole UserRoleItem `json:"user_role"`
}
UserRoleSaveResponse struct{}
UserRoleDeleteRequest struct{
Id int64 `path:"id"`
}
UserRoleDeleteResponse struct{}
UserRoleUpdateRequest struct{
Id int64 `path:"id"`
UserRole UserRoleItem `json:"user_role"`
}
UserRoleUpdateResponse struct{}
UserRoleSearchRequest struct{
Page int `json:"page"`
Size int `json:"size"`
}
UserRoleSearchResponse{
List []UserRoleItem `json:"list"`
Total int64 `json:"total"`
}
UserRoleItem struct{
}
)
... ...
syntax = "proto3";
option go_package ="./pb";
package pb;
message UserRoleGetReq {
int64 Id = 1;
}
message UserRoleGetResp{
UserRoleItem User = 1;
}
message UserRoleSaveReq {
}
message UserRoleSaveResp{
}
message UserRoleDeleteReq {
int64 Id = 1;
}
message UserRoleDeleteResp{
}
message UserRoleUpdateReq {
int64 Id = 1;
}
message UserRoleUpdateResp{
}
message UserRoleSearchReq {
int64 PageNumber = 1;
int64 PageSize = 2;
}
message UserRoleSearchResp{
repeated UserRoleItem List =1;
int64 Total =2;
}
message UserRoleItem {
}
service UserRoleService {
rpc UserRoleGet(UserRoleGetReq) returns(UserRoleGetResp);
rpc UserRoleSave(UserRoleSaveReq) returns(UserRoleSaveResp);
rpc UserRoleDelete(UserRoleDeleteReq) returns(UserRoleDeleteResp);
rpc UserRoleUpdate(UserRoleUpdateReq) returns(UserRoleUpdateResp);
rpc UserRoleSearch(UserRoleSearchReq) returns(UserRoleSearchResp);
}
... ...
... ... @@ -16,6 +16,7 @@ func Migrate(db *gorm.DB) {
&models.UserLoveFlag{},
&models.UserReadArticle{},
&models.User{},
&models.UserRole{},
&models.Role{},
&models.Company{},
&models.UserFollow{},
... ...
package models
import (
"gorm.io/gorm"
)
type UserRole struct {
Id int64 // 唯一标识
CompanyId int64 // 公司ID
UserId int64
RoleId int64
CreatedAt int64
UpdatedAt int64
DeletedAt int64
Version int
}
func (m *UserRole) TableName() string {
return "user_role"
}
func (m *UserRole) BeforeCreate(tx *gorm.DB) (err error) {
// m.CreatedAt = time.Now().Unix()
// m.UpdatedAt = time.Now().Unix()
return
}
func (m *UserRole) BeforeUpdate(tx *gorm.DB) (err error) {
// m.UpdatedAt = time.Now().Unix()
return
}
func (m *UserRole) CacheKeyFunc() string {
return ""
}
func (m *UserRole) CacheKeyFuncByObject(obj interface{}) string {
if v, ok := obj.(*UserRole); ok {
return v.CacheKeyFunc()
}
return ""
}
func (m *UserRole) CachePrimaryKeyFunc() string {
return ""
}
... ...
... ... @@ -242,6 +242,39 @@ func (repository *UserRepository) FindDepartmentUsers(ctx context.Context, conn
return total, dms, nil
}
func (repository *UserRepository) FindByCompanyRoles(ctx context.Context, conn transaction.Conn, companyId int64, roles []int64, queryOptions map[string]interface{}) (int64, []*domain.User, error) {
var (
tx = conn.DB()
ms []*models.User
dms = make([]*domain.User, 0)
total int64
)
queryFunc := func() (interface{}, error) {
tx = tx.Model(&ms).Order("id asc")
tx.Select("id", "name", "roles")
tx.Where("company_id = ?", companyId)
tx.Where("flag = ?", domain.UserAdmin)
tx.Where("id in (select user_id from user_role where role_id in(?)) ", roles)
if total, tx = transaction.PaginationAndCount(ctx, tx, queryOptions, &ms); tx.Error != nil {
return dms, tx.Error
}
return dms, nil
}
if _, err := repository.Query(queryFunc); err != nil {
return 0, nil, err
}
for _, item := range ms {
if dm, err := repository.ModelToDomainModel(item); err != nil {
return 0, dms, err
} else {
dms = append(dms, dm)
}
}
return total, dms, nil
}
func (repository *UserRepository) FindCompanyPositions(ctx context.Context, conn transaction.Conn, companyId int64, queryOptions map[string]interface{}) (int64, []*domain.User, error) {
var (
tx = conn.DB()
... ...
package repository
import (
"context"
"github.com/jinzhu/copier"
"github.com/pkg/errors"
"github.com/tiptok/gocomm/pkg/cache"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/db/models"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/db/transaction"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/domain"
"gorm.io/gorm"
)
type UserRoleRepository struct {
*cache.CachedRepository
}
func (repository *UserRoleRepository) Insert(ctx context.Context, conn transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
var (
err error
m = &models.UserRole{}
tx = conn.DB()
)
if m, err = repository.DomainModelToModel(dm); err != nil {
return nil, err
}
if tx = tx.Model(m).Save(m); tx.Error != nil {
return nil, tx.Error
}
dm.Id = m.Id
return repository.ModelToDomainModel(m)
}
func (repository *UserRoleRepository) Update(ctx context.Context, conn transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
var (
err error
m *models.UserRole
tx = conn.DB()
)
if m, err = repository.DomainModelToModel(dm); err != nil {
return nil, err
}
queryFunc := func() (interface{}, error) {
tx = tx.Model(m).Updates(m)
return nil, tx.Error
}
if _, err = repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
return nil, err
}
return repository.ModelToDomainModel(m)
}
func (repository *UserRoleRepository) UpdateWithVersion(ctx context.Context, transaction transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
var (
err error
m *models.UserRole
tx = transaction.DB()
)
if m, err = repository.DomainModelToModel(dm); err != nil {
return nil, err
}
oldVersion := dm.Version
m.Version += 1
queryFunc := func() (interface{}, error) {
tx = tx.Model(m).Select("*").Where("id = ?", m.Id).Where("version = ?", oldVersion).Updates(m)
if tx.RowsAffected == 0 {
return nil, domain.ErrUpdateFail
}
return nil, tx.Error
}
if _, err = repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
return nil, err
}
return repository.ModelToDomainModel(m)
}
func (repository *UserRoleRepository) Delete(ctx context.Context, conn transaction.Conn, dm *domain.UserRole) (*domain.UserRole, error) {
var (
tx = conn.DB()
m = &models.UserRole{Id: dm.Identify().(int64)}
)
queryFunc := func() (interface{}, error) {
tx = tx.Where("id = ?", m.Id).Delete(m)
return m, tx.Error
}
if _, err := repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
return dm, err
}
return repository.ModelToDomainModel(m)
}
func (repository *UserRoleRepository) DeleteByUser(ctx context.Context, conn transaction.Conn, userId int64) error {
var (
tx = conn.DB()
m = &models.UserRole{}
)
queryFunc := func() (interface{}, error) {
tx = tx.Where("user_id = ?", userId).Delete(m)
return m, tx.Error
}
if _, err := repository.Query(queryFunc, m.CacheKeyFunc()); err != nil {
return err
}
return nil
}
func (repository *UserRoleRepository) FindOne(ctx context.Context, conn transaction.Conn, id int64) (*domain.UserRole, error) {
var (
err error
tx = conn.DB()
m = new(models.UserRole)
)
queryFunc := func() (interface{}, error) {
tx = tx.Model(m).Where("id = ?", id).First(m)
if errors.Is(tx.Error, gorm.ErrRecordNotFound) {
return nil, domain.ErrNotFound
}
return m, tx.Error
}
cacheModel := new(models.UserRole)
cacheModel.Id = id
if err = repository.QueryCache(cacheModel.CacheKeyFunc, m, queryFunc); err != nil {
return nil, err
}
return repository.ModelToDomainModel(m)
}
func (repository *UserRoleRepository) Find(ctx context.Context, conn transaction.Conn, queryOptions map[string]interface{}) (int64, []*domain.UserRole, error) {
var (
tx = conn.DB()
ms []*models.UserRole
dms = make([]*domain.UserRole, 0)
total int64
)
queryFunc := func() (interface{}, error) {
tx = tx.Model(&ms).Order("id desc")
if total, tx = transaction.PaginationAndCount(ctx, tx, queryOptions, &ms); tx.Error != nil {
return dms, tx.Error
}
return dms, nil
}
if _, err := repository.Query(queryFunc); err != nil {
return 0, nil, err
}
for _, item := range ms {
if dm, err := repository.ModelToDomainModel(item); err != nil {
return 0, dms, err
} else {
dms = append(dms, dm)
}
}
return total, dms, nil
}
func (repository *UserRoleRepository) ModelToDomainModel(from *models.UserRole) (*domain.UserRole, error) {
to := &domain.UserRole{}
err := copier.Copy(to, from)
return to, err
}
func (repository *UserRoleRepository) DomainModelToModel(from *domain.UserRole) (*models.UserRole, error) {
to := &models.UserRole{}
err := copier.Copy(to, from)
return to, err
}
func NewUserRoleRepository(cache *cache.CachedRepository) domain.UserRoleRepository {
return &UserRoleRepository{CachedRepository: cache}
}
... ...
... ... @@ -39,6 +39,7 @@ type UserRepository interface {
FindOne(ctx context.Context, conn transaction.Conn, id int64) (*User, error)
FindOneByCompanyIdAndPhone(ctx context.Context, conn transaction.Conn, companyId int64, phone string, status []int) (*User, error)
Find(ctx context.Context, conn transaction.Conn, queryOptions map[string]interface{}) (int64, []*User, error)
FindByCompanyRoles(ctx context.Context, conn transaction.Conn, companyId int64, roles []int64, queryOptions map[string]interface{}) (int64, []*User, error)
FindDepartmentUsers(ctx context.Context, conn transaction.Conn, companyId int64, queryOptions map[string]interface{}) (int64, []*User, error)
FindCompanyPositions(ctx context.Context, conn transaction.Conn, companyId int64, queryOptions map[string]interface{}) (int64, []*User, error)
}
... ...
package domain
import (
"context"
"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/interanl/pkg/db/transaction"
)
type UserRole struct {
Id int64 // 唯一标识
CompanyId int64 // 公司ID
UserId int64
RoleId int64
CreatedAt int64
UpdatedAt int64
DeletedAt int64
Version int
}
type UserRoleRepository interface {
Insert(ctx context.Context, conn transaction.Conn, dm *UserRole) (*UserRole, error)
Update(ctx context.Context, conn transaction.Conn, dm *UserRole) (*UserRole, error)
UpdateWithVersion(ctx context.Context, conn transaction.Conn, dm *UserRole) (*UserRole, error)
Delete(ctx context.Context, conn transaction.Conn, dm *UserRole) (*UserRole, error)
DeleteByUser(ctx context.Context, conn transaction.Conn, userId int64) error
FindOne(ctx context.Context, conn transaction.Conn, id int64) (*UserRole, error)
Find(ctx context.Context, conn transaction.Conn, queryOptions map[string]interface{}) (int64, []*UserRole, error)
}
func (m *UserRole) Identify() interface{} {
if m.Id == 0 {
return nil
}
return m.Id
}
... ...
CREATE TABLE `comment` (
CREATE TABLE `department` (
`id` int(0) NOT NULL COMMENT '唯一标识',
PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Dynamic;
CREATE TABLE `department` (
CREATE TABLE `role` (
`id` int(0) NOT NULL COMMENT '唯一标识',
PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Dynamic;
CREATE TABLE `role` (
CREATE TABLE `user` (
`id` int(0) NOT NULL COMMENT '唯一标识',
PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Dynamic;
CREATE TABLE `user` (
CREATE TABLE `user_role` (
`id` int(0) NOT NULL COMMENT '唯一标识',
PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci ROW_FORMAT = Dynamic;
... ...