From e770cf34174c8226eaf703c303ee3d8397c38242 Mon Sep 17 00:00:00 2001
From: Srikanth Srungarapu <ssrungarapu@cloudera.com>
Date: Wed, 9 Sep 2015 23:37:10 -0700
Subject: [PATCH] HBASE-14385 Close the sockets that is missing in connection
 closure. -- ADDENDUM

---
 .../hadoop/hbase/ipc/TestRpcClientLeaks.java  | 115 ++++++++++++++++++
 1 file changed, 115 insertions(+)
 create mode 100644 hbase-server/src/test/java/org/apache/hadoop/hbase/ipc/TestRpcClientLeaks.java

diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/ipc/TestRpcClientLeaks.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/ipc/TestRpcClientLeaks.java
new file mode 100644
index 00000000000..296505510f1
--- /dev/null
+++ b/hbase-server/src/test/java/org/apache/hadoop/hbase/ipc/TestRpcClientLeaks.java
@@ -0,0 +1,115 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ * <p>
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.hadoop.hbase.ipc;
+
+import java.io.IOException;
+import java.net.Socket;
+import java.net.SocketAddress;
+import java.util.List;
+
+import com.google.common.collect.Lists;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.hbase.HBaseTestingUtility;
+import org.apache.hadoop.hbase.HConstants;
+import org.apache.hadoop.hbase.TableName;
+import org.apache.hadoop.hbase.client.Connection;
+import org.apache.hadoop.hbase.client.ConnectionFactory;
+import org.apache.hadoop.hbase.client.Get;
+import org.apache.hadoop.hbase.client.RetriesExhaustedException;
+import org.apache.hadoop.hbase.client.Table;
+import org.apache.hadoop.hbase.codec.Codec;
+import org.apache.hadoop.hbase.testclassification.SmallTests;
+import org.apache.hadoop.io.compress.CompressionCodec;
+import org.junit.AfterClass;
+import org.junit.BeforeClass;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.experimental.categories.Category;
+import org.junit.rules.ExpectedException;
+
+import static org.apache.hadoop.hbase.HBaseTestingUtility.fam1;
+import static org.junit.Assert.*;
+
+@Category(SmallTests.class)
+public class TestRpcClientLeaks {
+
+  public static class MyRpcClientImpl extends RpcClientImpl {
+    public static List<Socket> savedSockets = Lists.newArrayList();
+    @Rule public ExpectedException thrown = ExpectedException.none();
+
+    public MyRpcClientImpl(Configuration conf, String clusterId) {
+      super(conf, clusterId);
+    }
+
+    public MyRpcClientImpl(Configuration conf, String clusterId, SocketAddress address) {
+      super(conf, clusterId, address);
+    }
+
+    @Override
+    protected Connection createConnection(ConnectionId remoteId, Codec codec,
+        CompressionCodec compressor) throws IOException {
+      return new Connection(remoteId, codec, compressor) {
+        @Override
+        protected synchronized void setupConnection() throws IOException {
+          super.setupConnection();
+          synchronized (savedSockets) {
+            savedSockets.add(socket);
+          }
+          throw new IOException("Sample exception for " +
+            "verifying socket closure in case of exceptions.");
+        }
+      };
+    }
+  }
+
+  private static final HBaseTestingUtility UTIL = new HBaseTestingUtility();
+
+  @BeforeClass
+  public static void setup() throws Exception {
+    UTIL.startMiniCluster();
+  }
+
+  @AfterClass
+  public static void teardown() throws Exception {
+    UTIL.shutdownMiniCluster();
+  }
+
+  public static final Log LOG = LogFactory.getLog(TestRpcClientLeaks.class);
+
+  @Test(expected=RetriesExhaustedException.class)
+  public void testSocketClosed() throws IOException, InterruptedException {
+    String tableName = "testSocketClosed";
+    TableName name = TableName.valueOf(tableName);
+    UTIL.createTable(name, fam1).close();
+
+    Configuration conf = new Configuration(UTIL.getConfiguration());
+    conf.set(RpcClientFactory.CUSTOM_RPC_CLIENT_IMPL_CONF_KEY,
+      MyRpcClientImpl.class.getName());
+    conf.setInt(HConstants.HBASE_CLIENT_RETRIES_NUMBER, 2);
+    Connection connection = ConnectionFactory.createConnection(conf);
+    Table table = connection.getTable(TableName.valueOf(tableName));
+    table.get(new Get("asd".getBytes()));
+    connection.close();
+    for (Socket socket : MyRpcClientImpl.savedSockets) {
+      assertTrue("Socket + " +  socket + " is not closed", socket.isClosed());
+    }
+  }
+}
+