From 2be38aabc8e20921057cbf4f5ba15c09ebce2bd8 Mon Sep 17 00:00:00 2001
From: 克 刘 <4345453@qq.com>
Date: Mon, 08 Jun 2020 15:32:14 +0800
Subject: [PATCH] 扩展文件服务支持KidgrowAI 按医院/科室/日期/文件 方式存储

---
 kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-server/src/main/java/com/kidgrow/filecenter/config/AliyunOSSAutoConfigure.java |   29 +++++++++++++++++++++++++----
 1 files changed, 25 insertions(+), 4 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 a66e263..1dea60b 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
@@ -1,6 +1,9 @@
 package com.kidgrow.filecenter.config;
 
+import com.aliyun.oss.OSSClient;
 import com.aliyun.oss.common.auth.DefaultCredentialProvider;
+import com.kidgrow.common.constant.SecurityConstants;
+import com.kidgrow.common.utils.DateUtil;
 import com.kidgrow.filecenter.model.FileInfo;
 import com.kidgrow.filecenter.properties.FileServerProperties;
 import com.kidgrow.filecenter.service.impl.AbstractIFileService;
@@ -8,10 +11,11 @@
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-
-import com.aliyun.oss.OSSClient;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
+
+import javax.servlet.http.HttpServletRequest;
+import java.util.Date;
 
 /**
  * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
@@ -26,6 +30,8 @@
 public class AliyunOSSAutoConfigure {
     @Autowired
     private FileServerProperties fileProperties;
+
+
 
     /**
      * 阿里云文件存储client
@@ -44,6 +50,9 @@
         @Autowired
         private OSSClient ossClient;
 
+        @Autowired
+        private HttpServletRequest httpServletRequest;
+
         @Override
         protected String fileType() {
             return fileProperties.getType();
@@ -51,12 +60,24 @@
 
         @Override
         protected void uploadFile(MultipartFile file, FileInfo fileInfo) throws Exception {
-            ossClient.putObject(fileProperties.getOss().getBucketName(), fileInfo.getName(), file.getInputStream());
-            fileInfo.setUrl(fileProperties.getOss().getDomain() + "/" + fileInfo.getName());
+            String clientID= httpServletRequest.getHeader(SecurityConstants.CLIENT_HEADER);
+            String folderByDate =DateUtil.formatDate(new Date());
+            if(clientID=="hospital"){//医院端平台,根据医院ID,科室ID进行OSS存储
+                String orgID=httpServletRequest.getHeader(SecurityConstants.USER_ORG_ID_HEADER);
+                String depID=httpServletRequest.getHeader(SecurityConstants.USER_DEP_ID_HEADER);
+                String fileFolder="KidgrowAI/"+orgID+"/"+depID+"/";
+                ossClient.putObject(fileProperties.getOss().getBucketName(), fileFolder + folderByDate + "/" + fileInfo.getName(), file.getInputStream());
+                fileInfo.setUrl(fileProperties.getOss().getDomain() + fileFolder + folderByDate + "/" + fileInfo.getName());
+            }
+            else {
+                ossClient.putObject(fileProperties.getOss().getBucketName(), fileProperties.getOss().getFolder() + folderByDate + "/" + fileInfo.getName(), file.getInputStream());
+                fileInfo.setUrl(fileProperties.getOss().getDomain() + fileProperties.getOss().getFolder() + folderByDate + "/" + fileInfo.getName());
+            }
         }
 
         @Override
         protected boolean deleteFile(FileInfo fileInfo) {
+
             ossClient.deleteObject(fileProperties.getOss().getBucketName(), fileInfo.getName());
             return true;
         }

--
Gitblit v1.8.0