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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package com.iailab.module.model.matlab.controller.admin;
 
import com.iailab.framework.common.exception.enums.GlobalErrorCodeConstants;
import com.iailab.framework.common.page.PageData;
import com.iailab.framework.common.pojo.CommonResult;
import com.iailab.module.model.matlab.common.exceptions.IllegalityJarException;
import com.iailab.module.model.matlab.dto.MatlabJarFileInfoDTO;
import com.iailab.module.model.matlab.dto.MatlabRunDTO;
import com.iailab.module.model.matlab.dto.MlModelDTO;
import com.iailab.module.model.matlab.service.MlModelService;
import com.iailab.module.model.mpk.dto.MpkFileDTO;
import io.swagger.v3.oas.annotations.Operation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
 
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
import static com.iailab.framework.common.pojo.CommonResult.success;
 
 
/**
 * @author Dzd 
 * @since 1.0.0 2025-02-08
 */
@RestController
@RequestMapping("/model/matlab/model")
public class MlModelController {
    @Autowired
    private MlModelService mlModelService;
 
    @GetMapping("page")
    @Operation(summary = "分页")
    @PreAuthorize("@ss.hasPermission('ml:model:query')")
    public CommonResult<PageData<MlModelDTO>> page(@RequestParam Map<String, Object> params) {
        PageData<MlModelDTO> page = mlModelService.page(params);
        return success(page);
    }
 
    @GetMapping("{id}")
    @Operation(summary = "信息")
    @PreAuthorize("@ss.hasPermission('ml:model:query')")
    public CommonResult<MlModelDTO> get(@PathVariable("id") String id){
        MlModelDTO data = mlModelService.get(id);
 
        return success(data);
    }
 
    @PreAuthorize("@ss.hasPermission('ml:model:query')")
    @GetMapping("list")
    public CommonResult<List<MlModelDTO>> list(@RequestParam Map<String, Object> params) {
        List<MlModelDTO> list = mlModelService.list(params);
 
        return success(list);
    }
 
    @PostMapping
    @Operation(summary = "保存")
    @PreAuthorize("@ss.hasPermission('ml:model:create')")
    public CommonResult save(@RequestBody MlModelDTO dto){
        mlModelService.save(dto);
 
        return success();
    }
 
    @PutMapping
    @Operation(summary = "修改")
    @PreAuthorize("@ss.hasPermission('ml:model:update')")
    public CommonResult update(@RequestBody MlModelDTO dto){
        mlModelService.update(dto);
 
        return success();
    }
 
    @DeleteMapping
    @Operation(summary = "删除")
    @PreAuthorize("@ss.hasPermission('ml:model:delete')")
    public CommonResult delete(String id){
 
        mlModelService.delete(id);
 
        return success();
    }
 
    @PostMapping("/upload")
    @Operation(summary = "算法封装jar文件上传")
    public CommonResult<MatlabJarFileInfoDTO> upload(@RequestParam("file") MultipartFile file){
        MatlabJarFileInfoDTO result = null;
        try {
            result = mlModelService.uploadJarFile(file);
        } catch (IllegalityJarException e) {
            return CommonResult.error(GlobalErrorCodeConstants.BAD_REQUEST.getCode(),"非法jar包");
        }
        return success(result);
    }
 
    @PostMapping("/test")
    @Operation(summary = "Matlab测试运行")
    public CommonResult<String> test(@RequestBody MatlabRunDTO dto) {
        return mlModelService.test(dto);
    }
 
    @PostMapping("/importData")
    @Operation(summary = "导入参数")
    public CommonResult<List<HashMap<String,Object>>> importData(@RequestParam("file") MultipartFile file) throws Exception {
        List<HashMap<String,Object>> result = mlModelService.importData(file);
        return success(result);
    }
 
}