system_user_search_logic.go 2.0 KB
package user

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"
	"strings"

	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/api/internal/svc"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-discuss/cmd/discuss/api/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type SystemUserSearchLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewSystemUserSearchLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SystemUserSearchLogic {
	return &SystemUserSearchLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *SystemUserSearchLogic) SystemUserSearch(req *types.SystemUserSearchRequest) (resp *types.SystemUserSearchResponse, err error) {
	var (
		conn      = l.svcCtx.DefaultDBConn()
		users     []*domain.User
		userToken = contextdata.GetUserTokenFromCtx(l.ctx)
		total     int64
		roleMap   = make(map[int64]*domain.Role)
	)
	queryOptions := domain.IndexCompanyId(userToken.CompanyId)().WithOffsetLimit(req.Page, req.Size).WithOrder("id desc").
		WithKV("likeName", req.Name).
		WithKV("likePhone", req.Phone).
		WithKV("likePosition", req.Position).
		WithKV("enable", req.Enable).
		WithKV("departmentId", req.DepartmentId)
	if total, users, err = l.svcCtx.UserRepository.Find(l.ctx, conn, queryOptions); err != nil {
		return nil, xerr.NewErr(err)
	}
	resp = &types.SystemUserSearchResponse{
		Total: total,
		List:  make([]types.SystemUser, 0),
	}
	lo.ForEach(users, func(item *domain.User, index int) {
		var roles []string
		for _, id := range item.Roles {
			if role, _ := domain.LazyLoad(roleMap, l.ctx, conn, id, l.svcCtx.RoleRepository.FindOne); role != nil {
				roles = append(roles, role.Name)
			}
		}
		user := NewSystemUser(item)
		user.RolesDesc = strings.Join(roles, "、")
		resp.List = append(resp.List, user)
	})

	return
}