/* * Copyright (c) 2018-2025, ztt All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * Neither the name of the pig4cloud.com developer nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * Author: ztt */ package com.chinaztt.mes.basic.controller; import com.alibaba.excel.EasyExcel; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.chinaztt.mes.basic.entity.Warehouse; import com.chinaztt.mes.basic.excel.WarehouseData; import com.chinaztt.mes.basic.excel.WarehouseUploadListener; import com.chinaztt.mes.basic.service.WarehouseService; import com.chinaztt.mes.common.wrapper.QueryWrapperUtil; import com.chinaztt.ztt.common.core.util.R; import com.chinaztt.ztt.common.log.annotation.SysLog; import com.chinaztt.ztt.common.security.annotation.Inner; import com.chinaztt.ztt.common.security.service.ZttUser; import com.chinaztt.ztt.common.security.util.SecurityUtils; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import lombok.AllArgsConstructor; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.IOException; /** * 仓库基础数据 * * @author sunxl * @date 2020-09-18 09:32:01 */ @RestController @AllArgsConstructor @RequestMapping("/warehouse") @Api(value = "warehouse", tags = "仓库基础数据管理") public class WarehouseController { private final WarehouseService warehouseService; /** * 分页查询 * * @param page 分页对象 * @param warehouse 仓库基础数据 * @return */ @ApiOperation(value = "分页查询", notes = "分页查询") @GetMapping("/page") @PreAuthorize("@pms.hasPermission('basic_warehouse_view')") public R getWarehousePage(Page page, Warehouse warehouse) { return R.ok(warehouseService.getWarehouse(page, QueryWrapperUtil.gen(warehouse))); } /** * 通过id查询仓库基础数据 * * @param id id * @return R */ @ApiOperation(value = "通过id查询", notes = "通过id查询") @GetMapping("/{id}") @PreAuthorize("@pms.hasPermission('basic_warehouse_view')") public R getById(@PathVariable("id") Long id) { return R.ok(warehouseService.getById(id)); } /** * 新增仓库基础数据 * * @param warehouse 仓库基础数据 * @return R */ @ApiOperation(value = "新增仓库基础数据", notes = "新增仓库基础数据") @SysLog("新增仓库基础数据") @PostMapping @PreAuthorize("@pms.hasPermission('basic_warehouse_add')") public R save(@RequestBody Warehouse warehouse) { Warehouse warehouseByFind = warehouseService.getOne(Wrappers.lambdaQuery().eq(Warehouse::getWarehouseNo, warehouse.getWarehouseNo())); if (warehouseByFind != null) { throw new RuntimeException("仓库编号重复:" + warehouseByFind.getWarehouseName() + "-" + warehouseByFind.getWarehouseNo()); } return R.ok(warehouseService.save(warehouse)); } /** * excel上传 * * @return */ @PostMapping("/uploadExcel") public R upExcelLoad(@RequestParam("file") MultipartFile file) { ZttUser currentUser = SecurityUtils.getUser(); try { EasyExcel.read(file.getInputStream(), WarehouseData.class, new WarehouseUploadListener(warehouseService)).sheet().doRead(); } catch (IOException e) { e.printStackTrace(); } return R.ok(); } /** * 上传文件 文件名采用uuid,避免原始文件名中带"-"符号导致下载的时候解析出现异常 * * @param file 资源 * @return R(/ admin / bucketName / filename) */ @PostMapping("/upload") public R upload(@RequestParam("file") MultipartFile file) { ZttUser currentUser = SecurityUtils.getUser(); System.out.println(currentUser); return warehouseService.uploadFile(file); } /** * 修改仓库基础数据 * * @param warehouse 仓库基础数据 * @return R */ @ApiOperation(value = "修改仓库基础数据", notes = "修改仓库基础数据") @SysLog("修改仓库基础数据") @PutMapping @PreAuthorize("@pms.hasPermission('basic_warehouse_edit')") public R updateById(@RequestBody Warehouse warehouse) { return R.ok(warehouseService.updateById(warehouse)); } /** * 通过id删除仓库基础数据 * * @param id id * @return R */ @ApiOperation(value = "通过id删除仓库基础数据", notes = "通过id删除仓库基础数据") @SysLog("通过id删除仓库基础数据") @DeleteMapping("/{id}") @PreAuthorize("@pms.hasPermission('basic_warehouse_del')") public void removeById(@PathVariable Long id) { warehouseService.deleteWarehouse(id); } /** * @param fileName * @return */ @ApiOperation(value = "通过url删除附件", notes = "通过url删除附件") @SysLog("通过url删除附件") @DeleteMapping("/removeFile") public R removeFileById(String fileName) { return warehouseService.deleteFile(fileName); } /** * 获取文件 * * @param bucket 桶名称 * @param url 文件空间/名称 * @param response * @return */ @Inner(false) @GetMapping("/{bucket}/{url}") public void file(@PathVariable String bucket, @PathVariable String url, HttpServletResponse response) { warehouseService.getFile(bucket, url, response); } /** * 通过id删除图片 * * @param id id * @return R */ @ApiOperation(value = "通过id删除仓库基础数据", notes = "通过id删除仓库基础数据") @SysLog("通过id删除仓库基础数据") @DeleteMapping("/file/{id}") public void removeByFile(@PathVariable Long id) { warehouseService.deleteByFile(id); } }