From 3dcb4e0ebfd43190957f556d886917b2a2ffa064 Mon Sep 17 00:00:00 2001
From: zhaoxiaohao <913652501@qq.com>
Date: Wed, 12 Aug 2020 18:45:30 +0800
Subject: [PATCH] 修改上传图片的接口

---
 kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/service/impl/AbstractIFileService.java |  150 ++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 115 insertions(+), 35 deletions(-)

diff --git a/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/service/impl/AbstractIFileService.java b/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/service/impl/AbstractIFileService.java
index 71d54eb..a57b098 100644
--- a/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/service/impl/AbstractIFileService.java
+++ b/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/service/impl/AbstractIFileService.java
@@ -1,6 +1,7 @@
 package com.kidgrow.filecenter.service.impl;
 
 import cn.hutool.core.util.IdUtil;
+import com.aliyun.oss.model.OSSObject;
 import com.aliyun.oss.model.ObjectMetadata;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -18,17 +19,17 @@
 import lombok.extern.slf4j.Slf4j;
 import net.coobird.thumbnailator.Thumbnails;
 import org.apache.commons.collections4.MapUtils;
-import org.springframework.http.MediaType;
-import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.web.multipart.MultipartFile;
 import sun.misc.BASE64Decoder;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import sun.misc.BASE64Encoder;
 
 import java.io.*;
-import java.util.Date;
-import java.util.List;
-import java.util.Map;
-import java.util.UUID;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.util.*;
 
 /**
  * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
@@ -41,7 +42,10 @@
 @Slf4j
 public abstract class AbstractIFileService extends ServiceImpl<FileMapper, FileInfo> implements IFileService {
     private static final String FILE_SPLIT = ".";
+    private static Logger logger = LoggerFactory.getLogger(AbstractIFileService.class);
 
+    @Value("${kidgrow.file-server.oss.domain}")
+    private  String DOMAIN;
     @Override
     public FileInfo upload(MultipartFile file) throws Exception {
         FileInfo fileInfo = FileUtil.getFileInfo(file);
@@ -96,7 +100,8 @@
      */
     protected abstract void uploadFile(MultipartFile file, FileInfo fileInfo) throws Exception;
     protected abstract void fileUpLoadOss(FileInfo fileInfo,String newFilePath, InputStream fileStream);
