添加dataSourceId参数的方法

src/main/java/com/scbfkj/uni/process/Web.java

@@ -1,49 +1,22 @@
 package com.scbfkj.uni.process;
 
-import com.scbfkj.uni.library.DataFormatUtil;
 import com.scbfkj.uni.library.UniReturnUtil;
-import org.apache.commons.pool2.*;
-import org.apache.commons.pool2.impl.DefaultPooledObject;
-import org.apache.commons.pool2.impl.GenericKeyedObjectPool;
-import org.apache.commons.pool2.impl.GenericObjectPool;
+import com.scbfkj.uni.system.Config;
 import org.springframework.http.HttpEntity;
 import org.springframework.http.HttpHeaders;
-import org.springframework.http.HttpMethod;
 import org.springframework.http.MediaType;
 import org.springframework.http.converter.StringHttpMessageConverter;
-import org.springframework.util.StringUtils;
 import org.springframework.web.client.RestTemplate;
-import org.springframework.web.reactive.function.client.WebClient;
 
 import java.nio.charset.StandardCharsets;
-import java.util.*;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
 
 public class Web {
 
-    private final KeyedObjectPool<String, Web> webPool = new GenericKeyedObjectPool<>(new BaseKeyedPooledObjectFactory<String, Web>() {
-        @Override
-        public Web create(String s) throws Exception {
-            return new Web();
-        }
-
-        @Override
-        public PooledObject<Web> wrap(Web web) {
-            return new DefaultPooledObject<>(web);
-        }
-    });
-    private final static ObjectPool<WebClient> clientPool = new GenericObjectPool<>(new BasePooledObjectFactory<>() {
-
-        @Override
-        public WebClient create() throws Exception {
-            return WebClient.create();
-        }
-
-        @Override
-        public PooledObject<WebClient> wrap(WebClient webClient) {
-            return new DefaultPooledObject<>(webClient);
-        }
-    });
-
+    private static final DataBase DATA_BASE = new DataBase();
 
     private RestTemplate restTemplate; //webapi请求对象
 
@@ -56,8 +29,25 @@ public class Web {
     private Map<String, Object> baseHeader = new HashMap<>();
 
 
+    public Map<String, Object> execWebApiByDataSourceId(String dataSourceId, Object headers, String method, Object defaultBody) throws Exception {
+        Map<String, Object> config = queryConnectionStr(dataSourceId);
+
+        String host;
+        if (Objects.isNull(headers)) {
+            headers = config.get("headers");
+        }
+        if (Objects.isNull(method)) {
+            method = "post";
+        }
+        if (Objects.isNull(defaultBody)) {
+            defaultBody = config.get("body");
+        }
+        host = config.get("host").toString();
+        return execWebApi(headers, method, defaultBody, host);
+    }
+
     public Map<String, Object> execWebApi(Object headers, String method, Object defaultBody, String url) {
-        Map<String, Object> restTemplateResult = initWebApiParams(headers,method,url); // 通过连接配置 获取restTemplate
+        Map<String, Object> restTemplateResult = initWebApiParams(headers, method, url); // 通过连接配置 获取restTemplate
 
         if (!"0".equals(restTemplateResult.get("code"))) {
             return UniReturnUtil.fail(restTemplateResult.get("message").toString());
@@ -76,7 +66,7 @@ public class Web {
 
             Map<String, Object> result = sendWebApiRequest(defaultBody);
 
-            return UniReturnUtil.success(result) ;
+            return UniReturnUtil.success(result);
         } catch (Exception e) {
             return UniReturnUtil.fail("restTemplate执行异常" + UniReturnUtil.getMessage(e));
         }
@@ -105,7 +95,7 @@ public class Web {
         }
     }
 
-    private Map<String, Object> initWebApiParams(Object headers,String method,String url) {
+    private Map<String, Object> initWebApiParams(Object headers, String method, String url) {
         try {
 
             if (restTemplate == null) {
@@ -140,4 +130,22 @@ public class Web {
         }
         return UniReturnUtil.success(null);
     }
+
+
+    private static Map<String, Object> queryConnectionStr(String datasourceId) throws Exception {
+        List<Map<String, Object>> result = DATA_BASE.query(Config.getCenterConnectionStr(), """
+                select host,httpheaders,httpbody
+                from datasource
+                where datasourceid = ?""", datasourceId);
+        if (result.isEmpty()) {
+            throw new RuntimeException("数据源错误：没有找到数据源");
+        }
+        return result.stream().findFirst().map(it -> {
+            HashMap<String, Object> hashMap = new HashMap<>();
+            hashMap.put("url", it.get("host"));
+            hashMap.put("headers", it.get("httpheaders"));
+            hashMap.put("body", it.get("httpbody"));
+            return hashMap;
+        }).get();
+    }
 }