1
This commit is contained in:
334
src/main/java/cn/van/business/service/ImageConvertService.java
Normal file
334
src/main/java/cn/van/business/service/ImageConvertService.java
Normal file
@@ -0,0 +1,334 @@
|
||||
package cn.van.business.service;
|
||||
|
||||
import cn.van.business.model.pl.ImageConversion;
|
||||
import cn.van.business.repository.ImageConversionRepository;
|
||||
import cn.hutool.core.io.FileUtil;
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import cn.hutool.http.HttpUtil;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import javax.imageio.ImageIO;
|
||||
import javax.imageio.ImageReader;
|
||||
import javax.imageio.stream.ImageInputStream;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.IOException;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.nio.file.Paths;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.*;
|
||||
|
||||
/**
|
||||
* WebP图片IO支持
|
||||
* 注册WebP图片格式的读写器
|
||||
*/
|
||||
class WebPImageIO {
|
||||
private static final org.slf4j.Logger log = org.slf4j.LoggerFactory.getLogger(WebPImageIO.class);
|
||||
private static volatile boolean initialized = false;
|
||||
|
||||
/**
|
||||
* 初始化WebP支持
|
||||
* webp-imageio库在类加载时会通过ServiceLoader自动注册到ImageIO
|
||||
*/
|
||||
static synchronized void init() {
|
||||
if (initialized) {
|
||||
return;
|
||||
}
|
||||
|
||||
// 检查是否支持webp格式
|
||||
try {
|
||||
Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName("webp");
|
||||
if (readers.hasNext()) {
|
||||
log.debug("WebP图片格式支持已启用");
|
||||
} else {
|
||||
log.warn("未检测到WebP图片格式支持，webp转换可能失败。请确认webp-imageio依赖已正确加载");
|
||||
}
|
||||
} catch (Exception e) {
|
||||
log.warn("检查WebP支持时出错: {}", e.getMessage());
|
||||
}
|
||||
|
||||
initialized = true;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 图片转换服务类
|
||||
* 负责将webp格式的图片转换为jpg格式，并缓存转换结果
|
||||
*
|
||||
* @author System
|
||||
*/
|
||||
@Slf4j
|
||||
@Service
|
||||
public class ImageConvertService {
|
||||
|
||||
@Autowired
|
||||
private ImageConversionRepository imageConversionRepository;
|
||||
|
||||
/**
|
||||
* 图片存储根目录，默认使用系统临时目录
|
||||
*/
|
||||
@Value("${image.convert.storage-path:${java.io.tmpdir}/comment-images}")
|
||||
private String storagePath;
|
||||
|
||||
/**
|
||||
* 图片访问基础URL，用于生成转换后图片的访问地址
|
||||
* 如果为空，则返回本地文件路径
|
||||
*/
|
||||
@Value("${image.convert.base-url:}")
|
||||
private String baseUrl;
|
||||
|
||||
/**
|
||||
* 转换图片URL列表，将webp格式转换为jpg
|
||||
*
|
||||
* @param imageUrls 原始图片URL列表
|
||||
* @return 转换后的图片URL列表（webp已转换为jpg，其他格式保持不变）
|
||||
*/
|
||||
public List<String> convertImageUrls(List<String> imageUrls) {
|
||||
if (imageUrls == null || imageUrls.isEmpty()) {
|
||||
return Collections.emptyList();
|
||||
}
|
||||
|
||||
List<String> convertedUrls = new ArrayList<>();
|
||||
for (String imageUrl : imageUrls) {
|
||||
if (StrUtil.isBlank(imageUrl)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
try {
|
||||
String convertedUrl = convertImageUrl(imageUrl);
|
||||
convertedUrls.add(convertedUrl);
|
||||
} catch (Exception e) {
|
||||
log.warn("图片转换失败，使用原URL: {}, 错误: {}", imageUrl, e.getMessage());
|
||||
// 转换失败时使用原URL
|
||||
convertedUrls.add(imageUrl);
|
||||
}
|
||||
}
|
||||
|
||||
return convertedUrls;
|
||||
}
|
||||
|
||||
/**
|
||||
* 转换单个图片URL
|
||||
*
|
||||
* @param originalUrl 原始图片URL
|
||||
* @return 转换后的图片URL（如果不需要转换则返回原URL）
|
||||
*/
|
||||
private String convertImageUrl(String originalUrl) {
|
||||
if (StrUtil.isBlank(originalUrl)) {
|
||||
return originalUrl;
|
||||
}
|
||||
|
||||
// 检查是否为webp格式
|
||||
if (!isWebpFormat(originalUrl)) {
|
||||
return originalUrl;
|
||||
}
|
||||
|
||||
// 检查是否已转换
|
||||
Optional<ImageConversion> existing = imageConversionRepository.findByOriginalUrl(originalUrl);
|
||||
if (existing.isPresent()) {
|
||||
ImageConversion conversion = existing.get();
|
||||
log.debug("使用缓存的转换结果: {} -> {}", originalUrl, conversion.getConvertedUrl());
|
||||
return conversion.getConvertedUrl();
|
||||
}
|
||||
|
||||
// 执行转换
|
||||
try {
|
||||
String convertedUrl = performConversion(originalUrl);
|
||||
log.info("图片转换成功: {} -> {}", originalUrl, convertedUrl);
|
||||
return convertedUrl;
|
||||
} catch (Exception e) {
|
||||
log.error("图片转换失败: {}, 错误: {}", originalUrl, e.getMessage(), e);
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查URL是否为webp格式
|
||||
*
|
||||
* @param url 图片URL
|
||||
* @return 是否为webp格式
|
||||
*/
|
||||
private boolean isWebpFormat(String url) {
|
||||
if (StrUtil.isBlank(url)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// 检查URL中是否包含.webp
|
||||
String lowerUrl = url.toLowerCase();
|
||||
// 检查URL参数或路径中是否包含webp
|
||||
return lowerUrl.contains(".webp") ||
|
||||
lowerUrl.contains("format=webp") ||
|
||||
lowerUrl.contains("?webp") ||
|
||||
lowerUrl.contains("&webp");
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行图片转换
|
||||
*
|
||||
* @param originalUrl 原始webp图片URL
|
||||
* @return 转换后的jpg图片URL或路径
|
||||
*/
|
||||
private String performConversion(String originalUrl) throws IOException {
|
||||
// 确保存储目录存在
|
||||
Path storageDir = Paths.get(storagePath);
|
||||
if (!Files.exists(storageDir)) {
|
||||
Files.createDirectories(storageDir);
|
||||
}
|
||||
|
||||
// 生成转换后的文件名（基于原URL的MD5）
|
||||
String fileName = generateFileName(originalUrl);
|
||||
Path outputPath = storageDir.resolve(fileName);
|
||||
|
||||
// 如果转换后的文件已存在，直接返回
|
||||
if (Files.exists(outputPath)) {
|
||||
String convertedUrl = generateConvertedUrl(fileName);
|
||||
// 如果数据库中没有记录，保存记录
|
||||
if (!imageConversionRepository.existsByOriginalUrl(originalUrl)) {
|
||||
saveConversionRecord(originalUrl, convertedUrl, Files.size(outputPath));
|
||||
}
|
||||
return convertedUrl;
|
||||
}
|
||||
|
||||
// 下载原始图片
|
||||
byte[] imageData = downloadImage(originalUrl);
|
||||
if (imageData == null || imageData.length == 0) {
|
||||
throw new IOException("下载图片失败或图片数据为空: " + originalUrl);
|
||||
}
|
||||
|
||||
// 转换图片格式
|
||||
BufferedImage bufferedImage = null;
|
||||
try {
|
||||
// 初始化WebP支持
|
||||
WebPImageIO.init();
|
||||
|
||||
// 尝试使用ImageIO读取
|
||||
BufferedImage tempImage = null;
|
||||
try (ByteArrayInputStream bais = new ByteArrayInputStream(imageData)) {
|
||||
tempImage = ImageIO.read(bais);
|
||||
}
|
||||
|
||||
if (tempImage != null) {
|
||||
bufferedImage = tempImage;
|
||||
} else {
|
||||
// 如果ImageIO无法读取，尝试使用WebP特定的读取器
|
||||
Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName("webp");
|
||||
if (readers.hasNext()) {
|
||||
ImageReader reader = readers.next();
|
||||
try (ByteArrayInputStream bais = new ByteArrayInputStream(imageData);
|
||||
ImageInputStream iis = ImageIO.createImageInputStream(bais)) {
|
||||
reader.setInput(iis);
|
||||
bufferedImage = reader.read(0);
|
||||
} finally {
|
||||
reader.dispose();
|
||||
}
|
||||
}
|
||||
|
||||
if (bufferedImage == null) {
|
||||
throw new IOException("无法读取webp图片格式，请确认已添加webp-imageio依赖");
|
||||
}
|
||||
}
|
||||
|
||||
// 如果是RGBA格式，转换为RGB
|
||||
if (bufferedImage.getType() == BufferedImage.TYPE_4BYTE_ABGR ||
|
||||
bufferedImage.getType() == BufferedImage.TYPE_INT_ARGB) {
|
||||
BufferedImage rgbImage = new BufferedImage(
|
||||
bufferedImage.getWidth(),
|
||||
bufferedImage.getHeight(),
|
||||
BufferedImage.TYPE_INT_RGB
|
||||
);
|
||||
rgbImage.createGraphics().drawImage(bufferedImage, 0, 0, null);
|
||||
bufferedImage = rgbImage;
|
||||
}
|
||||
|
||||
// 保存为jpg格式
|
||||
boolean success = ImageIO.write(bufferedImage, "jpg", outputPath.toFile());
|
||||
if (!success) {
|
||||
throw new IOException("无法写入jpg格式");
|
||||
}
|
||||
|
||||
String convertedUrl = generateConvertedUrl(fileName);
|
||||
long fileSize = Files.size(outputPath);
|
||||
|
||||
// 保存转换记录
|
||||
saveConversionRecord(originalUrl, convertedUrl, fileSize);
|
||||
|
||||
log.info("图片转换完成: {} -> {} (大小: {} bytes)", originalUrl, convertedUrl, fileSize);
|
||||
return convertedUrl;
|
||||
|
||||
} finally {
|
||||
if (bufferedImage != null) {
|
||||
bufferedImage.flush();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 下载图片
|
||||
*
|
||||
* @param url 图片URL
|
||||
* @return 图片字节数组
|
||||
*/
|
||||
private byte[] downloadImage(String url) {
|
||||
try {
|
||||
return HttpUtil.downloadBytes(url);
|
||||
} catch (Exception e) {
|
||||
log.error("下载图片失败: {}, 错误: {}", url, e.getMessage());
|
||||
throw new RuntimeException("下载图片失败: " + e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成文件名（基于原URL的MD5）
|
||||
*
|
||||
* @param originalUrl 原始URL
|
||||
* @return 文件名（不含扩展名）
|
||||
*/
|
||||
private String generateFileName(String originalUrl) {
|
||||
// 使用URL的MD5作为文件名，避免重复和特殊字符问题
|
||||
String md5 = cn.hutool.core.util.HashUtil.md5Hex(originalUrl);
|
||||
return md5 + ".jpg";
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成转换后的URL
|
||||
*
|
||||
* @param fileName 文件名
|
||||
* @return 转换后的URL或本地路径
|
||||
*/
|
||||
private String generateConvertedUrl(String fileName) {
|
||||
if (StrUtil.isNotBlank(baseUrl)) {
|
||||
// 如果配置了baseUrl，返回HTTP访问地址
|
||||
return baseUrl.endsWith("/") ? baseUrl + fileName : baseUrl + "/" + fileName;
|
||||
} else {
|
||||
// 否则返回本地文件路径
|
||||
return Paths.get(storagePath, fileName).toString();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存转换记录
|
||||
*
|
||||
* @param originalUrl 原始URL
|
||||
* @param convertedUrl 转换后的URL
|
||||
* @param fileSize 文件大小
|
||||
*/
|
||||
private void saveConversionRecord(String originalUrl, String convertedUrl, long fileSize) {
|
||||
ImageConversion conversion = new ImageConversion();
|
||||
conversion.setOriginalUrl(originalUrl);
|
||||
conversion.setConvertedUrl(convertedUrl);
|
||||
conversion.setFileSize(fileSize);
|
||||
conversion.setConvertedAt(LocalDateTime.now());
|
||||
|
||||
try {
|
||||
imageConversionRepository.save(conversion);
|
||||
} catch (Exception e) {
|
||||
log.warn("保存转换记录失败: {}, 错误: {}", originalUrl, e.getMessage());
|
||||
// 不抛出异常，因为转换本身已成功
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user