yoyuz/my_site
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(files): expire stale upload sessions

Browse Source
This commit is contained in:
yoyuzh
2026-04-08 15:27:39 +08:00
parent 06a95bc489
commit f582e600aa
5 changed files with 63 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
backend/src/main/java/com/yoyuzh/files/UploadSessionService.java
View File

@@ -6,7 +6,9 @@ import com.yoyuzh.auth.User;
import com.yoyuzh.common.BusinessException;
import com.yoyuzh.common.ErrorCode;
import com.yoyuzh.config.FileStorageProperties;
import com.yoyuzh.files.storage.FileContentStorage;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
@@ -23,12 +25,18 @@ public class UploadSessionService {
private static final long DEFAULT_CHUNK_SIZE = 8L * 1024 * 1024;
private static final long SESSION_TTL_HOURS = 24;
private static final List<UploadSessionStatus> EXPIRABLE_STATUSES = List.of(
UploadSessionStatus.CREATED,
UploadSessionStatus.UPLOADING,
UploadSessionStatus.COMPLETING
);
private static final TypeReference<List<UploadedPart>> UPLOADED_PARTS_TYPE = new TypeReference<>() {
};
private final UploadSessionRepository uploadSessionRepository;
private final StoredFileRepository storedFileRepository;
private final FileService fileService;
private final FileContentStorage fileContentStorage;
private final ObjectMapper objectMapper = new ObjectMapper();
private final long maxFileSize;
private final Clock clock;
@@ -37,18 +45,21 @@ public class UploadSessionService {
public UploadSessionService(UploadSessionRepository uploadSessionRepository,
StoredFileRepository storedFileRepository,
FileService fileService,
FileContentStorage fileContentStorage,
FileStorageProperties properties) {
this(uploadSessionRepository, storedFileRepository, fileService, properties, Clock.systemUTC());
this(uploadSessionRepository, storedFileRepository, fileService, fileContentStorage, properties, Clock.systemUTC());
}
UploadSessionService(UploadSessionRepository uploadSessionRepository,
StoredFileRepository storedFileRepository,
FileService fileService,
FileContentStorage fileContentStorage,
FileStorageProperties properties,
Clock clock) {
this.uploadSessionRepository = uploadSessionRepository;
this.storedFileRepository = storedFileRepository;
this.fileService = fileService;
this.fileContentStorage = fileContentStorage;
this.maxFileSize = properties.getMaxFileSize();
this.clock = clock;
}
@@ -166,6 +177,29 @@ public class UploadSessionService {
}
}
@Scheduled(fixedDelay = 60 * 60 * 1000L)
@Transactional
public int pruneExpiredSessions() {
LocalDateTime now = LocalDateTime.ofInstant(clock.instant(), clock.getZone());
List<UploadSession> expiredSessions = uploadSessionRepository.findByStatusInAndExpiresAtBefore(
EXPIRABLE_STATUSES,
now
);
for (UploadSession session : expiredSessions) {
try {
fileContentStorage.deleteBlob(session.getObjectKey());
} catch (RuntimeException ignored) {
// Expiration is authoritative in the database even if remote object cleanup fails.
}
session.setStatus(UploadSessionStatus.EXPIRED);
session.setUpdatedAt(now);
}
if (!expiredSessions.isEmpty()) {
uploadSessionRepository.saveAll(expiredSessions);
}
return expiredSessions.size();
}
private void validateTarget(User user, String normalizedPath, String filename, long size) {
long effectiveMaxUploadSize = Math.min(maxFileSize, user.getMaxUploadSizeBytes());
if (size > effectiveMaxUploadSize) {