From 9a869e800b235636097abebadb1bd1cae4206cbc Mon Sep 17 00:00:00 2001 From: value <z1292839451@163.com> Date: 星期二, 04 六月 2024 21:21:57 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- user-server/src/main/java/com/yuanchu/mom/controller/CompaniesController.java | 41 ++++++++++++++++++++++++++++++++++++++++- 1 files changed, 40 insertions(+), 1 deletions(-) diff --git a/user-server/src/main/java/com/yuanchu/mom/controller/CompaniesController.java b/user-server/src/main/java/com/yuanchu/mom/controller/CompaniesController.java index b1e4805..ca5d2d6 100644 --- a/user-server/src/main/java/com/yuanchu/mom/controller/CompaniesController.java +++ b/user-server/src/main/java/com/yuanchu/mom/controller/CompaniesController.java @@ -1,10 +1,18 @@ package com.yuanchu.mom.controller; +import com.baomidou.mybatisplus.core.toolkit.ObjectUtils; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.yuanchu.mom.annotation.ValueAuth; import com.yuanchu.mom.annotation.ValueClassify; +import com.yuanchu.mom.common.GetLook; import com.yuanchu.mom.dto.PersonDto; +import com.yuanchu.mom.exception.ErrorException; +import com.yuanchu.mom.mapper.PowerMapper; +import com.yuanchu.mom.mapper.UserMapper; import com.yuanchu.mom.pojo.Company; import com.yuanchu.mom.pojo.Person; +import com.yuanchu.mom.pojo.Power; +import com.yuanchu.mom.pojo.User; import com.yuanchu.mom.service.UserService; import com.yuanchu.mom.util.HeaderToken; import com.yuanchu.mom.vo.Result; @@ -13,7 +21,9 @@ import lombok.AllArgsConstructor; import org.springframework.web.bind.annotation.*; +import java.util.ArrayList; import java.util.List; +import java.util.Map; @RestController @AllArgsConstructor @@ -25,11 +35,40 @@ private UserService userService; + UserMapper userMapper; + + PowerMapper powerMapper; + + private GetLook getLook; + + @ValueClassify("鐢ㄦ埛绠$悊") + @PostMapping("/getSampleUser") + @ApiOperation(value = "鐩存帴鑾峰彇涓庤嚜宸辩浉鍚屽崟浣嶇殑浜哄憳鍒楄〃") + public Result<?> getSampleUser(){ + return Result.success(); + } + @ValueClassify("鐢ㄦ埛绠$悊") @ApiOperation(value = "鑾峰彇浜轰簨绯荤粺缁勭粐鏋舵瀯") @GetMapping(value = "/selectCompaniesList") public Result<List<Company>> selectCompaniesList() { - return Result.success(headerToken.companyUrl()); + //鍒ゆ柇鏄惁鏈夌洿鎺ヨ幏鍙栫涓夋柟閮ㄩ棬鐨勬潈闄� + Integer userId = getLook.selectPowerByMethodAndUserId(null).get("userId"); + User user = userMapper.selectById(userId); + Integer roleId = user.getRoleId(); + Power power = powerMapper.selectOne(Wrappers.<Power>lambdaQuery().eq(Power::getRoleId, roleId).eq(Power::getMenuMethod, "getSampleUser")); + if (ObjectUtils.isNotEmpty(power)){ + List<Company> companies = new ArrayList<>(); + Company company = new Company(); + if (ObjectUtils.isEmpty(user.getCompanyId())){ + throw new ErrorException("璇ョ敤鎴锋病鏈夋墍灞炲崟浣�!"); + } + company.setCompanyId(user.getCompanyId()); + companies.add(company); + return Result.success(companies); + }else { + return Result.success(headerToken.companyUrl()); + } } @ValueAuth -- Gitblit v1.9.3