目录修改

14 files changed, 427 insertions, 63 deletions

diff --git a/pom.xml b/pom.xml
index 23da1b2..99509f6 100644
--- a/pom.xml
+++ b/pom.xml
@@ -59,6 +59,21 @@
             <groupId>org.projectlombok</groupId>
             <artifactId>lombok</artifactId>
         </dependency>
+        <dependency>
+            <groupId>org.apache.httpcomponents</groupId>
+            <artifactId>httpclient</artifactId>
+            <version>4.4.1</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.httpcomponents</groupId>
+            <artifactId>httpcore</artifactId>
+            <version>4.4.1</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.httpcomponents</groupId>
+            <artifactId>httpmime</artifactId>
+            <version>4.4.1</version>
+        </dependency>
     </dependencies>
 
     <build>
diff --git a/src/main/java/com/sentinel/license/LicenseApplication.java b/src/main/java/com/license/LicenseApplication.java
index adf5b56..25c1c46 100644
--- a/src/main/java/com/sentinel/license/LicenseApplication.java
+++ b/src/main/java/com/license/LicenseApplication.java
@@ -1,4 +1,4 @@
-package com.sentinel.license;
+package com.license;
 
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
diff --git a/src/main/java/com/sentinel/license/bean/Code.java b/src/main/java/com/license/bean/Code.java
index 5eb3c76..b127d7b 100644
--- a/src/main/java/com/sentinel/license/bean/Code.java
+++ b/src/main/java/com/license/bean/Code.java
@@ -1,4 +1,4 @@
-package com.sentinel.license.bean;
+package com.license.bean;
 
 
 /**
diff --git a/src/main/java/com/sentinel/license/bean/LicenseInfo.java b/src/main/java/com/license/bean/LicenseInfo.java
index 929d4a4..988a03c 100644
--- a/src/main/java/com/sentinel/license/bean/LicenseInfo.java
+++ b/src/main/java/com/license/bean/LicenseInfo.java
@@ -1,4 +1,4 @@
-package com.sentinel.license.bean;
+package com.license.bean;
 
 import lombok.Data;
 
@@ -19,5 +19,7 @@ public class LicenseInfo {
 
     // 1 获取指纹 2 更新license 3 获取license信息 4  验证license信息
     private Integer type;
+    //提供服务标识
+    private String serverId;
 
 }
diff --git a/src/main/java/com/sentinel/license/bean/Ret.java b/src/main/java/com/license/bean/Ret.java
index 8018f0d..e464646 100644
--- a/src/main/java/com/sentinel/license/bean/Ret.java
+++ b/src/main/java/com/license/bean/Ret.java
@@ -1,4 +1,4 @@
-package com.sentinel.license.bean;
+package com.license.bean;
 
 
 import java.util.Arrays;
diff --git a/src/main/java/com/sentinel/license/bean/Rets.java b/src/main/java/com/license/bean/Rets.java
index db00370..9844880 100644
--- a/src/main/java/com/sentinel/license/bean/Rets.java
+++ b/src/main/java/com/license/bean/Rets.java
@@ -1,4 +1,5 @@
-package com.sentinel.license.bean;
+package com.license.bean;
+
 
 public class Rets {
 
diff --git a/src/main/java/com/license/cache/CacheManager.java b/src/main/java/com/license/cache/CacheManager.java
new file mode 100644
index 0000000..8f4b56f
--- /dev/null
+++ b/src/main/java/com/license/cache/CacheManager.java
@@ -0,0 +1,53 @@
+package com.license.cache;
+
+import org.springframework.stereotype.Component;
+
+import java.util.HashMap;
+
+/**
+ * @author fengjunfeng
+ * @date 2022/1/20
+ * @time 10:24
+ * @description
+ **/
+@Component
+public class CacheManager {
+    /**
+     */
+    private static HashMap<String, String> map = new HashMap<>();
+
+//    /**
+//     * 保存时间
+//     */
+//    private static int liveTime = 60;
+
+    /**
+     * 信息缓存
+     * @param key
+     * @param v
+     * @return
+     */
+    public static boolean addCache(String key,String v){
+        map.put(key,v);
+        return true;
+    }
+
+    /**
+     * 缓存信息
+     * @param key
+     * @return
+     */
+    public static boolean delCache(String key){
+        map.remove(key);
+        return true;
+    }
+
+    /**
+     * @param key
+     * @return
+     */
+    public static String getCache(String key){
+        return map.get(key);
+    }
+
+}
diff --git a/src/main/java/com/license/config/RestTemplateConfig.java b/src/main/java/com/license/config/RestTemplateConfig.java
new file mode 100644
index 0000000..9a16fef
--- /dev/null
+++ b/src/main/java/com/license/config/RestTemplateConfig.java
@@ -0,0 +1,121 @@
+package com.license.config;
+
+import org.apache.http.client.HttpClient;
+import org.apache.http.client.config.RequestConfig;
+import org.apache.http.config.Registry;
+import org.apache.http.config.RegistryBuilder;
+import org.apache.http.conn.socket.ConnectionSocketFactory;
+import org.apache.http.conn.socket.PlainConnectionSocketFactory;
+import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
+import org.apache.http.impl.client.HttpClientBuilder;
+import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.client.ClientHttpRequestFactory;
+import org.springframework.http.client.ClientHttpResponse;
+import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
+import org.springframework.http.converter.HttpMessageConverter;
+import org.springframework.http.converter.StringHttpMessageConverter;
+import org.springframework.web.client.ResponseErrorHandler;
+import org.springframework.web.client.RestTemplate;
+
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.util.List;
+
+/**
+*
+* @author fengjunfeng
+* @date   2022/1/20
+* @time   11:12
+* @description
+*
+**/
+@Configuration
+public class RestTemplateConfig {
+    @Value("${http.maxTotal}")
+    private String maxTotal;
+    @Value("${http.defaultMaxPerRoute}")
+    private String defaultMaxPerRoute;
+    @Value("${http.connectTimeout}")
+    private String connectTimeout;
+    @Value("${http.connectionRequestTimeout}")
+    private String connectionRequestTimeout;
+    @Value("${http.socketTimeout}")
+    private String socketTimeout;
+
+
+
+    @Bean
+    public RestTemplate restTemplate() {
+        RestTemplate restTemplate = new RestTemplate(httpRequestFactory());
+        List<HttpMessageConverter<?>> converterList = restTemplate.getMessageConverters();
+
+        //重新设置StringHttpMessageConverter字符集为UTF-8，解决中文乱码问题
+        HttpMessageConverter<?> converterTarget = null;
+        for (HttpMessageConverter<?> item : converterList) {
+            if (StringHttpMessageConverter.class == item.getClass()) {
+                converterTarget = item;
+                break;
+            }
+        }
+        if (null != converterTarget) {
+            converterList.remove(converterTarget);
+        }
+
+
+        converterList.add(1, new StringHttpMessageConverter(StandardCharsets.UTF_8));
+
+        /**
+         * restTemplate 异常处理类 , 解决返回不了除了200之外的状态码
+         */
+        ResponseErrorHandler responseErrorHandler = new ResponseErrorHandler() {
+            @Override
+            public boolean hasError(ClientHttpResponse clientHttpResponse) throws IOException {
+                return true;
+            }
+
+            @Override
+            public void handleError(ClientHttpResponse clientHttpResponse) throws IOException {
+
+            }
+        };
+        restTemplate.setErrorHandler(responseErrorHandler);
+
+        return restTemplate;
+    }
+
+    @Bean
+    public ClientHttpRequestFactory httpRequestFactory() {
+        return new HttpComponentsClientHttpRequestFactory(httpClient());
+    }
+
+    @Bean
+    public HttpClient httpClient() {
+        Registry<ConnectionSocketFactory> registry = RegistryBuilder.<ConnectionSocketFactory>create()
+                .register("http", PlainConnectionSocketFactory.getSocketFactory())
+                .register("https", SSLConnectionSocketFactory.getSocketFactory())
+                .build();
+        PoolingHttpClientConnectionManager connectionManager = new PoolingHttpClientConnectionManager(registry);
+        connectionManager.setMaxTotal(Integer.valueOf(this.maxTotal));
+        connectionManager.setDefaultMaxPerRoute(Integer.valueOf(this.defaultMaxPerRoute));
+        RequestConfig requestConfig = RequestConfig.custom()
+                //服务器返回数据(response)的时间，超过抛出read timeout
+                .setSocketTimeout(Integer.valueOf(this.socketTimeout))
+                //连接上服务器(握手成功)的时间，超出抛出connect timeout
+                .setConnectTimeout(Integer.valueOf(this.connectTimeout))
+                //从连接池中获取连接的超时时间，超时间未拿到可用连接，会抛出org.apache.http.conn.ConnectionPoolTimeoutException: Timeout waiting for connection from pool
+                .setConnectionRequestTimeout(Integer.valueOf(this.connectionRequestTimeout))
+                .build();
+        return HttpClientBuilder.create()
+                .setDefaultRequestConfig(requestConfig)
+                .setConnectionManager(connectionManager)
+                .setConnectionManagerShared(true)
+                .build();
+    }
+
+
+
+
+}
diff --git a/src/main/java/com/license/controller/LicenseController.java b/src/main/java/com/license/controller/LicenseController.java
new file mode 100644
index 0000000..01b3f71
--- /dev/null
+++ b/src/main/java/com/license/controller/LicenseController.java
@@ -0,0 +1,105 @@
+package com.license.controller;
+
+import cn.hutool.core.map.MapUtil;
+import cn.hutool.json.JSONObject;
+import cn.hutool.json.JSONUtil;
+import cn.hutool.log.Log;
+
+import com.license.bean.LicenseInfo;
+import com.license.cache.CacheManager;
+import com.license.utils.Constant;
+import com.license.utils.GetIpProtUtil;
+import com.license.utils.HaspUtil;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.*;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.client.RestTemplate;
+
+import java.net.SocketException;
+import java.net.UnknownHostException;
+import java.util.Map;
+import java.util.concurrent.CompletableFuture;
+
+/**
+ * @author fengjunfeng
+ * @date 2021/12/28
+ * @time 16:00
+ * @description
+ **/
+@RestController
+public class LicenseController {
+    private static final Log log = Log.get();
+    @Autowired
+    public RestTemplate restTemplate;
+    @Value("${bifang-api.server.port}")
+    private String port;
+    /**
+     * license锁信息交互
+     *
+     * @return
+     */
+    @PostMapping("/licenseOperation")
+    public Object licenseOperation(@RequestBody LicenseInfo licenseInfo) {
+//        if (!verify(licenseInfo.getServerId())){
+//            return null;
+//        }
+        if (licenseInfo.getType().equals(Constant.LICENSE_C2V)){
+            //获取C2V信息
+            return HaspUtil.readC2V(licenseInfo.getVendorCode(),licenseInfo.getFeatureId());
+        }else if (licenseInfo.getType().equals(Constant.LICENSE_UPDATE)){
+            //上传license信息
+            return HaspUtil.updateKeyWithLicense(licenseInfo.getV2C());
+        }else if (licenseInfo.getType().equals(Constant.LICENSE_QUERY)){
+            //获取license信息
+            return HaspUtil.getLicenseInfo(licenseInfo.getVendorCode(),licenseInfo.getFeatureId());
+        }else if (licenseInfo.getType().equals(Constant.LICENSE_VERIFY)){
+            //验证license信息
+            return HaspUtil.verify(licenseInfo.getVendorCode(),licenseInfo.getFeatureId());
+        }else {
+            log.info("type error");
+            return null;
+        }
+    }
+
+    /**
+    *
+    * @author fengjunfeng
+    * @date   2022/1/20
+    * @time   11:14
+    * @description
+    *
+    **/
+    public boolean verify(String serverId){
+        try {
+            HttpHeaders headers = new HttpHeaders();
+            headers.setContentType(MediaType.APPLICATION_JSON);
+            headers.set(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE);
+            HttpEntity<JSONObject> httpEntity = new HttpEntity<>(null, headers);
+            String url = "http://" + GetIpProtUtil.getLocalIP() +":" + port + "/getServerId";
+            ResponseEntity<Map> entity = restTemplate.exchange(url, HttpMethod.GET,httpEntity,Map.class);
+            log.info("license status: {}", JSONUtil.toJsonStr(entity));
+            String str = MapUtil.get(entity.getBody(), "data", String.class);
+            if (str.equals(serverId)){
+                return true;
+            }
+        }catch (SocketException | UnknownHostException e){
+            log.error("获取ip出错");
+        }
+        return false;
+    }
+
+
+    @PostMapping("/test")
+    public Object test() {
+        verify("1111");
+
+        if (CacheManager.getCache("test") == null){
+            CacheManager.addCache("test","11111");
+        }else {
+            System.out.println(CacheManager.getCache("test"));
+        }
+        return null;
+    }
+
+}
diff --git a/src/main/java/com/sentinel/license/utils/Constant.java b/src/main/java/com/license/utils/Constant.java
index 19948a7..a2fb297 100644
--- a/src/main/java/com/sentinel/license/utils/Constant.java
+++ b/src/main/java/com/license/utils/Constant.java
@@ -1,4 +1,4 @@
-package com.sentinel.license.utils;
+package com.license.utils;
 
 /**
  * @author fengjunfeng
diff --git a/src/main/java/com/license/utils/GetIpProtUtil.java b/src/main/java/com/license/utils/GetIpProtUtil.java
new file mode 100644
index 0000000..7bfc472
--- /dev/null
+++ b/src/main/java/com/license/utils/GetIpProtUtil.java
@@ -0,0 +1,86 @@
+package com.license.utils;
+
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import java.net.InetAddress;
+import java.net.NetworkInterface;
+import java.net.SocketException;
+import java.net.UnknownHostException;
+import java.util.Enumeration;
+
+public class GetIpProtUtil {
+
+    /**
+     * 获取Linux下的IP地址
+     *
+     * @return IP地址
+     * @throws SocketException
+     */
+    private static String getLinuxLocalIp() throws SocketException {
+        String ip = "";
+        try {
+            for (Enumeration<NetworkInterface> en = NetworkInterface.getNetworkInterfaces(); en.hasMoreElements();) {
+                NetworkInterface intf = en.nextElement();
+                String name = intf.getName();
+                if (!name.contains("docker") && !name.contains("lo")) {
+                    for (Enumeration<InetAddress> enumIpAddr = intf.getInetAddresses(); enumIpAddr.hasMoreElements();) {
+                        InetAddress inetAddress = enumIpAddr.nextElement();
+                        if (!inetAddress.isLoopbackAddress()) {
+                            String ipaddress = inetAddress.getHostAddress().toString();
+                            if (!ipaddress.contains("::") && !ipaddress.contains("0:0:")
+                                    && !ipaddress.contains("fe80")) {
+                                ip = ipaddress;
+                            }
+                        }
+                    }
+                }
+            }
+        } catch (SocketException ex) {
+            ip = "127.0.0.1";
+            ex.printStackTrace();
+        }
+        return ip;
+    }
+    /**
+     * 判断操作系统是否是Windows
+     *
+     * @return
+     */
+    public static boolean isWindowsOS() {
+        boolean isWindowsOS = false;
+        // 注：这里的system，系统指的是 JRE (runtime)system，不是指 OS
+        String osName = System.getProperty("os.name");
+        if (osName.toLowerCase().indexOf("windows") > -1) {
+            isWindowsOS = true;
+        }
+        return isWindowsOS;
+    }
+    /**
+     * 获取本地IP地址
+     *
+     * @throws SocketException
+     */
+    public static String getLocalIP() throws UnknownHostException, SocketException {
+        if (isWindowsOS()) {
+            return InetAddress.getLocalHost().getHostAddress();
+        } else {
+            return getLinuxLocalIp();
+        }
+    }
+
+    public static String getLocalPort() {
+        return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest().getServerPort() + "";
+    }
+
+    public static String getIp() {
+        String ip = "127.0.0.1";
+        return ip;
+    }
+
+
+//    public static void main(String[] args) throws MalformedObjectNameException, SocketException, UnknownHostException {
+//        System.out.println(getLocalIP());
+//    }
+
+}
diff --git a/src/main/java/com/sentinel/license/utils/HaspUtil.java b/src/main/java/com/license/utils/HaspUtil.java
index f48c788..0cf8a38 100644
--- a/src/main/java/com/sentinel/license/utils/HaspUtil.java
+++ b/src/main/java/com/license/utils/HaspUtil.java
@@ -1,4 +1,4 @@
-package com.sentinel.license.utils;
+package com.license.utils;
 import Aladdin.Hasp;
 import Aladdin.HaspStatus;
 import cn.hutool.core.util.XmlUtil;
