add TempFileService

src/main/java/com/imyeyu/api/modules/common/service/TempFileService.java

@@ -0,0 +1,22 @@
+package com.imyeyu.api.modules.common.service;
+
+import com.imyeyu.api.modules.common.bean.TempFileMetaData;
+import com.imyeyu.api.modules.common.vo.TempFileResponse;
+import com.imyeyu.java.bean.timi.TimiException;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.File;
+import java.util.List;
+
+/**
+ * @author 夜雨
+ * @since 2025-09-27 01:35
+ */
+public interface TempFileService {
+
+	List<TempFileResponse> store(List<MultipartFile> files) throws TimiException;
+
+	File get(String id) throws TimiException;
+
+	TempFileMetaData metadata(String id) throws TimiException;
+}

src/main/java/com/imyeyu/api/modules/common/service/implement/TempFileServiceImplement.java

@@ -0,0 +1,132 @@
+package com.imyeyu.api.modules.common.service.implement;
+
+import com.imyeyu.api.modules.common.bean.SettingKey;
+import com.imyeyu.api.modules.common.bean.TempFileMetaData;
+import com.imyeyu.api.modules.common.service.AttachmentService;
+import com.imyeyu.api.modules.common.service.SettingService;
+import com.imyeyu.api.modules.common.service.TempFileService;
+import com.imyeyu.api.modules.common.vo.TempFileResponse;
+import com.imyeyu.io.IO;
+import com.imyeyu.io.IOSize;
+import com.imyeyu.java.TimiJava;
+import com.imyeyu.java.bean.timi.TimiCode;
+import com.imyeyu.java.bean.timi.TimiException;
+import com.imyeyu.network.Network;
+import com.imyeyu.spring.TimiSpring;
+import com.imyeyu.utils.Time;
+import jakarta.annotation.PostConstruct;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * @author 夜雨
+ * @since 2025-09-27 01:47
+ */
+@Slf4j
+@Service
+@RequiredArgsConstructor
+public class TempFileServiceImplement implements TempFileService {
+
+	private static final Long TTL = Time.H * 6;
+	private static final Long LIMIT = IOSize.GB * 10;
+
+	private final SettingService settingService;
+	private final AttachmentService attachmentService;
+
+	private Path storagePath;
+
+	private final Map<String, TempFileMetaData> metadataMap = new ConcurrentHashMap<>();
+	private final Map<String, Long> usageMap = new ConcurrentHashMap<>();
+
+	@PostConstruct
+	public void init() throws IOException {
+		storagePath = Paths.get(settingService.getAsString(SettingKey.TEMP_FILE_PATH));
+		IO.destroy(storagePath.toFile());
+		Files.createDirectories(storagePath);
+	}
+
+	public List<TempFileResponse> store(List<MultipartFile> files) throws TimiException {
+		long newFileSize = files.stream().mapToLong(MultipartFile::getSize).sum();
+		long currentUsage = usageMap.getOrDefault(TimiSpring.getRequestIP(), 0L);
+		TimiException.requiredTrue(currentUsage + newFileSize < LIMIT, "out of storage limit(10 GB)");
+
+		try {
+			List<TempFileResponse> result = new ArrayList<>();
+			for (int i = 0; i < files.size(); i++) {
+				MultipartFile file = files.get(i);
+
+				String fileId = UUID.randomUUID().toString();
+				String fileName = fileId;
+				if (TimiJava.isNotEmpty(file.getOriginalFilename())) {
+					fileName = fileId + "." + Network.uriFileExtension(file.getOriginalFilename());
+				}
+				Path filePath = storagePath.resolve(fileName);
+				Files.copy(file.getInputStream(), filePath);
+
+				// 创建元数据
+				TempFileMetaData metadata = new TempFileMetaData();
+				metadata.setId(fileId);
+				metadata.setPath(filePath);
+				metadata.setName(fileName);
+				metadata.setOriginalName(file.getOriginalFilename());
+				metadata.setLastAccessAt(Time.now());
+				metadataMap.put(metadata.getId(), metadata);
+
+				TempFileResponse resp = new TempFileResponse();
+				resp.setId(metadata.getId());
+				resp.setExpireAt(metadata.getLastAccessAt() + TTL);
+
+				usageMap.put(TimiSpring.getRequestIP(), currentUsage + newFileSize);
+				result.add(resp);
+			}
+			return result;
+		} catch (Exception e) {
+			log.error("store temp file error", e);
+			throw new TimiException(TimiCode.ERROR, "store temp file error", e);
+		}
+	}
+
+	@Override
+	public File get(String id) throws TimiException {
+		TempFileMetaData metaData = metadataMap.get(id);
+		TimiException.required(metaData, "not found temp file");
+		metaData.setLastAccessAt(Time.now());
+		return metaData.getPath().toFile();
+	}
+
+	@Override
+	public TempFileMetaData metadata(String id) throws TimiException {
+		return metadataMap.get(id);
+	}
+
+	@Scheduled(fixedRate = 3600000)
+	public void cleanup() {
+		List<String> expiredIds = metadataMap.values()
+		                                     .stream()
+		                                     .filter(metadata -> metadata.getLastAccessAt() + TTL < Time.now())
+		                                     .map(TempFileMetaData::getId)
+		                                     .toList();
+		for (int i = 0; i < expiredIds.size(); i++) {
+			TempFileMetaData removed = metadataMap.remove(expiredIds.get(i));
+			if (TimiJava.isNotEmpty(removed)) {
+				File file = removed.getPath().toFile();
+				IO.destroy(file);
+				usageMap.computeIfPresent(TimiSpring.getRequestIP(), (ip, usage) -> usage - file.length());
+			}
+		}
+	}
+}