feat: add protect check when endpoint fetch

netease-im · Dec 12, 2024 · ed0a867 · ed0a867
1 parent 8e1c4ea
commit ed0a867
Showing 1 changed file with 37 additions and 1 deletion.
diff --git a/src/main/java/com/netease/nim/im/server/sdk/core/endpoint/DynamicEndpointFetcher.java b/src/main/java/com/netease/nim/im/server/sdk/core/endpoint/DynamicEndpointFetcher.java
@@ -123,11 +123,21 @@ private boolean reload(String lbs) {
             }
             if (code == 200) {
                 String defaultEndpoint = json.getString("default.endpoint");
+                if (!check(defaultEndpoint)) {
+                    logger.error("default endpoint check error, skip update, endpoint = {}", defaultEndpoint);
+                    return false;
+                }
                 JSONArray backupEndpointsJson = json.getJSONArray("backup.endpoints");
                 Endpoints endpoints = new Endpoints();
                 List<String> backupEndpoints = new ArrayList<>();
+                boolean existsSkip = false;
                 if (backupEndpointsJson != null) {
                     for (Object backupEndpoint : backupEndpointsJson) {
+                        if (!check(String.valueOf(backupEndpoint))) {
+                            logger.error("backup endpoint check error, skip, endpoint = {}", defaultEndpoint);
+                            existsSkip = true;
+                            continue;
+                        }
                         backupEndpoints.add(String.valueOf(backupEndpoint));
                     }
                 }
@@ -143,7 +153,9 @@ private boolean reload(String lbs) {
                     ttl = 30;
                 }
                 this.nextFetchTime = System.currentTimeMillis() + ttl * 1000L;
-                this.md5 = json.getString("md5");
+                if (!existsSkip) {
+                    this.md5 = json.getString("md5");
+                }
                 this.endpoints = endpoints;
                 return true;
             } else {
@@ -162,4 +174,28 @@ private boolean reload(String lbs) {
     public Endpoints get() {
         return endpoints;
     }
+
+    private boolean check(String endpoint) {
+        if (okHttpClient == null) {
+            return true;
+        }
+        String url = endpoint + Constants.Endpoint.detectPath;
+        Request request = new Request.Builder().get()
+                .url(url)
+                .build();
+        boolean success;
+        try (Response response = okHttpClient.newCall(request).execute()) {
+            String string = response.body().string();
+            success = response.code() == 200;
+            if (logger.isDebugEnabled()) {
+                logger.debug("detect, endpoint = {}, path = {}, code = {}, response = {}", endpoint, Constants.Endpoint.detectPath, response.code(), string);
+            }
+        } catch (Exception e) {
+            if (logger.isDebugEnabled()) {
+                logger.debug("detect error, endpoint = {}, path = {}", endpoint, Constants.Endpoint.detectPath, e);
+            }
+            success = false;
+        }
+        return success;
+    }
 }