package org.springblade.modules.equipage.controller;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import io.swagger.annotations.ApiOperation;
|
import io.swagger.annotations.ApiParam;
|
import lombok.AllArgsConstructor;
|
import org.apache.commons.codec.Charsets;
|
import org.springblade.common.excel.RowWriteHandler;
|
import org.springblade.common.utils.arg;
|
import org.springblade.core.excel.util.ExcelUtil;
|
import org.springblade.core.mp.support.Condition;
|
import org.springblade.core.mp.support.Query;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.core.tool.utils.DateUtil;
|
import org.springblade.core.tool.utils.Func;
|
import org.springblade.modules.FTP.FtpUtil;
|
import org.springblade.modules.equipage.entity.Gun;
|
import org.springblade.modules.equipage.excel.GunExcel;
|
import org.springblade.modules.equipage.excel.GunImporter;
|
import org.springblade.modules.equipage.service.GunService;
|
import org.springblade.modules.equipage.vo.GunVo;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.net.URLEncoder;
|
import java.text.SimpleDateFormat;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
/**
|
* @author zhongrj
|
* @time 2021-07-06
|
* @desc 枪支控制层
|
*/
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/gun")
|
public class GunController {
|
|
private final GunService gunService;
|
|
/**
|
* 自定义分页
|
* @param query page,size
|
* @param gun 枪支信息对象
|
*/
|
@GetMapping("/page")
|
public R<IPage<GunVo>> page(GunVo gun, Query query) {
|
IPage<GunVo> pages = gunService.selectGunPage(Condition.getPage(query), gun);
|
return R.data(pages);
|
}
|
|
/**
|
* 新增
|
* @param gun 枪支信息对象
|
*/
|
@PostMapping("/save")
|
@ApiOperation(value = "新增", notes = "传入gun")
|
public R save(@RequestBody Gun gun) {
|
return R.status(gunService.save(gun));
|
}
|
|
/**
|
* 修改
|
* @param gun 枪支信息对象
|
*/
|
@PostMapping("/update")
|
public R update(@RequestBody Gun gun) {
|
return R.status(gunService.updateById(gun));
|
}
|
|
/**
|
* 新增或修改
|
* @param gun 枪支信息对象
|
*/
|
@PostMapping("/submit")
|
public R submit(@RequestBody Gun gun){
|
boolean status = false;
|
if (null==gun.getId()) {
|
//新增
|
status = gunService.save(gun);
|
|
//数据同步
|
String s1 =
|
"insert into sys_gun(id,issue_unit,card_number,gun_mode," +
|
"dept_id,person_in_charge,issue_time,valid_time,jurisdiction) " +
|
"values(" + "'" + gun.getId() + "'" + "," +
|
"'" + gun.getIssueUnit() + "'" + "," +
|
"'" + gun.getCardNumber() + "'" + "," +
|
"'" + gun.getGunMode() + "'" + "," +
|
"'" + gun.getDeptId() + "'" + "," +
|
"'" + gun.getPersonInCharge() + "'" + "," +
|
"'" + new SimpleDateFormat("yyyy-MM-dd").format(gun.getIssueTime()) + "'" + "," +
|
"'" + new SimpleDateFormat("yyyy-MM-dd").format(gun.getValidTime()) + "'" + "," +
|
"'" +gun.getJurisdiction() + "'" + ")";
|
FtpUtil.sqlFileUpload(s1);
|
}else {
|
//修改
|
status = gunService.updateById(gun);
|
//内网同步
|
String s1 =
|
"update sys_gun set issue_unit = " + "'" + gun.getIssueUnit() + "'" +
|
",card_number = " + "'" + gun.getCardNumber() + "'" +
|
",gun_mode = " + "'" + gun.getGunMode() + "'" +
|
",dept_id = " + "'" + gun.getDeptId() + "'" +
|
",person_in_charge = " + "'" + gun.getJurisdiction() + "'" +
|
",issue_time = " + "'" + new SimpleDateFormat("yyyy-MM-dd").format(gun.getIssueTime()) + "'" +
|
",valid_time = " + "'" + new SimpleDateFormat("yyyy-MM-dd").format(gun.getValidTime()) + "'" +
|
",jurisdiction = " + "'" + gun.getJurisdiction() + "'" +
|
" " +"where id = " + "'" + gun.getId() + "'";
|
FtpUtil.sqlFileUpload(s1);
|
}
|
return R.status(status);
|
}
|
|
/**
|
* 删除
|
* @param ids 枪支信息ids 数组
|
*/
|
@PostMapping("/remove")
|
public R remove(@ApiParam(value = "主键集合") @RequestParam String ids) {
|
//内网删除
|
List<Long> list = Func.toLongList(ids);
|
list.forEach(id ->{
|
//内网同步
|
String s1 = "delete from sys_gun where id = " + "'" + id + "'";
|
FtpUtil.sqlFileUpload(s1);
|
});
|
return R.status(gunService.removeByIds(Func.toLongList(ids)));
|
}
|
|
/**
|
* 详情
|
* @param gun 枪支信息对象
|
*/
|
@GetMapping("/detail")
|
@ApiOperation(value = "详情", notes = "传入gun")
|
public R<GunVo> detail(Gun gun) {
|
GunVo detail = gunService.selectGunInfo(gun);
|
return R.data(detail);
|
}
|
|
|
/**
|
* 导入枪支信息, excel
|
*/
|
@PostMapping("import-gun")
|
public R importGun(MultipartFile file, Integer isCovered, String deptId) {
|
GunImporter gunImporter = new GunImporter(gunService, false,deptId);
|
ExcelUtil.save(file, gunImporter, GunExcel.class);
|
return R.success("操作成功");
|
}
|
|
/**
|
* 枪支导出模板
|
*/
|
@GetMapping("export-template-gun")
|
@ApiOperation(value = "导出模板")
|
public void exportGun(HttpServletResponse response) throws IOException {
|
List<GunExcel> list = new ArrayList<>();
|
GunExcel gunExcel1 = new GunExcel();
|
gunExcel1.setDeptId("xxx保安公司");
|
gunExcel1.setCardNumber("x****");
|
gunExcel1.setGunMode("x****");
|
gunExcel1.setIssueUnit("*****");
|
gunExcel1.setPersonInCharge("张三");
|
list.add(gunExcel1);
|
String fileName = null;
|
try {
|
response.setContentType("application/vnd.ms-excel");
|
response.setCharacterEncoding(org.apache.commons.codec.Charsets.UTF_8.name());
|
fileName = URLEncoder.encode("枪支导入数据模板"+ DateUtil.time(), Charsets.UTF_8.name());
|
response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
|
//修改单元格格式为文本格式
|
EasyExcel.write(response.getOutputStream(), GunExcel.class).sheet("枪支数据表").registerWriteHandler(new RowWriteHandler()).doWrite(list);
|
} catch (Throwable var6) {
|
throw var6;
|
}
|
}
|
|
}
|