-    protected abstract String FilePath(String imgType,String folderByDate);
+    protected abstract String FilePath(String imgType,String folderByDate,String hospitalId,String departmentId);
+    protected abstract OSSObject down(String url);
 
     /**
      * 删除文件
@@ -135,22 +140,27 @@
     @Override
     public FileInfo uploadForThumbnails(MultipartFile file, String imgType) throws Exception {
         FileInfo fileInfo=this.upload(file,imgType);
+        List<String> lassStr= Arrays.asList("png","bmp","dib","gif","jfif","jpe","jpeg","jpg","tif","tiff","ico");
+        String suffix=file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")+1);
+        if(!lassStr.contains(suffix)){
+            return null;
+        }
         if (StringUtils.isNotBlank(fileInfo.getUrl())) {
             //生成缩略图上传
-            File nowFile= MultipartFileUtils.multipartFileToFile(file);
+//            File nowFile= MultipartFileUtils.multipartFileToFile(file);
+//            File nowFile= new File(file.getOriginalFilename());
             //暂存目录  发布后真实存在的磁盘目录
             String result = CommonConstant.TEMP_IMAGE_PATH;
             //文件扩展名
-            String suffix=file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
-            String newFileName= RandomValueUtils.uuid()+suffix;
+            String newFileName= RandomValueUtils.uuid()+"."+suffix;
             //生成缩略图
-            Thumbnails.of(nowFile).size(CommonConstant.TH_IMG_WIDTH, CommonConstant.TH_IMG_HEIGHT).toFile(result+newFileName);
+            Thumbnails.of(file.getInputStream()).size(CommonConstant.TH_IMG_WIDTH, CommonConstant.TH_IMG_HEIGHT).toFile(result+newFileName);
             File thuFile= new File(result+newFileName);
             if (thuFile.exists()) {
                 //将生成的文件转换为流
-                InputStream inputStream=new FileInputStream(new File(result+newFileName));
+                InputStream inputStream=new FileInputStream(thuFile);
                 String folderByDate = DateUtil.formatDate(new Date());
-                String filefloder=this.FilePath(imgType,folderByDate);
+                String filefloder=this.FilePath(imgType,folderByDate,null,null);
                 FileInfo newfileInfo=new FileInfo();
                 newfileInfo.setName(newFileName);
                 newfileInfo.setId(IdUtil.fastSimpleUUID());
@@ -163,8 +173,10 @@
                 //文件上传到oss
                 this.fileUpLoadOss(newfileInfo,filefloder+newFileName,inputStream);
                 inputStream.close();
-                //删除暂存文件
-                MultipartFileUtils.delteTempFile(thuFile);
+//                //删除暂存文件
+//                MultipartFileUtils.delteTempFile(thuFile);
+                boolean delete = thuFile.delete();
+
                 // 将文件信息保存到数据库
                 baseMapper.insert(newfileInfo);
                 fileInfo.setPath(newfileInfo.getUrl());
@@ -178,7 +190,7 @@
     }
 
     @Override
-    public FileInfo baseUplaod(String file, String imgType) {
+    public FileInfo baseUplaod(String file, String imgType, String hospitalId, String departmentId) {
         if(file==null||"".equals(file.trim())){
             return null;
         }
@@ -203,7 +215,9 @@
             newfileInfo.setSource("aliyun");
             newfileInfo.setImgType(imgType);
             newfileInfo.setCreateTime(new Date());
-            this.fileUpLoadOss(newfileInfo,"HospitalLogo/"+newFileName,input);
+            String folderByDate =DateUtil.formatDate(new Date());
+            String filePath = FilePath(imgType, folderByDate, hospitalId, departmentId);
+            this.fileUpLoadOss(newfileInfo,filePath,input);
             input.close();
             baseMapper.insert(newfileInfo);
             return newfileInfo;
@@ -214,29 +228,95 @@
     }
 
     @Override
-    public ResultBody baseDownLoad(Map<String, Object> map) {
-        List<FileInfo> fileInfos = baseMapper.selectByMap(map);
-        if(fileInfos.size()>0){
-            FileInfo fileInfo = fileInfos.get(0);
-            InputStream in = null;
-            byte[] data = null;
-            //读取图片字节数组
-            try{
-                in = new FileInputStream(fileInfo.getUrl());
-                data = new byte[in.available()];
-                in.read(data);
-                in.close();
-            }catch (IOException e){
+    public ResultBody baseDownLoad(Map<String, Object> map) throws Exception {
+        String urlDiZhi=MapUtils.getString(map,"url");
+        if(!urlDiZhi.contains(DOMAIN)){
+            ByteArrayOutputStream outPut = new ByteArrayOutputStream();
+            byte[] data = new byte[1024];
+            try {
+                // 创建URL
+                URL url = new URL(urlDiZhi);
+                // 创建链接
+                HttpURLConnection conn = (HttpURLConnection) url.openConnection();
+                conn.setRequestMethod("GET");
+                conn.setConnectTimeout(10 * 1000);
+                InputStream inStream = conn.getInputStream();
+                int len = -1;
+                while ((len = inStream.read(data)) != -1) {
+                    outPut.write(data, 0, len);
+                }
+                inStream.close();
+            } catch (IOException e) {
                 e.printStackTrace();
             }
-            //对字节数组Base64编码
+            String substring = urlDiZhi.substring(urlDiZhi.lastIndexOf(".") + 1, urlDiZhi.length());
+            String str="data:image/" + substring + ";base64,";
+            // 对字节数组Base64编码
             BASE64Encoder encoder = new BASE64Encoder();
-            //返回Base64编码过的字节数组字符串
-            String encode = encoder.encode(data);
-            return ResultBody.ok().data(encode);
+            System.out.println(encoder.encode(outPut.toByteArray()));
+            return ResultBody.ok().data(str+encoder.encode(outPut.toByteArray()).replaceAll("\n", "").replaceAll("\r", ""));
         }else {
-            return ResultBody.failed("获取数据失败");
+            List<FileInfo> fileInfos = baseMapper.selectByMap(map);
+            if (fileInfos.size() > 0) {
+                FileInfo fileInfo = fileInfos.get(0);
+                OSSObject down = this.down(fileInfo.getUrl().replace(DOMAIN, ""));
+                InputStream objectContent = down.getObjectContent();
+                //返回Base64编码过的字节数组字符串
+                byte[] data = null;
+                // 读取图片字节数组
+                try {
+                    ByteArrayOutputStream swapStream = new ByteArrayOutputStream();
+                    byte[] buff = new byte[100];
+                    int rc = 0;
+                    while ((rc = objectContent.read(buff, 0, 100)) > 0) {
+                        swapStream.write(buff, 0, rc);
+                    }
+                    data = swapStream.toByteArray();
+                } catch (IOException e) {
+                    e.printStackTrace();
+                } finally {
+                    if (objectContent != null) {
+                        try {
+                            objectContent.close();
+                            down.close();
+                        } catch (IOException e) {
+                            throw new Exception("输入流关闭异常");
+                        }
+                    }
+                }
+                String encode = "data:" + fileInfo.getContentType() + ";base64," + Base64.getEncoder().encodeToString(data);
+                return ResultBody.ok().data(encode);
+            } else {
+                return ResultBody.failed("获取数据失败");
+            }
         }
     }
+
+    public static void main(String[] args) {
+
+        ByteArrayOutputStream outPut = new ByteArrayOutputStream();
+        byte[] data = new byte[1024];
+        try {
+            // 创建URL
+            URL url = new URL("http://192.168.2.25:8008/123/J-1.jpg");
+            // 创建链接
+            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
+            conn.setRequestMethod("GET");
+            conn.setConnectTimeout(10 * 1000);
+            InputStream inStream = conn.getInputStream();
+            int len = -1;
+            while ((len = inStream.read(data)) != -1) {
+                outPut.write(data, 0, len);
+            }
+            inStream.close();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        // 对字节数组Base64编码
+        BASE64Encoder encoder = new BASE64Encoder();
+        System.out.println(encoder.encode(outPut.toByteArray()));
+
+    }
+
 }
 

--
Gitblit v1.8.0