后端架构更新

2026-04-27 09:18:10 +08:00
parent 0caf62c3d2
commit 995e2ee65a
141 changed files with 6957 additions and 772 deletions
@@ -0,0 +1,85 @@
+package com.nanri.aiimage.modules.file.service.object;
+
+import com.nanri.aiimage.config.TransientStorageProperties;
+import io.minio.GetObjectArgs;
+import io.minio.MinioClient;
+import io.minio.PutObjectArgs;
+import io.minio.RemoveObjectArgs;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Service;
+
+import java.io.ByteArrayInputStream;
+import java.nio.charset.StandardCharsets;
+import java.util.Objects;
+
+@Service
+@RequiredArgsConstructor
+public class RustfsObjectStorageService {
+
+    private final TransientStorageProperties properties;
+
+    public boolean isConfigured() {
+        return notBlank(properties.getEndpoint())
+                && notBlank(properties.getBucket())
+                && notBlank(properties.getAccessKeyId())
+                && notBlank(properties.getAccessKeySecret());
+    }
+
+    public String uploadText(String objectKey, String content) {
+        if (!isConfigured()) {
+            throw new IllegalStateException("transient storage is not configured");
+        }
+        try (ByteArrayInputStream stream = new ByteArrayInputStream(
+                Objects.requireNonNullElse(content, "").getBytes(StandardCharsets.UTF_8))) {
+            buildClient().putObject(PutObjectArgs.builder()
+                    .bucket(properties.getBucket())
+                    .object(objectKey)
+                    .stream(stream, stream.available(), -1)
+                    .contentType("application/json")
+                    .build());
+            return objectKey;
+        } catch (Exception ex) {
+            throw new IllegalStateException("failed to upload payload to transient storage", ex);
+        }
+    }
+
+    public String readObjectAsString(String objectKey) {
+        if (!isConfigured()) {
+            throw new IllegalStateException("transient storage is not configured");
+        }
+        try (var stream = buildClient().getObject(GetObjectArgs.builder()
+                .bucket(properties.getBucket())
+                .object(objectKey)
+                .build())) {
+            return new String(stream.readAllBytes(), StandardCharsets.UTF_8);
+        } catch (Exception ex) {
+            throw new IllegalStateException("failed to read payload from transient storage", ex);
+        }
+    }
+
+    public void deleteObject(String objectKey) {
+        if (!isConfigured() || !notBlank(objectKey)) {
+            return;
+        }
+        try {
+            buildClient().removeObject(RemoveObjectArgs.builder()
+                    .bucket(properties.getBucket())
+                    .object(objectKey)
+                    .build());
+        } catch (Exception ex) {
+            throw new IllegalStateException("failed to delete payload from transient storage", ex);
+        }
+    }
+
+    private MinioClient buildClient() {
+        return MinioClient.builder()
+                .endpoint(properties.getEndpoint())
+                .credentials(properties.getAccessKeyId(), properties.getAccessKeySecret())
+                .region(properties.getRegion())
+                .build();
+    }
+
+    private boolean notBlank(String value) {
+        return value != null && !value.isBlank();
+    }
+}