From 623cee8be4846e5762ff2949e519335ef8dee2bb Mon Sep 17 00:00:00 2001
From: suerprisePlus <15810472099@163.com>
Date: 星期一, 09 九月 2024 18:30:55 +0800
Subject: [PATCH] 大模型知识图谱接口对接
---
src/main/java/com/yb/service/AgentService.java | 61 +++++++++++++++++++++++++++++-
1 files changed, 59 insertions(+), 2 deletions(-)
diff --git a/src/main/java/com/yb/service/AgentService.java b/src/main/java/com/yb/service/AgentService.java
index eb93be4..6107210 100644
--- a/src/main/java/com/yb/service/AgentService.java
+++ b/src/main/java/com/yb/service/AgentService.java
@@ -1,11 +1,15 @@
package com.yb.service;
+import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.yb.config.XzConfig;
import com.yb.entity.ReqEntity;
import com.yb.util.EntityHttpUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.HashMap;
@Service
public class AgentService {
@@ -35,7 +39,7 @@
//
public String getAgentPoiMap(String func, String filed) throws Exception {
- String poiObj= xzService.getQueryFiled(func, filed, xzConfig.layerId, xzConfig.dbid, xzConfig.queryEntity);
+ String poiObj = xzService.getQueryFiled(func, filed, xzConfig.layerId, xzConfig.dbid, xzConfig.queryEntity);
System.out.println("rec poiMap:" + poiObj);
return poiObj;
}
@@ -47,7 +51,60 @@
Double lon = Double.parseDouble(obj.getString("lon"));
Double lat = Double.parseDouble(obj.getString("lat"));
String wkt = xzService.getWKt(lon, lat, dis);
- return xzService.getQueryAround(type, wkt, typeFiled, xzConfig.layerId, xzConfig.dbid, xzConfig.queryEntity);
+ return xzService.getQueryAround(type, wkt, typeFiled, xzConfig.layerId, xzConfig.dbid, xzConfig.queryEntity);
+
+ }
+
+ public HashMap<String, Object> getQueryRelationship(String place, String filed) throws Exception {
+ String poiObj = xzService.getQueryEntity(place, filed, xzConfig.layerId, xzConfig.dbid, xzConfig.layerQuery);
+ HashMap<String, String> hashMap = new HashMap<String, String>();
+ hashMap.put("token", xzConfig.token);
+ hashMap.put("dbid", xzConfig.dbid);
+ hashMap.put("groupid", "");
+ String relationList = entityHttpUtil.getPostMessage(xzConfig.relationList, hashMap);
+ JSONObject relattionObj = JSONObject.parseObject(relationList);
+ String data = relattionObj.getString("data");
+ JSONArray relationArry = JSONArray.parseArray(data);
+ JSONObject itemObj = new JSONObject();
+ for (int i = 0; i < relationArry.size(); i++) {
+ JSONObject relationtem = relationArry.getJSONObject(i);
+ String name = relationtem.getString("name");
+ if (name.equals("鍖呭惈鍏崇郴")) {
+ itemObj = relationtem;
+ }
+ }
+ String distanceUrl = xzConfig.relationDistinct + "?dbid=" + xzConfig.dbid + "&token=" + xzConfig.token + "&relationid=" + itemObj.getString("code");
+ String relationDistance = entityHttpUtil.getMessage(distanceUrl);
+ JSONObject distanceObj = JSONObject.parseObject(relationList);
+ String distanceData = relattionObj.getString("data");
+ System.out.println("rec getQueryRelationship:" + distanceData);
+
+ String relateddataeUrl = xzConfig.relationRelateddata + "?start=1&count=1000000&&a_like=&b_like=&dbid=" + xzConfig.dbid + "&token=" + xzConfig.token + "&relationid=" + itemObj.getString("code");
+ String relationRelateddata = entityHttpUtil.getMessage(relateddataeUrl);
+ JSONObject relateddataeObj = JSONObject.parseObject(relationRelateddata);
+ String relateddataeData = relateddataeObj.getString("data");
+ JSONArray relateddataeArray = JSONArray.parseArray(JSONObject.parseObject(relateddataeData).getString("items"));
+ System.out.println("rec getQueryRelationship:" + relateddataeArray);
+ ArrayList<String> list = new ArrayList<>();
+ for (int k = 0; k < relateddataeArray.size(); k++) {
+ JSONObject reitem = relateddataeArray.getJSONObject(k);
+ list.add(reitem.getString("a_objectid"));
+ list.add(reitem.getString("b_objectid"));
+ }
+ HashMap<String, String> reMap = new HashMap<String, String>();
+ reMap.put("token", xzConfig.token);
+ reMap.put("dbid", xzConfig.dbid);
+ reMap.put("businessids", list.toString());
+ reMap.put("layerid", xzConfig.layerId);
+ reMap.put("querylatest", "true");
+ String reMapList = entityHttpUtil.getPostMessage(xzConfig.layerQuery, reMap);
+ JSONArray reArray = JSONArray.parseArray(JSONObject.parseObject(reMapList).getString("data"));
+ System.out.println("rec getQueryRelationship:" + reArray);
+ HashMap<String, Object> traiMap = new HashMap<>();
+ traiMap.put("center", JSONObject.parseObject(poiObj));
+ traiMap.put("positions", reArray);
+ System.out.println("rec getQueryRelationship:" + traiMap);
+ return traiMap;
}
}
--
Gitblit v1.9.3