From 8562e0cc26459ef9cd5a3d5490285cc23f92df70 Mon Sep 17 00:00:00 2001
From: 克 刘 <4345453@qq.com>
Date: Sat, 15 Aug 2020 08:47:42 +0800
Subject: [PATCH] FileCenter支持根据环境变量进行同一个OSS服务中创建不同根文件夹

---
 kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-server/src/main/java/com/kidgrow/filecenter/config/AliyunOSSAutoConfigure.java |  116 ++++++++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 85 insertions(+), 31 deletions(-)

diff --git a/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-server/src/main/java/com/kidgrow/filecenter/config/AliyunOSSAutoConfigure.java b/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-server/src/main/java/com/kidgrow/filecenter/config/AliyunOSSAutoConfigure.java
index aca4c2e..5364009 100644
--- a/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-server/src/main/java/com/kidgrow/filecenter/config/AliyunOSSAutoConfigure.java
+++ b/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-server/src/main/java/com/kidgrow/filecenter/config/AliyunOSSAutoConfigure.java
@@ -2,12 +2,16 @@
 
 import com.aliyun.oss.OSSClient;
 import com.aliyun.oss.common.auth.DefaultCredentialProvider;
+import com.aliyun.oss.model.GetObjectRequest;
+import com.aliyun.oss.model.OSSObject;
 import com.kidgrow.common.constant.SecurityConstants;
 import com.kidgrow.common.utils.DateUtil;
+import com.kidgrow.common.utils.StringUtils;
 import com.kidgrow.filecenter.model.FileInfo;
 import com.kidgrow.filecenter.properties.FileServerProperties;
 import com.kidgrow.filecenter.service.impl.AbstractIFileService;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -15,6 +19,7 @@
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletRequest;
+import java.io.InputStream;
 import java.util.Date;
 import java.util.UUID;
 
