package com.mindskip.xzs.controller.admin;
|
|
|
import cn.hutool.core.codec.Base64;
|
import com.mindskip.xzs.base.BaseApiController;
|
import com.mindskip.xzs.base.RestResponse;
|
import com.mindskip.xzs.service.ExcelImportService;
|
import com.mindskip.xzs.service.FileUpload;
|
import com.mindskip.xzs.viewmodel.admin.file.UeditorConfigVM;
|
import com.mindskip.xzs.viewmodel.admin.file.UploadResultVM;
|
import lombok.AllArgsConstructor;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.io.File;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.util.UUID;
|
|
|
/**
|
* @version 2.2.0
|
* @description: 上传
|
* Copyright (C), 2020-2021, 武汉思维跳跃科技有限公司
|
* @date 2021 /9/7 9:45
|
*/
|
@AllArgsConstructor
|
@RequestMapping("/api/admin/upload")
|
@RestController("AdminUploadController")
|
public class UploadController extends BaseApiController {
|
|
private final FileUpload fileUpload;
|
private static final Logger logger = LoggerFactory.getLogger(UploadController.class);
|
private static final String SCRAWL_UPLOAD = "scrawlUpload";
|
private static final String SCRAWL_FIELD_NAME = "scrawl";
|
private static final String IMAGE_UPLOAD = "imgUpload";
|
private static final String IMAGE_UPLOAD_FILE = "upFile";
|
private static final String STATE = "SUCCESS";
|
private final ExcelImportService excelImportService;
|
|
/**
|
* 图片上传
|
*
|
* @param request the request
|
* @return the object
|
*/
|
@ResponseBody
|
@RequestMapping("/configAndUpload")
|
public Object upload(HttpServletRequest request) {
|
String action = request.getParameter("action");
|
if (null == action)
|
return null;
|
String filePath = null;
|
switch (action) {
|
case IMAGE_UPLOAD:
|
MultipartHttpServletRequest multipartHttpServletRequest = (MultipartHttpServletRequest) request;
|
MultipartFile multipartFile = multipartHttpServletRequest.getFile(IMAGE_UPLOAD_FILE);
|
String fileName = multipartFile.getOriginalFilename();
|
long attachSize = multipartFile.getSize();
|
try (InputStream inputStream = multipartFile.getInputStream()) {
|
filePath = fileUpload.fileUpload(inputStream, attachSize, fileName, "image");
|
} catch (IOException e) {
|
logger.error(e.getMessage(), e);
|
}
|
return new UploadResultVM(fileName, fileName, filePath, multipartFile.getSize(), fileName, STATE);
|
case SCRAWL_UPLOAD:
|
String imageBase64 = request.getParameter(SCRAWL_FIELD_NAME);
|
String uuidName = UUID.randomUUID().toString();
|
try {
|
File file = File.createTempFile(uuidName, ".png");
|
try (FileOutputStream fileOutputStream = new FileOutputStream(file)) {
|
byte[] buff = Base64.decode(imageBase64);
|
fileOutputStream.write(buff);
|
filePath = fileUpload.fileUpload(file, uuidName + ".png", "latex");
|
}
|
} catch (IOException e) {
|
logger.error(e.getMessage(), e);
|
}
|
return new UploadResultVM(filePath, STATE);
|
default:
|
return new UeditorConfigVM(SCRAWL_UPLOAD, SCRAWL_FIELD_NAME, IMAGE_UPLOAD, IMAGE_UPLOAD_FILE);
|
}
|
}
|
|
|
/**
|
* excel题目导入
|
*
|
* @param request the request
|
* @return the rest response
|
*/
|
@RequestMapping("/excel")
|
@ResponseBody
|
public RestResponse excel(HttpServletRequest request) {
|
MultipartFile file = ((MultipartHttpServletRequest) request).getFile("file");
|
String excelTempPath = null;
|
try (InputStream inputStream = file.getInputStream()) {
|
excelTempPath = File.createTempFile(UUID.randomUUID().toString(), ".xlsx").getAbsolutePath();
|
excelImportService.adminQuestionImport(inputStream, excelTempPath, getCurrentUser());
|
} catch (IOException e) {
|
logger.error(e.getMessage(), "文件上传失败");
|
}
|
String filePath = fileUpload.fileUpload(new File(excelTempPath), "题目导入结果.xlsx", "excel");
|
return RestResponse.ok(filePath);
|
}
|
|
|
/**
|
* File rest response.
|
*
|
* @param request the request
|
* @return the rest response
|
*/
|
@RequestMapping("/file")
|
@ResponseBody
|
public RestResponse file(HttpServletRequest request) {
|
MultipartHttpServletRequest multipartHttpServletRequest = (MultipartHttpServletRequest) request;
|
MultipartFile multipartFile = multipartHttpServletRequest.getFile("file");
|
String fileName = multipartFile.getOriginalFilename();
|
long fileSize = multipartFile.getSize();
|
String filePath = null;
|
String folder = request.getParameter("folder");
|
try (InputStream inputStream = multipartFile.getInputStream()) {
|
filePath = fileUpload.fileUpload(inputStream, fileSize, fileName, folder);
|
} catch (IOException e) {
|
logger.error(e.getMessage(), e);
|
return RestResponse.fail(2, "文件上传失败");
|
}
|
return RestResponse.ok(filePath);
|
}
|
|
}
|
