JYW
2024-04-23 7dcb6f8665db1334940dca300c4b83c6cd81373d
cnas-server/src/main/java/com/yuanchu/mom/controller/DeviceController.java
@@ -1,6 +1,7 @@
package com.yuanchu.mom.controller;
import cn.hutool.core.lang.UUID;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.yuanchu.mom.annotation.ValueAuth;
import com.yuanchu.mom.pojo.Device;
@@ -8,22 +9,32 @@
import com.yuanchu.mom.utils.JackSonUtil;
import com.yuanchu.mom.vo.Result;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.io.File;
import java.util.HashMap;
import java.util.Map;
/**
 * 设备(DeviceController)表控制层
 */
@Api(tags = "设备")
@AllArgsConstructor
@RestController
@RequestMapping("/deviceScope")
public class DeviceController {
    @Resource
    private DeviceService deviceService;
    @Value("${file.path}")
    private String filePath;
    @ApiOperation(value = "查询设备详情列表")
    @PostMapping("/selectDeviceParameter")
@@ -72,5 +83,48 @@
        return Result.success(deviceService.search(status, deviceName, specificationModel, largeCategory));
    }
    //图片上传
    @ApiOperation(value = "设备图片上传")
    @PostMapping("/uploadFile")
    public Result uploadFile(MultipartFile file) {
        String urlString;
        String pathName;
        String filename = file.getOriginalFilename();
        try {
            String path = filePath;
            File realpath = new File(path);
            if (!realpath.exists()) {
                realpath.mkdirs();
            }
            pathName = UUID.randomUUID() + "_" + file.getOriginalFilename();
            urlString = realpath + "/" + pathName;
            file.transferTo(new File(urlString));
            HashMap<String, String> map = new HashMap<>();
            map.put("name", filename);
            map.put("url", pathName);
            return Result.success(map);
        } catch (Exception e) {
            e.printStackTrace();
            System.err.println("图片上传错误");
            return null;
        }
    }
    @ApiOperation(value = "获取设备负责人")
    @GetMapping("/selectDevicePrincipal")
    @ValueAuth
    public Result selectDevicePrincipal() {
        return Result.success(deviceService.selectDevicePrincipal());
    }
    @ApiOperation(value = "通过设备分类获取设备列表")
    @PostMapping("/selectDeviceByCategory")
    @ValueAuth
    @ApiImplicitParams({
            @ApiImplicitParam(name = "category", value = "设备分类", dataTypeClass = String.class),
    })
    public Result selectDeviceByCategory(String category) {
        return Result.success(deviceService.selectDeviceByCategory(category));
    }
}