mirror of https://github.com/halo-dev/halo
Add thumbnail generation router for image uploads
John Niang
parent
b1127e3c28
commit
503e9b8c7f
|
|
@ -0,0 +1,156 @@
|
|||
package run.halo.app.core.attachment.endpoint;
|
||||
|
||||
import static java.nio.file.StandardOpenOption.CREATE;
|
||||
import static java.nio.file.StandardOpenOption.READ;
|
||||
import static java.nio.file.StandardOpenOption.TRUNCATE_EXISTING;
|
||||
import static java.nio.file.StandardOpenOption.WRITE;
|
||||
import static org.imgscalr.Scalr.Method.AUTOMATIC;
|
||||
import static org.imgscalr.Scalr.Mode.FIT_TO_WIDTH;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import javax.imageio.ImageIO;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.imgscalr.Scalr;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.core.io.FileSystemResource;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.web.reactive.function.server.RouterFunction;
|
||||
import org.springframework.web.reactive.function.server.RouterFunctions;
|
||||
import org.springframework.web.reactive.function.server.ServerResponse;
|
||||
import org.springframework.web.reactive.resource.NoResourceFoundException;
|
||||
import reactor.core.publisher.Mono;
|
||||
import reactor.core.scheduler.Scheduler;
|
||||
import reactor.core.scheduler.Schedulers;
|
||||
import run.halo.app.core.attachment.AttachmentRootGetter;
|
||||
import run.halo.app.core.attachment.ThumbnailSize;
|
||||
|
||||
@Slf4j
|
||||
@Component
|
||||
class ThumbnailRouters {
|
||||
|
||||
private final Path uploadRoot;
|
||||
|
||||
private final Path thumbnailRoot;
|
||||
|
||||
private final Scheduler thumbnailGeneratingScheduler;
|
||||
|
||||
public ThumbnailRouters(AttachmentRootGetter attachmentRoot) {
|
||||
this.uploadRoot = attachmentRoot.get().resolve("upload");
|
||||
this.thumbnailRoot = attachmentRoot.get().resolve("thumbnails");
|
||||
this.thumbnailGeneratingScheduler = Schedulers.newBoundedElastic(
|
||||
// Currently, we only allow 10 concurrent thumbnail generations
|
||||
10,
|
||||
Schedulers.DEFAULT_BOUNDED_ELASTIC_QUEUESIZE,
|
||||
"thumbnail-generator-"
|
||||
);
|
||||
}
|
||||
|
||||
@Bean
|
||||
RouterFunction<ServerResponse> thumbnailRouter() {
|
||||
return RouterFunctions.route()
|
||||
.GET("/thumbnails/w{width}/upload/{*filename}", serverRequest -> {
|
||||
var width = serverRequest.pathVariable("width");
|
||||
String originalFilename = serverRequest.pathVariable("filename");
|
||||
var fileName = StringUtils.removeStart(originalFilename, "/");
|
||||
if (StringUtils.isBlank(fileName)) {
|
||||
log.trace("Filename is blank");
|
||||
return Mono.error(new NoResourceFoundException(fileName));
|
||||
}
|
||||
var size = ThumbnailSize.fromWidth(width);
|
||||
// try to resolve the thumbnail
|
||||
// build thumbnail path
|
||||
var thumbnailPath = thumbnailRoot.resolve("w" + size.getWidth())
|
||||
.resolve(fileName);
|
||||
var thumbnailResource = new FileSystemResource(thumbnailPath);
|
||||
if (thumbnailResource.isReadable()) {
|
||||
return ServerResponse.ok().bodyValue(thumbnailResource);
|
||||
}
|
||||
// generate for the attachment
|
||||
return Mono.fromCallable(
|
||||
() -> {
|
||||
var uploadPath = uploadRoot.resolve(fileName);
|
||||
return generateThumbnail(uploadPath, thumbnailPath, size);
|
||||
})
|
||||
.subscribeOn(this.thumbnailGeneratingScheduler)
|
||||
.switchIfEmpty(Mono.error(() -> new NoResourceFoundException(fileName)))
|
||||
.map(FileSystemResource::new)
|
||||
.flatMap(resource -> ServerResponse.ok().bodyValue(resource));
|
||||
})
|
||||
.build();
|
||||
}
|
||||
|
||||
private Path generateThumbnail(Path attachmentPath, Path thumbnailPath, ThumbnailSize size) {
|
||||
if (!Files.exists(attachmentPath)) {
|
||||
log.trace("Attachment path does not exist: {}", attachmentPath);
|
||||
return null;
|
||||
}
|
||||
if (Files.exists(thumbnailPath)) {
|
||||
return thumbnailPath;
|
||||
}
|
||||
var attachmentPathString = attachmentPath.toString();
|
||||
synchronized (attachmentPathString) {
|
||||
// double check
|
||||
if (Files.exists(thumbnailPath)) {
|
||||
return thumbnailPath;
|
||||
}
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("Generating thumbnail for path: {}, target: {}, size: {}",
|
||||
attachmentPath, thumbnailPath, size);
|
||||
}
|
||||
var formatName = getFormatName(attachmentPath);
|
||||
if (formatName == null) {
|
||||
log.info("Cannot determine image format for path: {}", attachmentPath);
|
||||
return null;
|
||||
}
|
||||
try (
|
||||
var inputStream = Files.newInputStream(attachmentPath, READ);
|
||||
) {
|
||||
var bufferedImage = ImageIO.read(inputStream);
|
||||
if (bufferedImage == null) {
|
||||
return null;
|
||||
}
|
||||
if (bufferedImage.getWidth() <= size.getWidth()) {
|
||||
// if the image is smaller than the thumbnail size, just copy it
|
||||
Files.createDirectories(thumbnailPath.getParent());
|
||||
Files.copy(attachmentPath, thumbnailPath);
|
||||
return thumbnailPath;
|
||||
}
|
||||
// TODO Handle image orientation
|
||||
var thumbnailBufferedImage =
|
||||
Scalr.resize(bufferedImage, AUTOMATIC, FIT_TO_WIDTH, size.getWidth());
|
||||
Files.createDirectories(thumbnailPath.getParent());
|
||||
try (var outputStream =
|
||||
Files.newOutputStream(thumbnailPath, CREATE, TRUNCATE_EXISTING, WRITE)
|
||||
) {
|
||||
ImageIO.write(thumbnailBufferedImage, formatName, outputStream);
|
||||
}
|
||||
log.info("Generated thumbnail for path: {}, target: {}, size: {}",
|
||||
attachmentPath, thumbnailPath, size);
|
||||
return thumbnailPath;
|
||||
} catch (IOException e) {
|
||||
log.warn("Failed to generate thumbnail for path: {}", attachmentPath, e);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private String getFormatName(Path imagePath) {
|
||||
try (var imageInputStream = ImageIO.createImageInputStream(
|
||||
Files.newInputStream(imagePath))
|
||||
) {
|
||||
var readers = ImageIO.getImageReaders(imageInputStream);
|
||||
if (!readers.hasNext()) {
|
||||
return null;
|
||||
}
|
||||
var reader = readers.next();
|
||||
return reader.getFormatName();
|
||||
} catch (IOException e) {
|
||||
log.warn("Failed to get image format for path: {}", imagePath, e);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue