package com.bizzan.bitrade.controller.common; import com.aliyun.oss.ClientException; import com.aliyun.oss.OSSClient; import com.aliyun.oss.OSSException; import com.bizzan.bitrade.annotation.AccessLog; import com.bizzan.bitrade.config.AliyunConfig; import com.bizzan.bitrade.constant.AdminModule; import com.bizzan.bitrade.controller.BaseController; import com.bizzan.bitrade.service.LocaleMessageSourceService; import com.bizzan.bitrade.util.GeneratorUtil; import com.bizzan.bitrade.util.MessageResult; import com.bizzan.bitrade.util.UploadFileUtil; import org.apache.commons.fileupload.servlet.ServletFileUpload; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.shiro.authz.annotation.RequiresPermissions; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.util.Assert; import org.springframework.util.Base64Utils; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.ByteArrayInputStream; import java.io.IOException; import java.io.InputStream; import java.text.SimpleDateFormat; import java.util.Date; @Controller @RequestMapping("/common/upload") public class UploadController extends BaseController { @Autowired private LocaleMessageSourceService sourceService; private static Log log = LogFactory.getLog(UploadController.class); private Logger logger = LoggerFactory.getLogger(UploadController.class); private String savePath = "data/upload/{:cate}/{yyyy}{mm}{dd}/{time}{rand:6}"; private String allowedFormat = ".jpg,.gif,.png"; private long maxAllowedSize = 1024 * 10000; @Autowired private AliyunConfig aliyunConfig; @RequestMapping(value = "/oss/image", method = RequestMethod.POST) @ResponseBody @AccessLog(module = AdminModule.COMMON, operation = "上传oss图片") public String uploadOssImage( HttpServletRequest request, HttpServletResponse response, @RequestParam("file") MultipartFile file) throws IOException { log.info(request.getSession().getServletContext().getResource("/")); response.setCharacterEncoding("UTF-8"); response.setContentType("text/html; charset=UTF-8"); if (!ServletFileUpload.isMultipartContent(request)) { return MessageResult.error(500, sourceService.getMessage("FORMAT_NOT_SUPPORTED")).toString(); } if (file == null) { return MessageResult.error(500, sourceService.getMessage("FILE_NOT_FOUND")).toString(); } String directory = new SimpleDateFormat("yyyy/MM/dd/").format(new Date()); OSSClient ossClient = new OSSClient(aliyunConfig.getOssEndpoint(), aliyunConfig.getAccessKeyId(), aliyunConfig.getAccessKeySecret()); try { String fileName = file.getOriginalFilename(); String suffix = fileName.substring(fileName.lastIndexOf("."), fileName.length()); String key = directory + GeneratorUtil.getUUID() + suffix; System.out.println(key); ossClient.putObject(aliyunConfig.getOssBucketName(), key, file.getInputStream()); String uri = aliyunConfig.toUrl(key); MessageResult mr = new MessageResult(0, sourceService.getMessage("SUCCESS")); mr.setData(uri); return mr.toString(); } catch (OSSException oe) { return MessageResult.error(500, oe.getErrorMessage()).toString(); } catch (ClientException ce) { System.out.println("Error Message: " + ce.getMessage()); return MessageResult.error(500, ce.getErrorMessage()).toString(); } catch (Throwable e) { e.printStackTrace(); return MessageResult.error(500, sourceService.getMessage("REQUEST_FAILED")).toString(); } finally { ossClient.shutdown(); } } @RequestMapping(value = "/local/image", method = RequestMethod.POST) @ResponseBody @AccessLog(module = AdminModule.COMMON, operation = "上传本地图片") public String uploadLocalImage(HttpServletRequest request, HttpServletResponse response, @RequestParam("file") MultipartFile file) throws IOException { log.info(request.getSession().getServletContext().getResource("/").toString()); response.setCharacterEncoding("UTF-8"); response.setContentType("text/html; charset=UTF-8"); Assert.isTrue(ServletFileUpload.isMultipartContent(request), sourceService.getMessage("FORM_FORMAT_ERROR")); Assert.isTrue(file != null, sourceService.getMessage("NOT_FIND_FILE")); //验证文件类型 String fileName = file.getOriginalFilename(); String suffix = fileName.substring(fileName.lastIndexOf("."), fileName.length()); if (!allowedFormat.contains(suffix.trim().toLowerCase())) { return MessageResult.error(sourceService.getMessage("FORMAT_NOT_SUPPORT")).toString(); } String result = UploadFileUtil.uploadFile(file, fileName); if (result != null) { MessageResult mr = new MessageResult(0, sourceService.getMessage("UPLOAD_SUCCESS")); mr.setData(result); return mr.toString(); } else { MessageResult mr = new MessageResult(0, sourceService.getMessage("FAILED_TO_WRITE")); mr.setData(result); return mr.toString(); } } @RequiresPermissions("common:upload:oss:base64") @RequestMapping(value = "/oss/base64", method = RequestMethod.POST) @ResponseBody @AccessLog(module = AdminModule.COMMON, operation = "base64上传oss") public MessageResult base64UpLoad(@RequestParam String base64Data) { MessageResult result = new MessageResult(); try { logger.debug("上传文件的数据：" + base64Data); String dataPrix = ""; String data = ""; logger.debug("对数据进行判断"); if (base64Data == null || "".equals(base64Data)) { throw new Exception("上传失败，上传图片数据为空"); } else { String[] d = base64Data.split("base64,"); if (d != null && d.length == 2) { dataPrix = d[0]; data = d[1]; } else { throw new Exception("上传失败，数据不合法"); } } logger.debug("对数据进行解析，获取文件名和流数据"); String suffix = ""; if ("data:image/jpeg;".equalsIgnoreCase(dataPrix)) {//data:image/jpeg;base64,base64编码的jpeg图片数据 suffix = ".jpg"; } else if ("data:image/x-icon;".equalsIgnoreCase(dataPrix)) {//data:image/x-icon;base64,base64编码的icon图片数据 suffix = ".ico"; } else if ("data:image/gif;".equalsIgnoreCase(dataPrix)) {//data:image/gif;base64,base64编码的gif图片数据 suffix = ".gif"; } else if ("data:image/png;".equalsIgnoreCase(dataPrix)) {//data:image/png;base64,base64编码的png图片数据 suffix = ".png"; } else { throw new Exception("上传图片格式不合法"); } String directory = new SimpleDateFormat("yyyy/MM/dd/").format(new Date()); String key = directory + GeneratorUtil.getUUID() + suffix; //因为BASE64Decoder的jar问题，此处使用spring框架提供的工具包 byte[] bs = Base64Utils.decodeFromString(data); OSSClient ossClient = new OSSClient(aliyunConfig.getOssEndpoint(), aliyunConfig.getAccessKeyId(), aliyunConfig.getAccessKeySecret()); try { //使用apache提供的工具类操作流 InputStream is = new ByteArrayInputStream(bs); //FileUtils.writeByteArrayToFile(new File(Global.getConfig(UPLOAD_FILE_PAHT), tempFileName), bs); ossClient.putObject(aliyunConfig.getOssBucketName(), key, is); String uri = aliyunConfig.toUrl(key); MessageResult mr = new MessageResult(0, "上传成功"); mr.setData(uri); logger.debug("上传成功,key:{}", key); return mr; } catch (Exception ee) { throw new Exception("上传失败，写入文件失败，" + ee.getMessage()); } } catch (Exception e) { logger.debug("上传失败," + e.getMessage()); result.setCode(500); result.setMessage("上传失败," + e.getMessage()); } return result; } }