package com.admin.controller;
import com.alibaba.fastjson.JSONObject;
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.common.utils.BinaryUtil;
import com.aliyun.oss.model.*;
import com.admin.config.BaseController;
import com.appapi.common.FileUploadApiController;
import com.appapi.util.GeneratorUtil;
import org.apache.commons.io.FilenameUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.sql.Date;
import java.util.HashMap;
import java.util.Map;
@Controller
@RequestMapping("/common")
public class FileUploadController extends BaseController implements InitializingBean {
private static final Logger logger = LoggerFactory.getLogger(FileUploadApiController.class);
//密匙keyId 可以在阿里云获取到
public static String accessId = "你的阿里云accessId ";
//密匙keySecret 可以在阿里云获取到
public static String accessKey = "你的阿里云accessKey ";
//你的oss所在域,要加http:// 不明白可以对照你的文件引用地址
public static String endpoint = "oss-cn-shenzhen.aliyuncs.com";
//你的bucketName 名称 即是你的OSS对象名称 不明白查oss开发文档专业术语
public static String bucket = "ygyxj";
private static String callbackUrl="";
private static String dir="文件保存地址/";
private static String baseUrl="https://ygyxj.oss-cn-shenzhen.aliyuncs.com/";
/*@Autowired
private OSS ossClient2;*/
// @Autowired(required=true)
private static OSSClient ossClient = new OSSClient(endpoint, accessId, accessKey);
private String host;
// @Autowired
// private StringRedisTemplate stringRedisTemplate;
@Override
public void afterPropertiesSet() throws Exception {
host = "http://" + bucket + "." + endpoint;
}
/**
* 前台签名直传, 由服务器签名,用户可直接上传图片
* 这种只支持 Aliyun(因为我编码查看文档时,只有阿里云做了这个设计) 优点是 上传不需要占用应用服务器带宽。 目前前端是使用的这个。
* 若需要更改,请自行修改前端上传逻辑
* @param request
* @param response
*/
@RequestMapping(method = RequestMethod.GET)
public void upload(HttpServletRequest request, HttpServletResponse response) {
try {
long expireTime = 30;
long expireEndTime = System.currentTimeMillis() + expireTime * 1000;
Date expiration = new Date(expireEndTime);
PolicyConditions policyConds = new PolicyConditions();
policyConds.addConditionItem(PolicyConditions.COND_CONTENT_LENGTH_RANGE, 0, 1048576000);
policyConds.addConditionItem(MatchMode.StartWith, PolicyConditions.COND_KEY, dir);
String postPolicy = ossClient.generatePostPolicy(expiration, policyConds);
byte[] binaryData = postPolicy.getBytes("utf-8");
String encodedPolicy = BinaryUtil.toBase64String(binaryData);
String postSignature = ossClient.calculatePostSignature(postPolicy);
JSONObject respJsonObj = new JSONObject();
respJsonObj.put("accessid", accessId);
respJsonObj.put("policy", encodedPolicy);
respJsonObj.put("signature", postSignature);
respJsonObj.put("dir", dir);
respJsonObj.put("host", host);
respJsonObj.put("expire", String.valueOf(expireEndTime / 1000));
// respMap.put("expire", formatISO8601Date(expiration));
JSONObject jasonCallback = new JSONObject();
jasonCallback.put("callbackUrl", callbackUrl);
jasonCallback.put("callbackBody",
"filename=${object}&size=${size}&mimeType=${mimeType}&height=${imageInfo.height}&width=${imageInfo.width}");
jasonCallback.put("callbackBodyType", "application/x-www-form-urlencoded");
String base64CallbackBody = BinaryUtil.toBase64String(jasonCallback.toString().getBytes());
respJsonObj.put("callback", base64CallbackBody);
// System.out.println(ja1.toString());
response.setHeader("Access-Control-Allow-Origin", "*");
response.setHeader("Access-Control-Allow-Methods", "GET, POST");
respJsonObj.put("code", 200);
respJsonObj.put("baseUrl", baseUrl);
response(request, response, respJsonObj.toJSONString());
} catch (Exception e) {
e.printStackTrace();
logger.error("[上传签名] 异常", e);
}
}
/**
* 后台通过服务器间接传文件
* @param file
* @return
* @throws IOException
*/
@PostMapping("/upload")
public @ResponseBody Map<String, Object> create(@RequestParam("file") MultipartFile file) throws IOException {
ObjectMetadata objectMetadata = new ObjectMetadata();
String fileName = file.getOriginalFilename();
String uuid = GeneratorUtil.genFileName();
objectMetadata.setContentLength(file.getSize());
objectMetadata.setContentType(file.getContentType());
// objectMetadata.setContentType(getcontentType(fileName.substring(fileName.lastIndexOf("."))));
String ext = FilenameUtils.getExtension(file.getOriginalFilename());
PutObjectRequest putObjectRequest = new PutObjectRequest(bucket, "park/" + uuid+"."+ext, file.getInputStream(), objectMetadata);
PutObjectResult re = ossClient.putObject(putObjectRequest);
Map<String, Object> data = new HashMap<>();
data.put("url", baseUrl + "park/" + uuid +"."+ext);
data.put("code", 200);
data.put("msg", "成功");
data.put("size", file.getSize());
return data;
// return baseUrl + "park/" + uuid +"."+ext;
// return "服务器错误";
}
/**
* Description: 判断OSS服务文件上传时文件的contentType
*
* @param FilenameExtension 文件后缀
* @return String
*/
public static String getcontentType(String FilenameExtension) {
if (FilenameExtension.equalsIgnoreCase(".bmp")) {
return "image/bmp";
}
if (FilenameExtension.equalsIgnoreCase(".gif")) {
return "image/gif";
}
if (FilenameExtension.equalsIgnoreCase(".jpeg") ||
FilenameExtension.equalsIgnoreCase(".jpg") ||
FilenameExtension.equalsIgnoreCase(".png")) {
return "image/jpeg";
}
if (FilenameExtension.equalsIgnoreCase(".html")) {
return "text/html";
}
if (FilenameExtension.equalsIgnoreCase(".txt")) {
return "text/plain";
}
if (FilenameExtension.equalsIgnoreCase(".vsd")) {
return "application/vnd.visio";
}
if (FilenameExtension.equalsIgnoreCase(".pptx") ||
FilenameExtension.equalsIgnoreCase(".ppt")) {
return "application/vnd.ms-powerpoint";
}
if (FilenameExtension.equalsIgnoreCase(".docx") ||
FilenameExtension.equalsIgnoreCase(".doc")) {
return "application/msword";
}
if (FilenameExtension.equalsIgnoreCase(".xml")) {
return "text/xml";
}
return "image/jpeg";
}
/**
* Post请求
*/
@RequestMapping(method = RequestMethod.POST)
public void doPost(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
Map<String, String[]> parameterMap = request.getParameterMap();
JSONObject responseJson = new JSONObject();
responseJson.put("code", 200);
for (String key : parameterMap.keySet()) {
responseJson.put(key, parameterMap.get(key)[0]);
}
response(request, response, responseJson.toJSONString(), HttpServletResponse.SC_OK);
}
/**
* 服务器响应结果
*
* @param request
* @param response
* @param results
* @param status
* @throws IOException
*/
private void response(HttpServletRequest request, HttpServletResponse response, String results, int status)
throws IOException {
String callbackFunName = request.getParameter("callback");
response.addHeader("Content-Length", String.valueOf(results.length()));
if (callbackFunName == null || callbackFunName.equalsIgnoreCase(""))
response.getWriter().println(results);
else
response.getWriter().println(callbackFunName + "( " + results + " )");
response.setStatus(status);
response.flushBuffer();
}
/**
* 服务器响应结果
*/
private void response(HttpServletRequest request, HttpServletResponse response, String results) throws IOException {
String callbackFunName = request.getParameter("callback");
if (callbackFunName == null || callbackFunName.equalsIgnoreCase(""))
response.getWriter().println(results);
else
response.getWriter().println(callbackFunName + "( " + results + " )");
response.setStatus(HttpServletResponse.SC_OK);
response.flushBuffer();
}
//删除文件
public void deleteFile(@RequestParam("fileName")String fileName){
OSSClient ossClientDelete = new OSSClient(endpoint, accessId, accessKey);
String file = fileName.replace("https://ygyxj.oss-cn-shenzhen.aliyuncs.com/","");
ossClientDelete.deleteObject(bucket, file);
logger.info(fileName+"删除成功");
ossClientDelete.shutdown();
}
}
原文链接:https://blog.csdn.net/wgh0315/article/details/107787695