From 598b7528eed13b0cc1f52f97e36d2c0515e56166 Mon Sep 17 00:00:00 2001 From: zss <zss@example.com> Date: 星期二, 04 六月 2024 17:53:32 +0800 Subject: [PATCH] 设备的数采配置+用户获取第三方权限+原始工时导出 --- user-server/src/main/java/com/yuanchu/mom/controller/CompaniesController.java | 91 +++++++++++++++++++++++++++++++++++++-------- 1 files changed, 75 insertions(+), 16 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 73bb8e9..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,37 +1,96 @@ package com.yuanchu.mom.controller; -import cn.hutool.http.HttpRequest; -import cn.hutool.http.HttpResponse; -import cn.hutool.http.HttpUtil; +import com.baomidou.mybatisplus.core.toolkit.ObjectUtils; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.yuanchu.mom.annotation.ValueAuth; -import com.yuanchu.mom.pojo.AuthApi; +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; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; -import io.swagger.annotations.Authorization; import lombok.AllArgsConstructor; -import org.springframework.beans.factory.annotation.Autowired; -import org.springframework.web.bind.annotation.GetMapping; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; +import java.util.ArrayList; import java.util.List; +import java.util.Map; @RestController @AllArgsConstructor @RequestMapping("/companies") -@Api("缁勭粐妯″潡") +@Api(tags="鐢ㄦ埛妯″潡") public class CompaniesController { - @Autowired private HeaderToken headerToken; - @ValueAuth - @ApiOperation(value = "鑾峰彇缁勭粐鍒楄〃") - @GetMapping(value = "/selectCompaniesList") - public Result<List<Company>> selectcompaniesList(){ - return Result.success(headerToken.companyUrl()); + 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() { + //鍒ゆ柇鏄惁鏈夌洿鎺ヨ幏鍙栫涓夋柟閮ㄩ棬鐨勬潈闄� + 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 + @ApiOperation("鑾峰彇浜轰簨绯荤粺缁勭粐涓嬬殑浜哄憳") + @PostMapping(value = "/selectSimpleList") + public Result<?> selectSimpleList(String companyId) { + return Result.success(headerToken.userUrl(companyId)); + } + + @ValueAuth + @ApiOperation("灏嗕汉浜嬬郴缁熷嬀閫夌殑鍐呭杞Щ鍒版湰绯荤粺") + @PostMapping(value = "/addPersonUser") + public Result<?> addPersonUser(@RequestBody PersonDto personDto) { + return Result.success(userService.addPersonUser(personDto)); + } + + @ValueAuth + @ApiOperation("閫氳繃鍛樺伐鍙疯幏鍙栧憳宸�") + @PostMapping(value = "/selectPersonUser") + public Result<?> selectPersonUser(String code) { + Person person = headerToken.selectPersonUser(code); + return Result.success(person); + } + } \ No newline at end of file -- Gitblit v1.9.3