main
HuangHai 2 months ago
parent 801e4d585a
commit 2247d10604

@ -50,9 +50,12 @@ public class MilvusDemoController extends Controller {
.withVectorFieldName("vector_field") .withVectorFieldName("vector_field")
.build(); .build();
// 修改这代码 // 修改这部分代码
R<SearchResults> searchResponse = client.search(searchParam); R<SearchResults> 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)); renderJson(wrapper.getIDScore(0));

@ -25,12 +25,11 @@ import org.slf4j.LoggerFactory;
import java.io.File; import java.io.File;
public class ResApplication extends JFinalConfig { public class ResApplication extends JFinalConfig {
// 将字段改为静态 private final Neo4jPlugin neo4jPlugin = Neo4jPlugin.getInstance();
private static final Neo4jPlugin neo4jPlugin = Neo4jPlugin.getInstance();
private static final Logger logger = LoggerFactory.getLogger(ResApplication.class);
private final MilvusPlugin milvusPlugin = MilvusPlugin.getInstance(); private final MilvusPlugin milvusPlugin = MilvusPlugin.getInstance();
private static final Logger logger = LoggerFactory.getLogger(ResApplication.class);
public static void main(String[] args) { public static void main(String[] args) {
System.out.println("当前环境: " + PropKit.getEnvPrefix()); System.out.println("当前环境: " + PropKit.getEnvPrefix());
UndertowServer.create(ResApplication.class, "undertow.properties").start(); UndertowServer.create(ResApplication.class, "undertow.properties").start();
@ -56,6 +55,7 @@ public class ResApplication extends JFinalConfig {
me.add("/res", ResourceController.class); me.add("/res", ResourceController.class);
//Neo4j测试 //Neo4j测试
me.add("/neo4j", Neo4jDemoController.class); me.add("/neo4j", Neo4jDemoController.class);
//Milvus测试
me.add("/milvus", MilvusDemoController.class); me.add("/milvus", MilvusDemoController.class);
} }
@ -169,6 +169,7 @@ public class ResApplication extends JFinalConfig {
public void onStop() { public void onStop() {
// JFinal 停止时关闭 Neo4j 连接 // JFinal 停止时关闭 Neo4j 连接
neo4jPlugin.close(); neo4jPlugin.close();
// JFinal 停止时关闭 Milvus 连接
milvusPlugin.close(); milvusPlugin.close();
} }
} }

@ -62,12 +62,12 @@ neo4j:
password: DsideaL4r5t6y7u password: DsideaL4r5t6y7u
# 向量数据库 # 向量数据库
mivlus: milvus:
# Milvus 服务器的主机地址 # 服务器的主机地址
ms_host: 10.10.14.207 ms_host: 10.10.14.207
# Milvus 服务器的端口号 # 服务器的端口号
ms_port: 19530 ms_port: 19530
# Milvus 集合的名称 # 集合的名称
ms_collection_name: ds_collection ms_collection_name: ds_collection
# Milvus 连接池的最大连接数 # 连接池的最大连接数
ms_max_connections: 50 ms_max_connections: 50

Loading…
Cancel
Save