82 lines
2.9 KiB
Java
82 lines
2.9 KiB
Java
package net.shapelight.common.config;
|
||
|
||
import io.minio.MinioClient;
|
||
import io.minio.PutObjectOptions;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.stereotype.Component;
|
||
|
||
import java.io.ByteArrayInputStream;
|
||
import java.io.ByteArrayOutputStream;
|
||
import java.io.IOException;
|
||
import java.io.InputStream;
|
||
import java.util.Base64;
|
||
|
||
@Slf4j
|
||
@Component
|
||
public class MinioUtils {
|
||
@Autowired
|
||
private MinioConfig minioConfig;
|
||
@Autowired
|
||
private MinioClient minioClient;
|
||
|
||
public String getFileBase64(String url){
|
||
String base64Image = "";
|
||
InputStream inStream = null;
|
||
ByteArrayOutputStream outStream = null;
|
||
try {
|
||
minioClient.statObject(minioConfig.getBucketName(), url);
|
||
inStream = minioClient.getObject(minioConfig.getBucketName(), url);
|
||
outStream = new ByteArrayOutputStream();
|
||
byte[] buffer = new byte[1024];
|
||
int length;
|
||
while ((length = inStream.read(buffer)) != -1) {
|
||
outStream.write(buffer, 0, length);
|
||
}
|
||
base64Image = Base64.getEncoder().encodeToString(outStream.toByteArray());
|
||
} catch (Exception e) {
|
||
log.error("Minio文件不存在：" + url);
|
||
e.printStackTrace();
|
||
} finally {
|
||
if (inStream != null) {
|
||
try {
|
||
inStream.close();
|
||
} catch (IOException e) {
|
||
log.error("inputStream close IOException:" + e.getMessage());
|
||
}
|
||
}
|
||
if (outStream != null) {
|
||
try {
|
||
outStream.close();
|
||
} catch (IOException e) {
|
||
log.error("outStream close IOException:" + e.getMessage());
|
||
}
|
||
}
|
||
}
|
||
return base64Image;
|
||
}
|
||
|
||
public boolean saveFileByBase64(String fileBase64,String fileName,String contentType){
|
||
if(fileBase64!=null && !fileBase64.isEmpty()){
|
||
try {
|
||
byte[] b = Base64.getDecoder().decode(fileBase64.replace("\n", ""));
|
||
InputStream inputStream = new ByteArrayInputStream(b);
|
||
PutObjectOptions putObjectOptions = new PutObjectOptions(b.length, -1);
|
||
// putObjectOptions.setContentType("image/jpeg");
|
||
putObjectOptions.setContentType(contentType);
|
||
minioClient.putObject(
|
||
minioConfig.getBucketName(), fileName, inputStream, putObjectOptions);
|
||
inputStream.close();
|
||
return true;
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
log.error(e.getMessage());
|
||
return false;
|
||
}
|
||
}else{
|
||
log.debug("Minio文件base64为空，不能写入");
|
||
return false;
|
||
}
|
||
}
|
||
}
|