package org.jeecg.modules.doc.controller;
|
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
import java.io.IOException;
|
import java.io.UnsupportedEncodingException;
|
import java.net.URLDecoder;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.system.query.QueryGenerator;
|
import org.jeecg.common.util.oConvertUtils;
|
import org.jeecg.modules.doc.api.JeecgSystemClient;
|
import org.jeecg.modules.doc.entity.PathPermission;
|
import org.jeecg.modules.doc.entity.UserStorageFlow;
|
import org.jeecg.modules.doc.service.IUserStorageFlowService;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.jeecg.modules.doc.vo.PathPermissionVO;
|
import org.jeecgframework.poi.excel.ExcelImportUtil;
|
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
|
import org.jeecgframework.poi.excel.entity.ExportParams;
|
import org.jeecgframework.poi.excel.entity.ImportParams;
|
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
|
import org.jeecg.common.system.base.controller.JeecgController;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
import org.springframework.web.servlet.ModelAndView;
|
import com.alibaba.fastjson.JSON;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.jeecg.common.aspect.annotation.AutoLog;
|
|
/**
|
* @Description: user_storage_flow
|
* @Author: jeecg-boot
|
* @Date: 2022-08-04
|
* @Version: V1.0
|
*/
|
@Api(tags="user_storage_flow")
|
@RestController
|
@RequestMapping("/userStorageFlow")
|
@Slf4j
|
public class UserStorageFlowController extends JeecgController<UserStorageFlow, IUserStorageFlowService> {
|
@Autowired
|
private IUserStorageFlowService userStorageFlowService;
|
|
@Autowired
|
private JeecgSystemClient jeecgSystemClient;
|
|
/**
|
* 分页列表查询
|
*
|
* @param userStorageFlow
|
* @param pageNo
|
* @param pageSize
|
* @param req
|
* @return
|
*/
|
@AutoLog(value = "user_storage_flow-分页列表查询")
|
@ApiOperation(value="user_storage_flow-分页列表查询", notes="user_storage_flow-分页列表查询")
|
@GetMapping(value = "/list")
|
public Result<?> queryPageList(UserStorageFlow userStorageFlow,
|
@RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
|
@RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
|
HttpServletRequest req) {
|
QueryWrapper<UserStorageFlow> queryWrapper = QueryGenerator.initQueryWrapper(userStorageFlow, req.getParameterMap());
|
Page<UserStorageFlow> page = new Page<UserStorageFlow>(pageNo, pageSize);
|
IPage<UserStorageFlow> pageList = userStorageFlowService.page(page, queryWrapper);
|
return Result.OK(pageList);
|
}
|
|
/**
|
* 添加
|
*
|
* @param userStorageFlow
|
* @return
|
*/
|
@AutoLog(value = "user_storage_flow-添加")
|
@ApiOperation(value="user_storage_flow-添加", notes="user_storage_flow-添加")
|
@PostMapping(value = "/add")
|
public Result<?> add(@RequestBody UserStorageFlow userStorageFlow) {
|
userStorageFlowService.save(userStorageFlow);
|
return Result.OK("添加成功！");
|
}
|
|
/**
|
* 编辑
|
*
|
* @param userStorageFlow
|
* @return
|
*/
|
@AutoLog(value = "user_storage_flow-编辑")
|
@ApiOperation(value="user_storage_flow-编辑", notes="user_storage_flow-编辑")
|
@PutMapping(value = "/edit")
|
public Result<?> edit(@RequestBody UserStorageFlow userStorageFlow) {
|
userStorageFlowService.updateById(userStorageFlow);
|
return Result.OK("编辑成功!");
|
}
|
|
/**
|
* 通过id删除
|
*
|
* @param id
|
* @return
|
*/
|
@AutoLog(value = "user_storage_flow-通过id删除")
|
@ApiOperation(value="user_storage_flow-通过id删除", notes="user_storage_flow-通过id删除")
|
@DeleteMapping(value = "/delete")
|
public Result<?> delete(@RequestParam(name="id",required=true) String id) {
|
userStorageFlowService.removeById(id);
|
return Result.OK("删除成功!");
|
}
|
|
/**
|
* 批量删除
|
*
|
* @param ids
|
* @return
|
*/
|
@AutoLog(value = "user_storage_flow-批量删除")
|
@ApiOperation(value="user_storage_flow-批量删除", notes="user_storage_flow-批量删除")
|
@DeleteMapping(value = "/deleteBatch")
|
public Result<?> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
|
this.userStorageFlowService.removeByIds(Arrays.asList(ids.split(",")));
|
return Result.OK("批量删除成功!");
|
}
|
|
/**
|
* 通过id查询
|
*
|
* @param id
|
* @return
|
*/
|
@AutoLog(value = "user_storage_flow-通过id查询")
|
@ApiOperation(value="user_storage_flow-通过id查询", notes="user_storage_flow-通过id查询")
|
@GetMapping(value = "/queryById")
|
public Result<?> queryById(@RequestParam(name="id",required=true) String id) {
|
UserStorageFlow userStorageFlow = userStorageFlowService.getById(id);
|
if(userStorageFlow==null) {
|
return Result.error("未找到对应数据");
|
}
|
return Result.OK(userStorageFlow);
|
}
|
|
/**
|
* 导出excel
|
*
|
* @param request
|
* @param userStorageFlow
|
*/
|
@RequestMapping(value = "/exportXls")
|
public ModelAndView exportXls(HttpServletRequest request, UserStorageFlow userStorageFlow) {
|
return super.exportXls(request, userStorageFlow, UserStorageFlow.class, "user_storage_flow");
|
}
|
|
/**
|
* 通过excel导入数据
|
*
|
* @param request
|
* @param response
|
* @return
|
*/
|
@RequestMapping(value = "/importExcel", method = RequestMethod.POST)
|
public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
|
return super.importExcel(request, response, UserStorageFlow.class);
|
}
|
|
|
|
@ApiOperation(value = "保存用户下载限额分配")
|
@PostMapping("/save")
|
public Result<?> savePathPerm(@RequestBody UserStorageFlow storageFlow) {
|
try{
|
storageFlow.setFlow(storageFlow.getFlow()*1024*1024);
|
userStorageFlowService.saveOrUpdate(storageFlow);
|
return Result.OK();
|
}catch (Exception e) {
|
return Result.error("权限设置失败！");
|
}
|
|
}
|
|
|
/**
|
* 查询所有角色和用户的下载限额
|
* 先查询出所有角色及角色对应的用户列表，将下载限额设置情况合并到列表，返回给前端展示。
|
* @return Result<List<PathPermissionVO>>
|
*/
|
@ApiOperation(value="查询角色用户下载限额", notes="查询角色用户下载限额")
|
@GetMapping("/queryAll")
|
public Result<List<PathPermissionVO>> queryAllPathPermission() {
|
// 查询所有已设置下载限额
|
LambdaQueryWrapper<UserStorageFlow> queryWrapper = new LambdaQueryWrapper<>();
|
|
List<UserStorageFlow> list = userStorageFlowService.list(queryWrapper);
|
// 角色/用户对应的下载限额转换成MAP,方便查询
|
Map<String, UserStorageFlow> permMap = list.stream()
|
.collect(Collectors.toMap(path -> path.getRoleId()==null?path.getUserId():path.getRoleId(), pp -> pp));
|
// 调用system的接口查询角色和用户
|
List<PathPermissionVO> vos = jeecgSystemClient.queryRoleUsers();
|
// 将查到的文件权限绑定到对应角色，给前台展示
|
if (vos != null && vos.size() > 0) {
|
vos.stream().forEach( r -> {
|
|
convertVo(r, r.getRoleId(), permMap);
|
});
|
}
|
return Result.OK(vos);
|
}
|
|
|
/**
|
* 将权限实体转换成vo对象
|
* @param r
|
* @param permMap
|
*/
|
private void convertVo(PathPermissionVO r,String key, Map<String, UserStorageFlow> permMap) {
|
UserStorageFlow p = permMap.get(key);
|
if (p != null) {
|
r.setId(p.getId());
|
if (p.getFlow() != null) {
|
r.setFlow(p.getFlow() / 1024 / 1024);
|
}
|
|
}
|
r.setKey(key);
|
if (r.getChildren() != null) {
|
r.getChildren().stream().forEach( u -> {
|
|
u.setChildren(null);
|
convertVo(u, u.getUserId(), permMap);
|
});
|
}
|
|
}
|
}
|
