package org.jeecg.modules.weekly.controller;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.shiro.SecurityUtils;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.aspect.annotation.AutoLog;
|
import org.jeecg.common.system.base.controller.JeecgController;
|
import org.jeecg.common.system.query.QueryGenerator;
|
import org.jeecg.common.system.vo.LoginUser;
|
import org.jeecg.modules.weekly.entity.WekUserRelation;
|
import org.jeecg.modules.weekly.feign.SystemClient;
|
import org.jeecg.modules.weekly.feign.model.SysUser;
|
import org.jeecg.modules.weekly.service.IWekUserRelationService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.servlet.ModelAndView;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* @Description: wek_user_relation
|
* @Author: jeecg-boot
|
* @Date: 2024-01-09
|
* @Version: V1.0
|
*/
|
@Api(tags = "wek_user_relation")
|
@RestController
|
@RequestMapping("/wek/relation")
|
@Slf4j
|
public class WekUserRelationController extends JeecgController<WekUserRelation, IWekUserRelationService> {
|
@Autowired
|
private IWekUserRelationService wekUserRelationService;
|
@Autowired
|
private SystemClient systemClient;
|
|
/**
|
* 分页列表查询
|
*
|
* @param wekUserRelation
|
* @param pageNo
|
* @param pageSize
|
* @param req
|
* @return
|
*/
|
@AutoLog(value = "wek_user_relation-分页列表查询")
|
@ApiOperation(value = "wek_user_relation-分页列表查询", notes = "wek_user_relation-分页列表查询")
|
@GetMapping(value = "/list")
|
public Result<?> queryPageList(WekUserRelation wekUserRelation,
|
@RequestParam(name = "pageNo", defaultValue = "1") Integer pageNo,
|
@RequestParam(name = "pageSize", defaultValue = "10") Integer pageSize,
|
HttpServletRequest req) {
|
if (StringUtils.isEmpty(wekUserRelation.getUser())) {
|
return Result.error("用户不能为空!");
|
}
|
QueryWrapper<WekUserRelation> queryWrapper = QueryGenerator.initQueryWrapper(wekUserRelation, req.getParameterMap());
|
Page<WekUserRelation> page = new Page<WekUserRelation>(pageNo, pageSize);
|
IPage<WekUserRelation> pageList = wekUserRelationService.page(page, queryWrapper);
|
return Result.OK(pageList);
|
}
|
|
/**
|
* 添加
|
*
|
* @param wekUserRelation
|
* @return
|
*/
|
@AutoLog(value = "wek_user_relation-添加")
|
@ApiOperation(value = "wek_user_relation-添加", notes = "wek_user_relation-添加")
|
@PostMapping(value = "/add")
|
public Result<?> add(@RequestBody WekUserRelation wekUserRelation) {
|
if (StringUtils.isEmpty(wekUserRelation.getUser()) || StringUtils.isEmpty(wekUserRelation.getSuperiors())) {
|
return Result.error("用户或上级不能为空!");
|
}
|
QueryWrapper<WekUserRelation> queryWrapper = new QueryWrapper<>();
|
queryWrapper.lambda().eq(WekUserRelation::getUser, wekUserRelation.getUser());
|
queryWrapper.lambda().eq(WekUserRelation::getSuperiors, wekUserRelation.getSuperiors());
|
List<WekUserRelation> list = wekUserRelationService.list(queryWrapper);
|
if (list.size() < 1) {
|
wekUserRelationService.save(wekUserRelation);
|
} else {
|
return Result.error("上级已存在！");
|
}
|
|
return Result.OK("添加成功！");
|
}
|
|
/**
|
* 编辑
|
*
|
* @param wekUserRelation
|
* @return
|
*/
|
@AutoLog(value = "wek_user_relation-编辑")
|
@ApiOperation(value = "wek_user_relation-编辑", notes = "wek_user_relation-编辑")
|
@PutMapping(value = "/edit")
|
public Result<?> edit(@RequestBody WekUserRelation wekUserRelation) {
|
wekUserRelationService.updateById(wekUserRelation);
|
return Result.OK("编辑成功!");
|
}
|
|
/**
|
* 通过id删除
|
*
|
* @param id
|
* @return
|
*/
|
@AutoLog(value = "wek_user_relation-通过id删除")
|
@ApiOperation(value = "wek_user_relation-通过id删除", notes = "wek_user_relation-通过id删除")
|
@DeleteMapping(value = "/delete")
|
public Result<?> delete(@RequestParam(name = "id", required = true) String id) {
|
wekUserRelationService.removeById(id);
|
return Result.OK("删除成功!");
|
}
|
|
/**
|
* 批量删除
|
*
|
* @param ids
|
* @return
|
*/
|
@AutoLog(value = "wek_user_relation-批量删除")
|
@ApiOperation(value = "wek_user_relation-批量删除", notes = "wek_user_relation-批量删除")
|
@DeleteMapping(value = "/deleteBatch")
|
public Result<?> deleteBatch(@RequestParam(name = "ids", required = true) String ids) {
|
this.wekUserRelationService.removeByIds(Arrays.asList(ids.split(",")));
|
return Result.OK("批量删除成功!");
|
}
|
|
/**
|
* 通过id查询
|
*
|
* @param id
|
* @return
|
*/
|
@AutoLog(value = "wek_user_relation-通过id查询")
|
@ApiOperation(value = "wek_user_relation-通过id查询", notes = "wek_user_relation-通过id查询")
|
@GetMapping(value = "/queryById")
|
public Result<?> queryById(@RequestParam(name = "id", required = true) String id) {
|
WekUserRelation wekUserRelation = wekUserRelationService.getById(id);
|
if (wekUserRelation == null) {
|
return Result.error("未找到对应数据");
|
}
|
return Result.OK(wekUserRelation);
|
}
|
|
|
@AutoLog(value = "wek_user_relation-查询下级")
|
@ApiOperation(value = "wek_user_relation-查询下级", notes = "wek_user_relation-查询下级")
|
@GetMapping(value = "/queryJunior")
|
public Result<?> queryJunior(@RequestParam(name = "username", required = false) String username) {
|
//查询登录用户的下级
|
LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();
|
QueryWrapper<WekUserRelation> queryWrapper = new QueryWrapper<>();
|
queryWrapper.lambda().eq(WekUserRelation::getSuperiors, sysUser.getUsername());
|
List<WekUserRelation> list = wekUserRelationService.list(queryWrapper);
|
|
|
//如果登录用户是部门上级 1-普通用户 2-上级
|
Integer userIdentity = sysUser.getUserIdentity();
|
|
List<Map<String, Object>> root = new ArrayList<>();
|
List<Map<String, Object>> gList1 = new ArrayList<>();
|
List<Map<String, Object>> gList2 = new ArrayList<>();
|
Map<String, Object> group1 = new HashMap<>();
|
group1.put("title", "直属下级");
|
group1.put("children", gList1);
|
group1.put("type", 1);
|
|
Map<String, Object> group2 = new HashMap<>();
|
group2.put("title", "其他下级");
|
group2.put("children", gList2);
|
group2.put("type", 1);
|
|
Map<String, Object> gl1 = new HashMap<>();
|
gl1.put("username", "gl1");
|
gl1.put("realname", "概览");
|
gl1.put("type", 1);
|
gList1.add(gl1);
|
Map<String, Object> gl2 = new HashMap<>();
|
gl2.put("username", "gl2");
|
gl2.put("realname", "概览");
|
gl2.put("type", 1);
|
gList2.add(gl2);
|
|
list.forEach(i -> {
|
Map<String, Object> item = new HashMap<>();
|
item.put("username", i.getUser());
|
item.put("realname", i.getUserName());
|
item.put("type", 2);
|
gList1.add(item);
|
|
QueryWrapper<WekUserRelation> wrapper = new QueryWrapper<>();
|
wrapper.lambda().eq(WekUserRelation::getSuperiors, i.getUser());
|
List<WekUserRelation> subList = wekUserRelationService.list(wrapper);
|
subList.forEach(sub -> {
|
Map<String, Object> s = new HashMap<>();
|
s.put("username", sub.getUser());
|
s.put("realname", sub.getUserName());
|
s.put("type", 2);
|
gList2.add(s);
|
});
|
});
|
|
//如果是上级需要把部门所有人员添加为下级
|
if (userIdentity != null && userIdentity == 2) {
|
Set<String> namesToRemove = gList1.stream()
|
.map(map -> (String) map.get("username"))
|
.collect(Collectors.toSet());
|
List<SysUser> depUserList = systemClient.queryUserByDepCode(sysUser.getOrgCode()).getResult();
|
if (depUserList != null) {
|
depUserList = depUserList.stream().filter(i -> i.getUserIdentity() == null || (i.getUserIdentity()!=null && i.getUserIdentity() == 1)).collect(Collectors.toList());
|
List<Map<String, Object>> oList = new ArrayList<>();
|
depUserList.forEach(user -> {
|
Map<String, Object> o = new HashMap<>();
|
o.put("username", user.getUsername());
|
o.put("realname", user.getRealname());
|
o.put("type", 2);
|
oList.add(o);
|
});
|
// 移除 name 值相同的元素
|
oList.removeIf(map -> namesToRemove.contains(map.get("username")));
|
group2.put("children", oList);
|
}
|
|
}
|
|
|
if (list.size() < 1) {
|
return Result.error("未找到对应数据");
|
}
|
root.add(group1);
|
root.add(group2);
|
return Result.OK(root);
|
}
|
|
/**
|
* 导出excel
|
*
|
* @param request
|
* @param wekUserRelation
|
*/
|
@RequestMapping(value = "/exportXls")
|
public ModelAndView exportXls(HttpServletRequest request, WekUserRelation wekUserRelation) {
|
return super.exportXls(request, wekUserRelation, WekUserRelation.class, "wek_user_relation");
|
}
|
|
/**
|
* 通过excel导入数据
|
*
|
* @param request
|
* @param response
|
* @return
|
*/
|
@RequestMapping(value = "/importExcel", method = RequestMethod.POST)
|
public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
|
return super.importExcel(request, response, WekUserRelation.class);
|
}
|
|
}
|
