Solve #1416, Complete disable SmartClient on node

littlezhou · Nov 20, 2017 · 6887fd2 · 6887fd2
1 parent 96ae0e2
commit 6887fd2
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/smart-common/src/main/java/org/smartdata/SmartConstants.java b/smart-common/src/main/java/org/smartdata/SmartConstants.java
@@ -35,4 +35,7 @@ public class SmartConstants {
 
   public static final String SMART_ADMIN_PROTOCOL_NAME =
     "org.smartdata.protocol.SmartAdminProtocol";
+
+  public static final String SMART_CLIENT_DISABLED_ID_FILE =
+    "/tmp/SMART_CLIENT_DISABLED_ID_FILE";
 }
diff --git a/...p-support/smart-hadoop-client/src/main/java/org/smartdata/hdfs/client/SmartDFSClient.java b/...p-support/smart-hadoop-client/src/main/java/org/smartdata/hdfs/client/SmartDFSClient.java
@@ -24,9 +24,11 @@
 import org.apache.hadoop.hdfs.DFSInputStream;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.smartdata.SmartConstants;
 import org.smartdata.client.SmartClient;
 import org.smartdata.metrics.FileAccessEvent;
 
+import java.io.File;
 import java.io.IOException;
 import java.net.InetSocketAddress;
 import java.net.URI;
@@ -39,6 +41,9 @@ public class SmartDFSClient extends DFSClient {
   public SmartDFSClient(InetSocketAddress nameNodeAddress, Configuration conf,
       InetSocketAddress smartServerAddress) throws IOException {
     super(nameNodeAddress, conf);
+    if (isSmartClientDisabled()) {
+      return;
+    }
     try {
       smartClient = new SmartClient(conf, smartServerAddress);
       healthy = true;
@@ -51,7 +56,9 @@ public SmartDFSClient(InetSocketAddress nameNodeAddress, Configuration conf,
   public SmartDFSClient(final URI nameNodeUri, final Configuration conf,
       final InetSocketAddress smartServerAddress) throws IOException {
     super(nameNodeUri, conf);
-
+    if (isSmartClientDisabled()) {
+      return;
+    }
     try {
       smartClient = new SmartClient(conf, smartServerAddress);
       healthy = true;
@@ -65,6 +72,9 @@ public SmartDFSClient(URI nameNodeUri, Configuration conf,
       FileSystem.Statistics stats, InetSocketAddress smartServerAddress)
       throws IOException {
     super(nameNodeUri, conf, stats);
+    if (isSmartClientDisabled()) {
+      return;
+    }
     try {
       smartClient = new SmartClient(conf, smartServerAddress);
       healthy = true;
@@ -77,6 +87,9 @@ public SmartDFSClient(URI nameNodeUri, Configuration conf,
   public SmartDFSClient(Configuration conf,
       InetSocketAddress smartServerAddress) throws IOException {
     super(conf);
+    if (isSmartClientDisabled()) {
+      return;
+    }
     try {
       smartClient = new SmartClient(conf, smartServerAddress);
       healthy = true;
@@ -88,6 +101,9 @@ public SmartDFSClient(Configuration conf,
 
   public SmartDFSClient(Configuration conf) throws IOException {
     super(conf);
+    if (isSmartClientDisabled()) {
+      return;
+    }
     try {
       smartClient = new SmartClient(conf);
       healthy = true;
@@ -146,9 +162,14 @@ public synchronized void close() throws IOException {
         if (smartClient != null) {
           smartClient.close();
         }
-      }finally {
+      } finally {
         healthy = false;
       }
     }
   }
+
+  private boolean isSmartClientDisabled() {
+    File idFile = new File(SmartConstants.SMART_CLIENT_DISABLED_ID_FILE);
+    return idFile.exists();
+  }
 }