package com.ks.daylucky.controller.admin;
|
|
import com.ks.daylucky.pojo.DTO.ImageFromEnum;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
import org.yeshi.utils.JsonUtil;
|
import org.yeshi.utils.entity.FileUploadResult;
|
import org.yeshi.utils.tencentcloud.COSManager;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.io.IOException;
|
import java.util.UUID;
|
|
@Controller
|
@RequestMapping("/admin/upload")
|
public class FileUploadController {
|
|
|
/**
|
* 上传图片
|
*
|
* @param request
|
* @param from
|
* @return
|
*/
|
@ResponseBody
|
@RequestMapping("image")
|
public String uploadImage(HttpServletRequest request, String from, String dir) {
|
MultipartFile image = ((MultipartHttpServletRequest) request).getFile("image");
|
ImageFromEnum imageFromEnum = ImageFromEnum.valueOf(from);
|
if (imageFromEnum == null) {
|
return JsonUtil.loadFalseResult("from错误");
|
}
|
String contentType = image.getContentType();
|
|
try {
|
FileUploadResult result = COSManager.getInstance().uploadFile(image.getInputStream(), "/image/" + from + "/" + dir + "/" + UUID.randomUUID().toString() + "." + getExtensionName(contentType));
|
return JsonUtil.loadTrueResult(result.getUrl());
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
return JsonUtil.loadFalseResult("上传文件出错");
|
}
|
|
private String getExtensionName(String contentType) {
|
return contentType.split("/")[1];
|
}
|
|
|
}
|