Blizzard
ec9431bfa4
多机/容器化第一时间断裂的隐患(ARCHITECTURE_REVIEW §7 #1):此前报告 源(reportStore)与 .docx 产物(reportExport/reportRender)都写 mcp-go 本地 SUNDYNIX_REPORTS_DIR,gateway 再按同一路径 c.File 回流——mcp-go 与 gateway 不共享磁盘即断。 - blob 包从 gateway/internal 提到 sundynix-shared/blob,gateway 与 mcp-go 共用同一 MinIO;新增 PutBytes/GetBytes 走二进制(.docx)。 - mcp-go 注入 blob:report 源/产物优先 Put 到对象存储(键 reports/<id>.{json,docx}), 导出结果返回 minio://<key>;MinIO 未就绪回退本地盘(单机降级,getSource 兼容旧本地报告)。 - gateway ExportReport 识别 minio:// → GetBytes 流式下载,否则 c.File 本地降级。 - 测试:mcp 本地回落往返单测 + shared/blob 真 MinIO 二进制往返(BLOB_TEST_ENDPOINT 门控)。 四模块 build 绿,受影响测试全过。 - 附带:go mod tidy 引入 genproto 拆包冲突,四模块统一钉单体 genproto 新版解决 (注:勿 go work sync,会剪掉 indirect require 回落旧版重现冲突)。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
92 lines
3.1 KiB
Go
92 lines
3.1 KiB
Go
// Package blob 封装对象存储（MinIO）：大文档正文、报告源/产物落对象存储，
|
|
// 让多进程/多机不再依赖共享本地盘。gateway 与 mcp-go 共用此包（同一 MinIO）。
|
|
package blob
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/minio/minio-go/v7"
|
|
"github.com/minio/minio-go/v7/pkg/credentials"
|
|
)
|
|
|
|
// Store 是对象存储句柄；cli 为 nil 表示降级（未连上 MinIO，回退本地盘/内联）。
|
|
type Store struct {
|
|
cli *minio.Client
|
|
bucket string
|
|
}
|
|
|
|
// Open 连接 MinIO 并确保 bucket 存在。连接失败返回降级实例（cli=nil），不阻断服务启动。
|
|
func Open(endpoint, accessKey, secretKey, bucket string) *Store {
|
|
cli, err := minio.New(endpoint, &minio.Options{
|
|
Creds: credentials.NewStaticV4(accessKey, secretKey, ""),
|
|
Secure: false,
|
|
})
|
|
if err != nil {
|
|
log.Printf("[blob] MinIO 不可用，回退本地盘/内联: %v", err)
|
|
return &Store{}
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
exists, err := cli.BucketExists(ctx, bucket)
|
|
if err != nil {
|
|
log.Printf("[blob] MinIO 连接失败，回退本地盘/内联: %v", err)
|
|
return &Store{}
|
|
}
|
|
if !exists {
|
|
if err := cli.MakeBucket(ctx, bucket, minio.MakeBucketOptions{}); err != nil {
|
|
log.Printf("[blob] 建 bucket 失败，回退本地盘/内联: %v", err)
|
|
return &Store{}
|
|
}
|
|
}
|
|
log.Printf("[blob] MinIO connected %s bucket=%s", endpoint, bucket)
|
|
return &Store{cli: cli, bucket: bucket}
|
|
}
|
|
|
|
// Ready 报告对象存储是否可用。
|
|
func (s *Store) Ready() bool { return s != nil && s.cli != nil }
|
|
|
|
// Put 写入一段文本到对象键 key。
|
|
func (s *Store) Put(ctx context.Context, key, content string) error {
|
|
r := bytes.NewReader([]byte(content))
|
|
_, err := s.cli.PutObject(ctx, s.bucket, key, r, int64(len(content)), minio.PutObjectOptions{ContentType: "text/plain; charset=utf-8"})
|
|
return err
|
|
}
|
|
|
|
// PutBytes 写入二进制对象（如 .docx），可指定 Content-Type。
|
|
func (s *Store) PutBytes(ctx context.Context, key string, data []byte, contentType string) error {
|
|
if contentType == "" {
|
|
contentType = "application/octet-stream"
|
|
}
|
|
r := bytes.NewReader(data)
|
|
_, err := s.cli.PutObject(ctx, s.bucket, key, r, int64(len(data)), minio.PutObjectOptions{ContentType: contentType})
|
|
return err
|
|
}
|
|
|
|
// Get 读回对象键 key 的全部文本。
|
|
func (s *Store) Get(ctx context.Context, key string) (string, error) {
|
|
b, err := s.GetBytes(ctx, key)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(b), nil
|
|
}
|
|
|
|
// GetBytes 读回对象键 key 的原始字节（供二进制产物流式下载）。
|
|
func (s *Store) GetBytes(ctx context.Context, key string) ([]byte, error) {
|
|
obj, err := s.cli.GetObject(ctx, s.bucket, key, minio.GetObjectOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer obj.Close()
|
|
return io.ReadAll(obj)
|
|
}
|
|
|
|
// Delete 删除对象；返回错误供级联删按需失败闭合（其它 best-effort 调用点可忽略返回值）。
|
|
func (s *Store) Delete(ctx context.Context, key string) error {
|
|
return s.cli.RemoveObject(ctx, s.bucket, key, minio.RemoveObjectOptions{})
|
|
}
|