diff --git a/zyplayer-doc-es/src/main/java/com/zyplayer/doc/elasticsearch/controller/EsMappingController.java b/zyplayer-doc-es/src/main/java/com/zyplayer/doc/elasticsearch/controller/EsMappingController.java index f7590d95..56085cc3 100644 --- a/zyplayer-doc-es/src/main/java/com/zyplayer/doc/elasticsearch/controller/EsMappingController.java +++ b/zyplayer-doc-es/src/main/java/com/zyplayer/doc/elasticsearch/controller/EsMappingController.java @@ -2,7 +2,9 @@ package com.zyplayer.doc.elasticsearch.controller; import com.zyplayer.doc.core.json.DocResponseJson; import com.zyplayer.doc.core.json.ResponseJson; +import com.zyplayer.doc.data.repository.manage.entity.EsDatasource; import com.zyplayer.doc.data.service.elasticsearch.support.ElasticSearchUtil; +import com.zyplayer.doc.data.service.manage.EsDatasourceService; import org.apache.commons.lang3.StringUtils; import org.elasticsearch.action.search.SearchRequest; import org.elasticsearch.client.RequestOptions; @@ -45,16 +47,19 @@ public class EsMappingController { @Resource ElasticSearchUtil elasticSearchUtil; + @Resource + EsDatasourceService esDatasourceService; @PostMapping("/mappings") - public ResponseJson mappings(String index) throws IOException { + public ResponseJson mappings(Long id, String index) { + EsDatasource esDatasource = esDatasourceService.getById(id); GetMappingsRequest request = new GetMappingsRequest(); request.setMasterTimeout(TimeValue.timeValueMinutes(1)); if (StringUtils.isNotBlank(index)) { request.indices(index); } try { - RestHighLevelClient client = elasticSearchUtil.getEsClient("127.0.0.1:9200", "http"); + RestHighLevelClient client = elasticSearchUtil.getEsClient(esDatasource.getHostPort(), esDatasource.getScheme()); GetMappingsResponse getMappingResponse = client.indices().getMapping(request, RequestOptions.DEFAULT); Map allMappings = getMappingResponse.mappings(); return DocResponseJson.ok(allMappings); @@ -65,14 +70,15 @@ public class EsMappingController { } @PostMapping("/execute") - public ResponseJson execute(String index, String sql) throws IOException { + public ResponseJson execute(Long id, String index, String sql) { + EsDatasource esDatasource = esDatasourceService.getById(id); SearchTemplateRequest request = new SearchTemplateRequest(); request.setRequest(new SearchRequest().indices(index)); request.setScriptType(ScriptType.INLINE); request.setScript(sql); request.setScriptParams(new HashMap<>()); try { - RestHighLevelClient client = elasticSearchUtil.getEsClient("127.0.0.1:9200", "http"); + RestHighLevelClient client = elasticSearchUtil.getEsClient(esDatasource.getHostPort(), esDatasource.getScheme()); SearchTemplateResponse response = client.searchTemplate(request, RequestOptions.DEFAULT); List> resultList = new LinkedList<>(); for (SearchHit searchHit : response.getResponse().getHits()) { @@ -85,11 +91,12 @@ public class EsMappingController { } @GetMapping("/index") - public ResponseJson index(String index) throws IOException { + public ResponseJson index(Long id, String index) throws IOException { + EsDatasource esDatasource = esDatasourceService.getById(id); GetIndexRequest request = new GetIndexRequest(index); request.setMasterTimeout(TimeValue.timeValueMinutes(1)); try { - RestHighLevelClient client = elasticSearchUtil.getEsClient("127.0.0.1:9200", "http"); + RestHighLevelClient client = elasticSearchUtil.getEsClient(esDatasource.getHostPort(), esDatasource.getScheme()); GetIndexResponse indexResponse = client.indices().get(request, RequestOptions.DEFAULT); Map resultMap = new HashMap<>(); resultMap.put("mapping", indexResponse.getMappings().get(index)); diff --git a/zyplayer-doc-ui/es-ui/src/App.vue b/zyplayer-doc-ui/es-ui/src/App.vue index 701ac645..810fd6ca 100644 --- a/zyplayer-doc-ui/es-ui/src/App.vue +++ b/zyplayer-doc-ui/es-ui/src/App.vue @@ -6,8 +6,13 @@
+
+ + + +
- 控制台 +