DeviceController.java
3.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package vion.controller;
import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.dev33.satoken.stp.StpUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.github.liaochong.myexcel.core.DefaultExcelBuilder;
import com.github.liaochong.myexcel.core.watermark.Watermark;
import com.github.liaochong.myexcel.utils.AttachmentV2ExportUtil;
import com.github.liaochong.myexcel.utils.WatermarkUtil;
import jakarta.servlet.http.HttpServletResponse;
import lombok.RequiredArgsConstructor;
import org.apache.poi.ss.usermodel.Workbook;
import org.dromara.hutool.core.date.TimeUtil;
import org.dromara.hutool.core.lang.Assert;
import org.dromara.hutool.core.text.StrUtil;
import org.springframework.web.bind.annotation.*;
import vion.dto.DeviceDTO;
import vion.service.IDeviceService;
import vion.vo.DeviceVO;
import vion.vo.UserVO;
import java.awt.*;
import java.io.IOException;
import java.time.LocalDateTime;
import java.util.List;
/**
* 设备管理
*/
@RestController
@RequestMapping("/api/device")
@RequiredArgsConstructor
public class DeviceController {
private final IDeviceService deviceService;
@GetMapping
@SaCheckPermission(value = "device:list", orRole = "admin")
public Page<DeviceVO> list(DeviceDTO dto) {
Assert.notNull(dto.getStatus(), "status 不能为空！");
return deviceService.list(dto);
}
@GetMapping("/{id}")
@SaCheckPermission(value = "device:query", orRole = "admin")
public DeviceVO getById(@PathVariable Long id) {
return deviceService.getById(id);
}
@PostMapping
@SaCheckPermission(value = "device:edit", orRole = "admin")
public String update(@RequestBody List<DeviceDTO> dtoList) {
return deviceService.update(dtoList);
}
@PostMapping("/record")
@SaCheckPermission(value = "device:record", orRole = "admin")
public Object circulate(@RequestBody List<DeviceDTO> dtoList) {
return deviceService.recordDevice(dtoList);
}
@PostMapping("/circulate")
@SaCheckPermission(value = "device:circulate", orRole = "admin")
public Object circulateBatch(@RequestBody List<DeviceDTO> dtoList) {
return deviceService.circulateBatch(dtoList);
}
@GetMapping("/out/export")
@SaCheckPermission(value = "device:out:export", orRole = "admin")
public void deviceExport(DeviceDTO dto, HttpServletResponse response) {
UserVO user = (UserVO) StpUtil.getTokenSession().get("curLoginUser");
dto.setPageSize(30000);
Page<DeviceVO> voPage = deviceService.list(dto);
try (DefaultExcelBuilder<DeviceVO> defaultExcelBuilder = DefaultExcelBuilder.of(DeviceVO.class)) {
Workbook workbook = defaultExcelBuilder.build(voPage.getRecords());
// 水印添加指定字体，并在服务器上安装 SimSun 字体，解决中文字体变成方块的问题
Watermark watermark = new Watermark();
watermark.setText(user.getUsername() + "-" + user.getPhone());
watermark.setFont(new Font("SimSun", Font.PLAIN, 16));
WatermarkUtil.addWatermark(workbook, watermark);
AttachmentV2ExportUtil.export(workbook, StrUtil.format("设备列表_{}", TimeUtil.formatNormal(LocalDateTime.now())), response);
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}