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 |   50 +++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 49 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 08729b4..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,9 +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;
@@ -12,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
@@ -24,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
@@ -45,4 +85,12 @@
         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