package com.ruoyi.web.controller.tool;
|
|
import com.ruoyi.common.config.RuoYiConfig;
|
import com.ruoyi.common.utils.file.FileUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.codec.binary.Base64;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.StringUtils;
|
|
import java.io.ByteArrayOutputStream;
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.InputStream;
|
import java.net.HttpURLConnection;
|
import java.net.URL;
|
import java.nio.file.Files;
|
|
@Slf4j
|
@Component
|
public class ImageUtil {
|
|
/**
|
* 将网络图片URL转为Base64编码
|
*
|
* @param imageUrl 网络图片地址
|
* @return Base64编码的图片字符串
|
*/
|
public static String imageUrlToBase64(String imageUrl) {
|
if (StringUtils.isEmpty(imageUrl)) {
|
return null;
|
}
|
|
String imagePath = RuoYiConfig.getUploadPath() + imageUrl.replace("/profile/upload", "");
|
|
InputStream inputStream = null;
|
ByteArrayOutputStream outputStream = null;
|
|
try {
|
// 1. 获取本地文件
|
File imageFile = new File(imagePath);
|
if (!imageFile.exists()) {
|
log.error("图片文件不存在：{}", imagePath);
|
return null;
|
}
|
|
// 2. 读取图片数据
|
inputStream = Files.newInputStream(imageFile.toPath());
|
outputStream = new ByteArrayOutputStream();
|
|
byte[] buffer = new byte[8192];
|
int len;
|
while ((len = inputStream.read(buffer)) != -1) {
|
outputStream.write(buffer, 0, len);
|
}
|
|
// 3. 转换为Base64
|
byte[] imageBytes = outputStream.toByteArray();
|
String base64 = Base64.encodeBase64String(imageBytes);
|
|
// 4. 根据文件扩展名添加前缀
|
String contentType = getContentType(imagePath);
|
return "data:" + contentType + ";base64," + base64;
|
|
} catch (Exception e) {
|
log.error("本地图片转Base64失败，路径：{}", imagePath, e);
|
return null;
|
} finally {
|
try {
|
if (inputStream != null) {
|
inputStream.close();
|
}
|
if (outputStream != null) {
|
outputStream.close();
|
}
|
} catch (Exception e) {
|
log.error("关闭流失败", e);
|
}
|
}
|
}
|
|
/**
|
* 根据文件扩展名获取Content-Type
|
*/
|
private static String getContentType(String imagePath) {
|
if (StringUtils.isEmpty(imagePath)) {
|
return "image/jpeg";
|
}
|
String ext = imagePath.substring(imagePath.lastIndexOf(".") + 1).toLowerCase();
|
switch (ext) {
|
case "png":
|
return "image/png";
|
case "gif":
|
return "image/gif";
|
case "bmp":
|
return "image/bmp";
|
case "webp":
|
return "image/webp";
|
case "svg":
|
return "image/svg+xml";
|
case "jpg":
|
case "jpeg":
|
default:
|
return "image/jpeg";
|
}
|
}
|
}
|