@@ -31,7 +36,6 @@
 public class AliyunOSSAutoConfigure {
     @Autowired
     private FileServerProperties fileProperties;
-
 
 
     /**
@@ -59,42 +63,24 @@
             return fileProperties.getType();
         }
 
+        @Value("${spring.profiles.active}")
+        private String OssPathEn;
+
         @Override
         protected void uploadFile(MultipartFile file, FileInfo fileInfo) throws Exception {
-            String clientID= httpServletRequest.getHeader(SecurityConstants.CLIENT_HEADER);
-            String folderByDate =DateUtil.formatDate(new Date());
+            String clientID = httpServletRequest.getHeader(SecurityConstants.CLIENT_HEADER);
+            String folderByDate = DateUtil.formatDate(new Date());
             int begin = file.getOriginalFilename().indexOf(".");
             int last = file.getOriginalFilename().length();
             String fileType = file.getOriginalFilename().substring(begin, last);
-            String fileName= UUID.randomUUID().toString().replaceAll("-","")+fileType;
+            String fileName = UUID.randomUUID().toString().replaceAll("-", "") + fileType;
             String fileFolder = "";
-            if(clientID.equals("hospital")){//医院端平台,
-
-                if ((fileInfo.getImgType() != null)) {
-                    // 光片需要按照根据医院ID,科室ID,日期进行OSS存储
-                    if (fileInfo.getImgType().toLowerCase().equals("xray")) {
-                        String hospitalID = httpServletRequest.getHeader(SecurityConstants.USER_HOSPITAL_ID_HEADER);
-                        String depID = httpServletRequest.getHeader(SecurityConstants.USER_DEP_ID_HEADER);
-                        fileFolder = "Xray/" + hospitalID + "/" + depID + "/"+ folderByDate + "/";
-                    }
-                    // 医院Logo只保存在一个目录中
-                    else if (fileInfo.getImgType().toLowerCase().equals("logo")) {
-                        fileFolder = "HospitalLogo/";
-                    } else {
-                        fileFolder = "OtherImage/"+ folderByDate + "/";
-                    }
-
-                } else {
-                    fileFolder = "OtherImage/"+ folderByDate + "/";
-                }
-
-                ossClient.putObject(fileProperties.getOss().getBucketName(), fileFolder +  fileName, file.getInputStream());
-                fileInfo.setUrl(fileProperties.getOss().getDomain() + fileFolder  + fileName);
-
-            }
-            else {
-                ossClient.putObject(fileProperties.getOss().getBucketName(), fileProperties.getOss().getFolder() + folderByDate + "/" + fileName, file.getInputStream());
-                fileInfo.setUrl(fileProperties.getOss().getDomain() + fileProperties.getOss().getFolder() + folderByDate + "/" + fileName);
+            if (clientID.equals("hospital")) {//医院端平台,
+                fileFolder = FilePath(fileInfo.getImgType(), folderByDate, null, null);
+                fileUpLoadOss(fileInfo, fileFolder + fileName, file.getInputStream());
+            } else {
+                fileFolder = FilePath(fileInfo.getImgType(), folderByDate, null, null);
+                fileUpLoadOss(fileInfo, fileFolder + fileName, file.getInputStream());
             }
         }
 
@@ -104,5 +90,73 @@
             ossClient.deleteObject(fileProperties.getOss().getBucketName(), fileInfo.getName());
             return true;
         }
+
+        /**
+         * 给OSS上上传文件
+         *
+         * @param fileInfo    返回对象
+         * @param newFilePath 上传到的文件路径
+         * @param fileStream  要上传的文件流
+         */
+        public void fileUpLoadOss(FileInfo fileInfo, String newFilePath, InputStream fileStream) {
+            ossClient.putObject(fileProperties.getOss().getBucketName(), newFilePath, fileStream);
+            fileInfo.setUrl(fileProperties.getOss().getDomain() + newFilePath);
+
+        }
+
+        public OSSObject down(String url) {
+            GetObjectRequest request = new GetObjectRequest(fileProperties.getOss().getBucketName(), url);
+            return ossClient.getObject(request);
+        }
+
+        /**
+         * 组合sso上的文件路径
+         *
+         * @param imgType
+         * @return
+         */
+        public String FilePath(String imgType, String folderByDate, String hospitalId, String departmentId) {
+            //返回的文件路径
+            String fileFolder = OssPathEn+"/";
+
+
+            if (StringUtils.isNotBlank(imgType)) {
+                // 光片需要按照根据医院ID,科室ID,日期进行OSS存储
+                if (imgType.toLowerCase().equals("xray")) {
+                    String hospitalID = httpServletRequest.getHeader(SecurityConstants.USER_HOSPITAL_ID_HEADER);
+                    String depID = httpServletRequest.getHeader(SecurityConstants.USER_DEP_ID_HEADER);
+                    if (org.apache.commons.lang3.StringUtils.isBlank(hospitalID) &&
+                            org.apache.commons.lang3.StringUtils.isBlank(depID)) {
+                        fileFolder += "Xray/" + hospitalID + "/" + depID + "/" + folderByDate + "/";
+                    } else {
+                        if (org.apache.commons.lang3.StringUtils.isBlank(hospitalId) &&
+                                org.apache.commons.lang3.StringUtils.isBlank(departmentId)) {
+                            fileFolder += "Xray/" + hospitalId + "/" + departmentId + "/" + folderByDate + "/";
+                        } else {
+                            fileFolder += "Xray/" + folderByDate + "/";
+                        }
+                    }
+                } else if (imgType.toLowerCase().equals("app")) {
+                    //移动端APP上传
+                    if (org.apache.commons.lang3.StringUtils.isBlank(hospitalId) &&
+                            org.apache.commons.lang3.StringUtils.isBlank(departmentId)) {
+                        fileFolder += "AppXray/" + hospitalId + "/" + departmentId + "/" + folderByDate + "/";
+                    } else {
+                        fileFolder += "AppXray/" + folderByDate + "/";
+                    }
+                }
+                // 医院Logo只保存在一个目录中
+                else if (imgType.toLowerCase().equals("logo")) {
+                    fileFolder += "HospitalLogo/";
+                } else if (imgType.toLowerCase().equals("doctor")) {
+                    fileFolder += "DoctorImage/";
+                } else {
+                    fileFolder += "OtherImage/" + folderByDate + "/";
+                }
+            } else {
+                fileFolder += "OtherImage/" + folderByDate + "/";
+            }
+            return fileFolder;
+        }
     }
 }

--
Gitblit v1.8.0