diff --git a/dsRes/src/main/java/com/dsideal/Res/Milvus/Controller/MilvusDemoController.java b/dsRes/src/main/java/com/dsideal/Res/Milvus/Controller/MilvusDemoController.java index 8aaa2f9c..63a63bd3 100644 --- a/dsRes/src/main/java/com/dsideal/Res/Milvus/Controller/MilvusDemoController.java +++ b/dsRes/src/main/java/com/dsideal/Res/Milvus/Controller/MilvusDemoController.java @@ -50,9 +50,12 @@ public class MilvusDemoController extends Controller { .withVectorFieldName("vector_field") .build(); - // 修改这行代码 + // 修改这部分代码 R searchResponse = client.search(searchParam); - SearchResultsWrapper wrapper = new SearchResultsWrapper(searchResponse.getData()); + if (searchResponse.getStatus() != R.Status.Success.getCode()) { + throw new RuntimeException("Milvus search failed: " + searchResponse.getMessage()); + } + SearchResultsWrapper wrapper = new SearchResultsWrapper(searchResponse.getData().getResults()); // 返回搜索结果 renderJson(wrapper.getIDScore(0)); diff --git a/dsRes/src/main/java/com/dsideal/Res/ResApplication.java b/dsRes/src/main/java/com/dsideal/Res/ResApplication.java index 110556f9..9f46ca43 100644 --- a/dsRes/src/main/java/com/dsideal/Res/ResApplication.java +++ b/dsRes/src/main/java/com/dsideal/Res/ResApplication.java @@ -25,12 +25,11 @@ import org.slf4j.LoggerFactory; import java.io.File; public class ResApplication extends JFinalConfig { - // 将字段改为静态 - private static final Neo4jPlugin neo4jPlugin = Neo4jPlugin.getInstance(); - - private static final Logger logger = LoggerFactory.getLogger(ResApplication.class); + private final Neo4jPlugin neo4jPlugin = Neo4jPlugin.getInstance(); private final MilvusPlugin milvusPlugin = MilvusPlugin.getInstance(); + private static final Logger logger = LoggerFactory.getLogger(ResApplication.class); + public static void main(String[] args) { System.out.println("当前环境: " + PropKit.getEnvPrefix()); UndertowServer.create(ResApplication.class, "undertow.properties").start(); @@ -56,6 +55,7 @@ public class ResApplication extends JFinalConfig { me.add("/res", ResourceController.class); //Neo4j测试 me.add("/neo4j", Neo4jDemoController.class); + //Milvus测试 me.add("/milvus", MilvusDemoController.class); } @@ -169,6 +169,7 @@ public class ResApplication extends JFinalConfig { public void onStop() { // JFinal 停止时关闭 Neo4j 连接 neo4jPlugin.close(); + // JFinal 停止时关闭 Milvus 连接 milvusPlugin.close(); } } diff --git a/dsRes/src/main/resources/application_dev.yaml b/dsRes/src/main/resources/application_dev.yaml index af51c38c..e57a6f63 100644 --- a/dsRes/src/main/resources/application_dev.yaml +++ b/dsRes/src/main/resources/application_dev.yaml @@ -62,12 +62,12 @@ neo4j: password: DsideaL4r5t6y7u # 向量数据库 -mivlus: - # Milvus 服务器的主机地址 +milvus: + # 服务器的主机地址 ms_host: 10.10.14.207 - # Milvus 服务器的端口号 + # 服务器的端口号 ms_port: 19530 - # Milvus 集合的名称 + # 集合的名称 ms_collection_name: ds_collection - # Milvus 连接池的最大连接数 + # 连接池的最大连接数 ms_max_connections: 50