iailab-plat.git

dengzedong

2024-10-10 a8f2f4e6f1a9173d7eb9779c7879674015c8cda0

提交 \| 用户 \| 时间
449017	1	package com.iailab.module.model.mpk.controller.admin;
D	2
	3	import com.iailab.framework.common.page.PageData;
	4	import com.iailab.framework.common.pojo.CommonResult;
	5	import com.iailab.module.model.mpk.dto.MpkFileDTO;
0255ea	6	import com.iailab.module.model.mpk.service.MpkFileService;
449017	7	import io.swagger.v3.oas.annotations.Operation;
D	8	import org.apache.commons.io.IOUtils;
	9	import org.springframework.beans.factory.annotation.Autowired;
b425df	10	import org.springframework.security.access.prepost.PreAuthorize;
d8db4b	11	import org.springframework.util.CollectionUtils;
449017	12	import org.springframework.web.bind.annotation.*;
D	13	import org.springframework.web.multipart.MultipartFile;
	14
	15	import javax.servlet.http.HttpServletResponse;
	16	import java.io.IOException;
	17	import java.net.URLEncoder;
	18	import java.util.*;
	19
	20	import static com.iailab.framework.common.pojo.CommonResult.success;
	21
	22	/**
	23	* @author dzd
	24	* @Description
	25	* @createTime 2024年08月14日
	26	*/
	27	@RestController
	28	@RequestMapping("/model/mpk/file")
	29	public class MpkFileController {
	30	@Autowired
0255ea	31	private MpkFileService mpkFileService;
449017	32
D	33	@GetMapping("page")
	34	@Operation(summary = "分页")
b425df	35	@PreAuthorize("@ss.hasPermission('mpk:file:query')")
449017	36	public CommonResult<PageData<MpkFileDTO>> page(@RequestParam Map<String, Object> params) {
0255ea	37	PageData<MpkFileDTO> page = mpkFileService.page(params);
449017	38	return success(page);
D	39	}
	40
b425df	41	@PreAuthorize("@ss.hasPermission('mpk:file:query')")
449017	42	@GetMapping("{id}")
D	43	public CommonResult<MpkFileDTO> info(@PathVariable("id") String id) {
0255ea	44	MpkFileDTO schedule = mpkFileService.get(id);
d8db4b	45	List<String> menuAndGroup = new ArrayList<>();
潘	46	menuAndGroup.add(schedule.getMenuName());
	47	menuAndGroup.add(schedule.getGroupName());
	48	schedule.setMenuAndGroup(menuAndGroup);
449017	49	return success(schedule);
D	50	}
	51
b425df	52	@PreAuthorize("@ss.hasPermission('mpk:file:query')")
449017	53	@GetMapping("list")
D	54	public CommonResult<List<MpkFileDTO>> list() {
0255ea	55	List<MpkFileDTO> list = mpkFileService.list(new HashMap<>());
449017	56
D	57	return success(list);
	58	}
	59
b425df	60	@PreAuthorize("@ss.hasPermission('mpk:file:create')")
449017	61	@PostMapping
b425df	62	public CommonResult<Boolean> save(@RequestBody MpkFileDTO dto) {
d8db4b	63	if (!CollectionUtils.isEmpty(dto.getMenuAndGroup())) {
潘	64	dto.setMenuName(dto.getMenuAndGroup().get(0));
	65	if (dto.getMenuAndGroup().size() > 1) {
	66	dto.setGroupName(dto.getMenuAndGroup().get(1));
	67	}
	68	}
0255ea	69	mpkFileService.save(dto);
b425df	70	return CommonResult.success(true);
449017	71	}
D	72
b425df	73	@PreAuthorize("@ss.hasPermission('mpk:file:delete')")
449017	74	@DeleteMapping
b425df	75	public CommonResult<Boolean> delete(String id) {
0255ea	76	mpkFileService.delete(id);
b425df	77	return CommonResult.success(true);
449017	78	}
D	79
b425df	80	@PreAuthorize("@ss.hasPermission('mpk:file:update')")
449017	81	@PutMapping
b425df	82	public CommonResult<Boolean> update(@RequestBody MpkFileDTO dto) {
d8db4b	83	if (!CollectionUtils.isEmpty(dto.getMenuAndGroup())) {
潘	84	dto.setMenuName(dto.getMenuAndGroup().get(0));
	85	if (dto.getMenuAndGroup().size() > 1) {
	86	dto.setGroupName(dto.getMenuAndGroup().get(1));
	87	}
	88	}
0255ea	89	mpkFileService.update(dto);
b425df	90	return CommonResult.success(true);
449017	91	}
D	92
	93	@GetMapping("generat")
1fea3e	94	public void generat(String id, String remark,String zipFileName, HttpServletResponse response) {
D	95	try {
	96	byte[] data = mpkFileService.generatorCode(id, remark,zipFileName);
449017	97
1fea3e	98	response.reset();
D	99	response.setHeader("Content-Disposition", "attachment; filename=\"" + URLEncoder.encode(zipFileName, "UTF-8") + "\"");
	100	response.addHeader("Content-Length", "" + data.length);
	101	response.setContentType("application/octet-stream; charset=UTF-8");
449017	102
1fea3e	103	IOUtils.write(data, response.getOutputStream());
D	104	} catch (Exception e) {
912a1e	105	throw new RuntimeException("代码生成异常",e);
1fea3e	106	}
449017	107	}
D	108
	109	@GetMapping("packageModel")
	110	public void packageModel(String ids ,String projectId,String log ,String projectName,String version,String zipFileName,HttpServletResponse response) throws IOException {
	111	byte[] data;
	112	try {
0255ea	113	data = mpkFileService.packageModel(Arrays.asList(ids.split(",")),projectId,projectName,zipFileName,log,version);
449017	114	} catch (InterruptedException e) {
D	115	throw new RuntimeException("模型打包失败",e);
	116	}
	117
	118	response.reset();
	119	response.setHeader("Content-Disposition", "attachment; filename=\"" + URLEncoder.encode(zipFileName, "UTF-8") + "\"");
	120	response.addHeader("Content-Length", "" + data.length);
	121	response.setContentType("application/octet-stream; charset=UTF-8");
	122
	123	IOUtils.write(data, response.getOutputStream());
	124	}
	125
	126	@PostMapping("/upload")
	127	@Operation(summary = "python文件上传")
c12dae	128	public CommonResult<Map<String,String>> upload(@RequestParam("file") MultipartFile file) throws Exception {
0255ea	129	Map<String,String> result = mpkFileService.savePyFile(file);
449017	130	return success(result);
D	131	}
c12dae	132
D	133	@PostMapping("/publish")
	134	public CommonResult<String> publish(@RequestBody Map<String, Object> params) {
	135	return mpkFileService.publish(params);
	136	}
449017	137	}