引言 眾所周知,數據流分析是實現污點分析的一種常用技術 數據流分析分為過程內的數據流分析與過程間的數據流分析。前者是對一個方法體內的數據流分析,主要是基於CFG分析,不涉及方法調用;後者是基於不同方法間的數據流分析,主要是基於ICFG+CG分析,會涉及方法調用。 一、過程內數據流分析 1. CFG的 ...
支持Fast DFS、伺服器、OSS等上傳方式
介紹
在實際的業務中,可以根據客戶的需求設置不同的文件上傳需求,支持普通伺服器上傳+分散式上傳(Fast DFS)+雲服務上傳OSS(OSS)
軟體架構
為了方便演示使用,本項目使用的是前後端不分離的架構
前端:Jquery.uploadFile
後端:SpringBoot
前期準備:FastDFS、OSS(華為)、伺服器
實現邏輯
通過 application 配置對上傳文件進行一個自定義配置,從而部署在不同客戶環境可以自定義選擇方式。
優點:
- 一鍵切換;
- 支持當前主流方式;
缺點:
- 遷移數據難度增加:因為表示FileID在對象存儲和伺服器上傳都是生成的UUID,而FastDFS是返回存取ID,當需要遷移的時候,通過腳本可以快速將FastDFS的數據遷移上雲,因為存儲ID可以共用。但是對象存儲和伺服器上傳的UUID無法被FastDFS使用,增加遷移成本
核心代碼
package com.example.file.util;
import com.example.file.common.ResultBean;
import com.github.tobato.fastdfs.domain.StorePath;
import com.github.tobato.fastdfs.proto.storage.DownloadByteArray;
import com.github.tobato.fastdfs.service.FastFileStorageClient;
import com.obs.services.ObsClient;
import com.obs.services.exception.ObsException;
import com.obs.services.model.DeleteObjectRequest;
import com.obs.services.model.GetObjectRequest;
import com.obs.services.model.ObsObject;
import com.obs.services.model.PutObjectResult;
import io.micrometer.common.util.StringUtils;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.web.multipart.MultipartFile;
import java.io.*;
import java.util.Objects;
import java.util.UUID;
@Slf4j
public class FileUtil {
/**
*
* @param file 文件
* @param uploadFlag 標識
* @param uploadPath 上傳路徑
* @param endPoint 功能變數名稱
* @param ak ak
* @param sk sk
* @param bucketName 桶名字
* @return fileId 用於下載
*/
public static ResultBean uploadFile(MultipartFile file, String uploadFlag, String uploadPath,
FastFileStorageClient fastFileStorageClient,
String endPoint, String ak, String sk,
String bucketName) {
if (StringUtils.isBlank(uploadFlag)){
ResultBean.error("uploadFlag is null");
}
switch (uploadFlag){
case "fastDFS":
return uploadFileByFastDFS(file,fastFileStorageClient);
case "huaweiOOS":
return uploadFileByHuaweiObject(file, endPoint, ak, ak, bucketName);
case "server":
default:
return uploadFileByOrigin(file,uploadPath);
}}
/**
* 上傳文件fastDFS
* @param file 文件名
* @return
*/
private static ResultBean uploadFileByFastDFS(MultipartFile file,FastFileStorageClient fastFileStorageClient){
Long size=file.getSize();
String fileName=file.getOriginalFilename();
String extName=fileName.substring(fileName.lastIndexOf(".")+1);
InputStream inputStream=null;
try {
inputStream=file.getInputStream();
//1-上傳的文件流 2-文件的大小 3-文件的尾碼 4-可以不管他
StorePath storePath=fastFileStorageClient.uploadFile(inputStream,size,extName,null);
log.info("[uploadFileByFastDFS][FullPath]"+storePath.getFullPath());
return ResultBean.success(storePath.getPath());
}catch (Exception e){
log.info("[ERROR][uploadFileByFastDFS]"+e.getMessage());
return ResultBean.error(e.getMessage());
}finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 對象存儲上傳
* @param file 文件名
* @return
*/
private static ResultBean uploadFileByHuaweiObject(MultipartFile file, String huaweiEndPoint,String huaweiobsAk,String huaweiobsSk,
String bucketName){
String fileName = file.getOriginalFilename();
fileName = renameToUUID(fileName);
InputStream inputStream=null;
try {
inputStream=file.getInputStream();
// 創建ObsClient實例
ObsClient obsClient = new ObsClient(huaweiobsAk, huaweiobsSk, huaweiEndPoint);
PutObjectResult result = obsClient.putObject(bucketName, fileName, inputStream);
obsClient.close();
return ResultBean.success(fileName);
}catch (ObsException e){
log.info("[ERROR][uploadFileByHuaweiObject]"+e.getErrorMessage());
return ResultBean.error(e.getErrorMessage());
}catch (Exception e){
log.info("[ERROR][uploadFileByHuaweiObject]"+e.getMessage());
return ResultBean.error(e.getMessage());
}finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 上傳文件原本方法
* @param file 文件名
* @return
*/
private static ResultBean uploadFileByOrigin(MultipartFile file,String uploadPath){
String fileName = file.getOriginalFilename();
fileName = renameToUUID(fileName);
File targetFile = new File(uploadPath);
if (!targetFile.exists()) {
targetFile.mkdirs();
}
FileOutputStream out = null;
try {
out = new FileOutputStream(uploadPath + fileName);
out.write(file.getBytes());
} catch (IOException e) {
log.info("[ERROR][uploadFileByOrigin]"+e.getMessage());
return ResultBean.error(e.getMessage());
}finally {
try {
out.flush();
} catch (IOException e) {
e.printStackTrace();
}
try {
out.close();
} catch (IOException e) {
e.printStackTrace();
}
}
return ResultBean.success(fileName);
}
/**
* 下載
* @return
*/
public static byte[] downloadFile(String fileId,String uploadFlag,String uploadPath
,FastFileStorageClient fastFileStorageClient, String group,
String endPoint,String ak,String sk,
String bucketName) {
byte[] result=null;
switch (uploadFlag){
case "fastDFS":
result =downloadFileByFastDFS(fileId,fastFileStorageClient,group);
break;
case "huaweiOOS":
result =downloadFileByHuaweiObject(fileId, endPoint, ak, sk, bucketName);
break;
case "server":
default:
String path2 = uploadPath + fileId;
path2 = path2.replace("//", "/");
result=downloadFileByOrigin(path2);
break;
}
return result;
}
/**
* 下載文件fastDFS
* @param fileId 文件名
* @return
*/
private static byte[] downloadFileByFastDFS(String fileId,FastFileStorageClient fastFileStorageClient,
String group){
DownloadByteArray callback=new DownloadByteArray();
byte[] group1s=null;
try {
group1s = fastFileStorageClient.downloadFile(group, fileId, callback);
}catch (Exception e){
log.info("[ERROR][downloadFileByFastDFS]"+e.getMessage());
}
return group1s;
}
/**
* 下載文件對象存儲
* @param fileId 文件名
* @return
*/
private static byte[] downloadFileByHuaweiObject(String fileId, String huaweiEndPoint,String huaweiobsAk,String huaweiobsSk,
String bucketName){
byte[] bytes =null;
try {
// 創建ObsClient實例
ObsClient obsClient = new ObsClient(huaweiobsAk, huaweiobsSk, huaweiEndPoint);
// 構造GetObjectRequest請求
GetObjectRequest getObjectRequest = new GetObjectRequest(bucketName, fileId);
// 執行下載操作
ObsObject obsObject = obsClient.getObject(getObjectRequest);
bytes = inputStreamToByteArray(obsObject.getObjectContent());
// 關閉OBS客戶端
obsClient.close();
return bytes;
}catch (ObsException e){
log.info("[ERROR][downloadFileByHuaweiObject]"+e.getErrorMessage());
}catch (Exception e) {
log.info("[ERROR][downloadFileByHuaweiObject]"+e.getMessage());
}
return bytes;
}
/**
*
* @param input
* @return
* @throws IOException
*/
private static byte[] inputStreamToByteArray(InputStream input) throws IOException {
ByteArrayOutputStream output = new ByteArrayOutputStream();
byte[] buffer = new byte[4096];
int n = 0;
while (-1 != (n = input.read(buffer))) {
output.write(buffer, 0, n);
}
return output.toByteArray();
}
/**
* 下載文件
* @param fileId 文件名
* @return
*/
private static byte[] downloadFileByOrigin(String fileId){
File file =new File(fileId);
InputStream inputStream=null;
byte[] buff = new byte[1024];
byte[] result=null;
BufferedInputStream bis = null;
ByteArrayOutputStream os = null;
try {
os=new ByteArrayOutputStream();
bis = new BufferedInputStream(new FileInputStream(file));
int i = bis.read(buff);
while (i != -1) {
os.write(buff, 0, buff.length);
i = bis.read(buff);
}
result=os.toByteArray();
os.flush();
} catch (Exception e) {
log.info("[ERROR][downloadFile]"+e.getMessage());
}finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
return result;
}
/**
* 刪除文件
* @param fileId 文件ID
* @return 刪除失敗返回-1,否則返回0
*/
public static boolean deleteFile(String fileId,String fastDFSFlag,FastFileStorageClient fastFileStorageClient,
String group,String uploadPath) {
boolean result=false;
if (StringUtils.isNotBlank(fastDFSFlag)&&
fastDFSFlag.trim().equalsIgnoreCase("true")){
result =deleteFileByFastDFS(fileId,fastFileStorageClient,group);
}else {
String path2 = uploadPath + fileId;
path2 = path2.replace("//", "/");
result=deleteByOrigin(path2);
}
return result;
}
private static boolean deleteByOrigin(String fileName) {
File file = new File(fileName);
// 如果文件路徑所對應的文件存在,並且是一個文件,則直接刪除
if (file.exists() && file.isFile()) {
if (file.delete()) {
return true;
} else {
return false;
}
} else {
return false;
}
}
private static boolean deleteFileByFastDFS(String fileId,FastFileStorageClient fastFileStorageClient,
String group) {
try {
String groupFieId=group+"/"+fileId;
StorePath storePath = StorePath.praseFromUrl(groupFieId);
fastFileStorageClient.deleteFile(storePath.getGroup(), storePath.getPath());
} catch (Exception e) {
log.info("[ERROR][deleteFileByFastDFS]"+e.getMessage());
return false;
}
return true;
}
/**
* 生成fileId
* @param fileName
* @return
*/
private static String renameToUUID(String fileName) {
return UUID.randomUUID() + "." + fileName.substring(fileName.lastIndexOf(".") + 1);
}
/**
* 刪除文件對象存儲
* @param fileId 文件名
* @return
*/
private static boolean deleteFileByHuaweiObject(String fileId, String huaweiEndPoint,String huaweiobsAk,String huaweiobsSk,
String bucketName){
try {
// 創建ObsClient實例
ObsClient obsClient = new ObsClient(huaweiobsAk, huaweiobsSk, huaweiEndPoint);
// 構造GetObjectRequest請求
DeleteObjectRequest getObjectRequest = new DeleteObjectRequest(bucketName, fileId);
// 執行刪除操作
obsClient.deleteObject(getObjectRequest);
// 關閉OBS客戶端
obsClient.close();
}catch (ObsException e){
log.info("[ERROR][deleteFileByHuaweiObject]"+e.getErrorMessage());
}catch (Exception e) {
log.info("[ERROR][downloadFileByHuaweiObject]"+e.getMessage());
}
return true;
}
/**
* 文件數據輸出(image)
* @param fileId
* @param bytes
* @param res
*/
public static void fileDataOut(String fileId, byte[] bytes, HttpServletResponse res){
String[] prefixArray = fileId.split("\\.");
String prefix=prefixArray[1];
File file =new File(fileId);
res.reset();
res.setCharacterEncoding("utf-8");
// res.setHeader("content-type", "");
res.addHeader("Content-Length", "" + bytes.length);
if(prefix.equals("svg"))
prefix ="svg+xml";
res.setContentType("image/"+prefix);
res.setHeader("Accept-Ranges","bytes");
OutputStream os = null;
try {
// os = res.getOutputStream();
// os.write(bytes);
// os.flush();
res.getOutputStream().write(bytes);
} catch (IOException e) {
System.out.println("not find img..");
} finally {
if (os != null) {
try {
os.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}
參考資料
假設個人實戰使用可以採用docker快速安裝,但是未安裝過FastDFS建議普通安裝部署,瞭解一下tracker和storage的使用,以及部署搭建集群。
FastDFS普通安裝部署:https://www.cnblogs.com/chenliugou/p/15322389.html
OpenFeign和FastDFS的類衝突:https://www.cnblogs.com/chenliugou/p/18113183
Gitee地址:https://gitee.com/chen-liugou/file/new/master?readme=true