refactor(phase1): 完成包名重构 (org.example → com.superbiz.agent)

Task 4.1: 包名统一重构 - 重命名 41 个 Java 文件的包名 - 更新所有 import 语句 - 恢复枚举类（FaultCategory、DiagnosisStatus、SourceType） - 更新测试类的 import 重构范围： - domain/entity: 3 个实体类 - domain/model: 2 个数据类 - domain/enums: 3 个枚举类 - repository: 3 个接口 - service/session: 2 个类（接口 + 实现） - config: 9 个配置类 - controller: 2 个控制器 - agent/tool: 4 个工具类 - client: 1 个客户端 - Main.java: 主类验证结果： - 编译成功，无错误 - 所有测试通过 (27/27) - ApiDocumentRepositoryTest: 7/7 ✅ - CaseLibraryRepositoryTest: 6/6 ✅ - DiagnosisRecordRepositoryTest: 6/6 ✅ - RedisSessionManagerTest: 8/8 ✅ Progress: 21/33 tasks completed (64%)
2026-06-23 14:56:04 +08:00
parent 8bd758dbaf
commit c3a232540a
59 changed files with 103 additions and 104 deletions
@@ -0,0 +1,196 @@
+package com.superbiz.agent.client;
+
+import io.milvus.client.MilvusServiceClient;
+import io.milvus.grpc.DataType;
+import io.milvus.param.ConnectParam;
+import io.milvus.param.IndexType;
+import io.milvus.param.MetricType;
+import io.milvus.param.R;
+import io.milvus.param.RpcStatus;
+import io.milvus.param.collection.*;
+import io.milvus.param.index.CreateIndexParam;
+import com.superbiz.agent.config.MilvusProperties;
+import com.superbiz.agent.constant.MilvusConstants;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.concurrent.TimeUnit;
+
+/**
+ * Milvus 客户端工厂类
+ * 负责创建和初始化 Milvus 客户端连接
+ */
+@Component
+public class MilvusClientFactory {
+
+    private static final Logger logger = LoggerFactory.getLogger(MilvusClientFactory.class);
+
+    @Autowired
+    private MilvusProperties milvusProperties;
+
+    /**
+     * 创建并初始化 Milvus 客户端
+     * 
+     * 简化版本：直接连接并创建 collection
+     * 
+     * @return MilvusServiceClient 实例
+     * @throws RuntimeException 如果连接或初始化失败
+     */
+    public MilvusServiceClient createClient() {
+        MilvusServiceClient client = null;
+
+        try {
+            // 1. 连接到 Milvus
+            logger.info("正在连接到 Milvus: {}:{}", milvusProperties.getHost(), milvusProperties.getPort());
+            client = connectToMilvus();
+            logger.info("成功连接到 Milvus");
+
+            // 2. 检查并创建 biz collection（如果不存在）
+            if (!collectionExists(client, MilvusConstants.MILVUS_COLLECTION_NAME)) {
+                logger.info("collection '{}' 不存在，正在创建...", MilvusConstants.MILVUS_COLLECTION_NAME);
+                createBizCollection(client);
+                logger.info("成功创建 collection '{}'", MilvusConstants.MILVUS_COLLECTION_NAME);
+                
+                // 创建索引
+                createIndexes(client);
+                logger.info("成功创建索引");
+            } else {
+                logger.info("collection '{}' 已存在", MilvusConstants.MILVUS_COLLECTION_NAME);
+            }
+
+            // 3. 加载 collection 到内存（搜索必须）
+            logger.info("正在加载 collection '{}' 到内存...", MilvusConstants.MILVUS_COLLECTION_NAME);
+            R<RpcStatus> loadResp = client.loadCollection(LoadCollectionParam.newBuilder()
+                    .withCollectionName(MilvusConstants.MILVUS_COLLECTION_NAME)
+                    .build());
+            if (loadResp.getStatus() == 0) {
+                logger.info("collection '{}' 已加载", MilvusConstants.MILVUS_COLLECTION_NAME);
+            } else {
+                logger.warn("collection '{}' 加载失败: {}", MilvusConstants.MILVUS_COLLECTION_NAME, loadResp.getMessage());
+            }
+
+            return client;
+
+        } catch (Exception e) {
+            logger.error("创建 Milvus 客户端失败", e);
+            if (client != null) {
+                client.close();
+            }
+            throw new RuntimeException("创建 Milvus 客户端失败: " + e.getMessage(), e);
+        }
+    }
+
+    /**
+     * 连接到 Milvus
+     */
+    private MilvusServiceClient connectToMilvus() {
+        ConnectParam.Builder builder = ConnectParam.newBuilder()
+                .withHost(milvusProperties.getHost())
+                .withPort(milvusProperties.getPort())
+                .withDatabaseName(milvusProperties.getDatabase())
+                .withConnectTimeout(milvusProperties.getTimeout(), TimeUnit.MILLISECONDS);
+
+        // Zilliz Cloud: token + SSL
+        if (milvusProperties.getToken() != null && !milvusProperties.getToken().isEmpty()) {
+            builder.withToken(milvusProperties.getToken());
+            builder.withSecure(true);
+        }
+        // 本地 Milvus: username + password
+        else if (milvusProperties.getUsername() != null && !milvusProperties.getUsername().isEmpty()) {
+            builder.withAuthorization(milvusProperties.getUsername(), milvusProperties.getPassword());
+        }
+
+        return new MilvusServiceClient(builder.build());
+    }
+
+    /**
+     * 检查 collection 是否存在
+     */
+    private boolean collectionExists(MilvusServiceClient client, String collectionName) {
+        R<Boolean> response = client.hasCollection(HasCollectionParam.newBuilder()
+                .withCollectionName(collectionName)
+                .build());
+
+        if (response.getStatus() != 0) {
+            throw new RuntimeException("检查 collection 失败: " + response.getMessage());
+        }
+
+        return response.getData();
+    }
+
+    /**
+     * 创建 biz collection
+     */
+    private void createBizCollection(MilvusServiceClient client) {
+        // 定义字段
+        FieldType idField = FieldType.newBuilder()
+                .withName("id")
+                .withDataType(DataType.VarChar)
+                .withMaxLength(MilvusConstants.ID_MAX_LENGTH)
+                .withPrimaryKey(true)
+                .build();
+
+        FieldType vectorField = FieldType.newBuilder()
+                .withName("vector")
+                .withDataType(DataType.FloatVector)  // 改为 FloatVector
+                .withDimension(milvusProperties.getVectorDim())
+                .build();
+
+        FieldType contentField = FieldType.newBuilder()
+                .withName("content")
+                .withDataType(DataType.VarChar)
+                .withMaxLength(MilvusConstants.CONTENT_MAX_LENGTH)
+                .build();
+
+        FieldType metadataField = FieldType.newBuilder()
+                .withName("metadata")
+                .withDataType(DataType.JSON)
+                .build();
+
+        // 创建 collection schema
+        CollectionSchemaParam schema = CollectionSchemaParam.newBuilder()
+                .withEnableDynamicField(false)
+                .addFieldType(idField)
+                .addFieldType(vectorField)
+                .addFieldType(contentField)
+                .addFieldType(metadataField)
+                .build();
+
+        // 创建 collection
+        CreateCollectionParam createParam = CreateCollectionParam.newBuilder()
+                .withCollectionName(MilvusConstants.MILVUS_COLLECTION_NAME)
+                .withDescription("Business knowledge collection")
+                .withSchema(schema)
+                .withShardsNum(MilvusConstants.DEFAULT_SHARD_NUMBER)
+                .build();
+
+        R<RpcStatus> response = client.createCollection(createParam);
+        if (response.getStatus() != 0) {
+            throw new RuntimeException("创建 collection 失败: " + response.getMessage());
+        }
+    }
+
+    /**
+     * 为 collection 创建索引
+     */
+    private void createIndexes(MilvusServiceClient client) {
+        // 为 vector 字段创建索引（FloatVector 使用 IVF_FLAT 和 L2 距离）
+        CreateIndexParam vectorIndexParam = CreateIndexParam.newBuilder()
+                .withCollectionName(MilvusConstants.MILVUS_COLLECTION_NAME)
+                .withFieldName("vector")
+                .withIndexType(IndexType.IVF_FLAT)
+                .withMetricType(MetricType.L2)  // L2 距离（欧氏距离）
+                .withExtraParam("{\"nlist\":128}")
+                .withSyncMode(Boolean.FALSE)
+                .build();
+
+        R<RpcStatus> response = client.createIndex(vectorIndexParam);
+        if (response.getStatus() != 0) {
+            throw new RuntimeException("创建 vector 索引失败: " + response.getMessage());
+        }
+        
+        logger.info("成功为 vector 字段创建索引");
+    }
+}