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 |  121 ++++++++++++++++++++++++++++++++-------
 1 files changed, 98 insertions(+), 23 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 2783c87..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,13 +19,16 @@
 import lombok.extern.slf4j.Slf4j;
 import net.coobird.thumbnailator.Thumbnails;
 import org.apache.commons.collections4.MapUtils;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.web.multipart.MultipartFile;
 import sun.misc.BASE64Decoder;
-import sun.misc.BASE64Encoder;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import sun.misc.BASE64Encoder;
 
 import java.io.*;
+import java.net.HttpURLConnection;
+import java.net.URL;
 import java.util.*;
 
 /**
@@ -40,6 +44,8 @@
     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);
@@ -94,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);
 
     /**
      * 删除文件
@@ -153,7 +160,7 @@
                 //将生成的文件转换为流
                 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());
@@ -183,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;
         }
@@ -208,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;
@@ -219,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