@@ -163,17 +163,19 @@ public class HaspUtil {
 //		log.info("license status: {}", status);
 		if(status == HaspStatus.HASP_STATUS_OK) {
 			//如果已经安装过license ，读取指纹和 hasp id
-			String info = hasp.getSessionInfo(KEY_ID_FORMAT);
-            Document document = XmlUtil.parseXml(info);
-            Element rootEle = document.getDocumentElement();
-            Element haspElement=XmlUtil.getElement(rootEle,"hasp");
-            String id = haspElement.getAttribute("id");
-            String licenseInfo = hasp.getInfo("<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n" +
-                    "<haspscope>\n" +
-                    "    <hasp " +
-                    "id= \"" + id +"\""+
-                    "/>\n" +
-                    "</haspscope>",KEY_C2V_FORMAT,vendorCode);
+            String licenseInfo = hasp.getSessionInfo(KEY_C2V_FORMAT);
+
+//			String info = hasp.getSessionInfo(KEY_ID_FORMAT);
+//            Document document = XmlUtil.parseXml(info);
+//            Element rootEle = document.getDocumentElement();
+//            Element haspElement=XmlUtil.getElement(rootEle,"hasp");
+//            String id = haspElement.getAttribute("id");
+//            String licenseInfo = hasp.getInfo("<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n" +
+//                    "<haspscope>\n" +
+//                    "    <hasp " +
+//                    "id= \"" + id +"\""+
+//                    "/>\n" +
+//                    "</haspscope>",KEY_C2V_FORMAT,vendorCode);
 			status = hasp.getLastError();
 			log.info("getLicense c2v license status: {}", status);
 			if (status != HaspStatus.HASP_STATUS_OK) {
diff --git a/src/main/java/com/sentinel/license/controller/LicenseController.java b/src/main/java/com/sentinel/license/controller/LicenseController.java
deleted file mode 100644
index 67638c6..0000000
--- a/src/main/java/com/sentinel/license/controller/LicenseController.java
+++ /dev/null
@@ -1,45 +0,0 @@
-package com.sentinel.license.controller;
-
-import cn.hutool.log.Log;
-import com.sentinel.license.bean.LicenseInfo;
-import com.sentinel.license.utils.Constant;
-import com.sentinel.license.utils.HaspUtil;
-import org.springframework.web.bind.annotation.*;
-
-/**
- * @author fengjunfeng
- * @date 2021/12/28
- * @time 16:00
- * @description
- **/
-@RestController
-public class LicenseController {
-    private static final Log log = Log.get();
-
-
-    /**
-     * license锁信息交互
-     *
-     * @return
-     */
-    @PostMapping("/licenseOperation")
-    public Object licenseOperation(@RequestBody LicenseInfo licenseInfo) {
-        if (licenseInfo.getType().equals(Constant.LICENSE_C2V)){
-            //获取C2V信息
-            return HaspUtil.readC2V(licenseInfo.getVendorCode(),licenseInfo.getFeatureId());
-        }else if (licenseInfo.getType().equals(Constant.LICENSE_UPDATE)){
-            //上传license信息
-            return HaspUtil.updateKeyWithLicense(licenseInfo.getV2C());
-        }else if (licenseInfo.getType().equals(Constant.LICENSE_QUERY)){
-            //获取license信息
-            return HaspUtil.getLicenseInfo(licenseInfo.getVendorCode(),licenseInfo.getFeatureId());
-        }else if (licenseInfo.getType().equals(Constant.LICENSE_VERIFY)){
-            //验证license信息
-            return HaspUtil.verify(licenseInfo.getVendorCode(),licenseInfo.getFeatureId());
-        }else {
-            log.info("type error");
-            return null;
-        }
-    }
-
-}
diff --git a/src/main/resources/application.properties b/src/main/resources/application.properties
index e68636a..11359b6 100644
--- a/src/main/resources/application.properties
+++ b/src/main/resources/application.properties
@@ -8,3 +8,27 @@ server.tomcat.accept-count=1000
 server.tomcat.min-spare-threads=100
 
 #server.address=127.0.0.1
+
+bifang-api.server.port=8080
+#设置连接超时时间，单位毫秒
+http.connectTimeout=120000
+#http clilent中从connetcion pool中获得一个connection的超时时间,单位毫秒
+http.connectionRequestTimeout=120000
+#请求获取数据的超时时间，单位毫秒。 如果访问一个接口，多少时间内无法返回数据，就直接放弃此次调用
+http.socketTimeout=120000
+http.maxTotal=20
+http.defaultMaxPerRoute=2
+
+
+
+# 开启https,配置跟证书对应
+#server.ssl.enabled=true
+#server.ssl.key-store=D:/tomcat.keystore
+## server.ssl.key-store-type=JKS
+#server.ssl.key-store-type=JKS
+## 密码
+#server.ssl.key-store-password=tomcat
+## springboot2.x不需要配置
+#server.ssl.key-password=tomcat
+## 别名
+#server.ssl.key-alias=tomcat