svn commit: r1467499 - /hbase/trunk/hbase-server/src/main/java/org/apache/hadoop/hbase/util/hbck/TableLockChecker.java

[en...@apache.org]

Author: enis
Date: Fri Apr 12 22:32:51 2013
New Revision: 1467499

URL: http://svn.apache.org/r1467499
Log:
HBASE-8205. HBCK support for table locks. Addendum for adding new class

Added:
    hbase/trunk/hbase-server/src/main/java/org/apache/hadoop/hbase/util/hbck/TableLockChecker.java

Added: hbase/trunk/hbase-server/src/main/java/org/apache/hadoop/hbase/util/hbck/TableLockChecker.java
URL: http://svn.apache.org/viewvc/hbase/trunk/hbase-server/src/main/java/org/apache/hadoop/hbase/util/hbck/TableLockChecker.java?rev=1467499&view=auto
==============================================================================
--- hbase/trunk/hbase-server/src/main/java/org/apache/hadoop/hbase/util/hbck/TableLockChecker.java (added)
+++ hbase/trunk/hbase-server/src/main/java/org/apache/hadoop/hbase/util/hbck/TableLockChecker.java Fri Apr 12 22:32:51 2013
@@ -0,0 +1,86 @@
+/**
+ * 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
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * 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.util.hbck;
+
+import java.io.IOException;
+
+import org.apache.hadoop.hbase.InterProcessLock.MetadataHandler;
+import org.apache.hadoop.hbase.master.TableLockManager;
+import org.apache.hadoop.hbase.protobuf.ProtobufUtil;
+import org.apache.hadoop.hbase.protobuf.generated.ZooKeeperProtos;
+import org.apache.hadoop.hbase.util.Bytes;
+import org.apache.hadoop.hbase.util.EnvironmentEdgeManager;
+import org.apache.hadoop.hbase.util.HBaseFsck;
+import org.apache.hadoop.hbase.util.HBaseFsck.ErrorReporter;
+import org.apache.hadoop.hbase.zookeeper.ZooKeeperWatcher;
+
+/**
+ * Utility to check and fix table locks. Need zookeeper connection.
+ */
+public class TableLockChecker {
+
+  private ZooKeeperWatcher zkWatcher;
+  private ErrorReporter errorReporter;
+  long expireTimeout;
+
+  public TableLockChecker(ZooKeeperWatcher zkWatcher, ErrorReporter errorReporter) {
+    this.zkWatcher = zkWatcher;
+    this.errorReporter = errorReporter;
+    expireTimeout = zkWatcher.getConfiguration().getLong(
+        TableLockManager.TABLE_LOCK_EXPIRE_TIMEOUT,
+        TableLockManager.DEFAULT_TABLE_LOCK_EXPIRE_TIMEOUT_MS);
+  }
+
+  public void checkTableLocks() throws IOException {
+    TableLockManager tableLockManager
+      = TableLockManager.createTableLockManager(zkWatcher.getConfiguration(), zkWatcher, null);
+    final long expireDate = EnvironmentEdgeManager.currentTimeMillis() - expireTimeout;
+
+    MetadataHandler handler = new MetadataHandler() {
+      @Override
+      public void handleMetadata(byte[] ownerMetadata) {
+        ZooKeeperProtos.TableLock data = TableLockManager.fromBytes(ownerMetadata);
+        String msg = "Table lock acquire attempt found:";
+        if (data != null) {
+           msg = msg +
+              String.format("[tableName=%s, lockOwner=%s, threadId=%s, " +
+              "purpose=%s, isShared=%s, createTime=%s]", Bytes.toString(data.getTableName().toByteArray()),
+              ProtobufUtil.toServerName(data.getLockOwner()), data.getThreadId(),
+              data.getPurpose(), data.getIsShared(), data.getCreateTime());
+        }
+
+        if (data.hasCreateTime() && data.getCreateTime() < expireDate) {
+          errorReporter.reportError(HBaseFsck.ErrorReporter.ERROR_CODE.EXPIRED_TABLE_LOCK, msg);
+        } else {
+          errorReporter.print(msg);
+        }
+      }
+    };
+
+    tableLockManager.visitAllLocks(handler);
+  }
+
+  public void fixExpiredTableLocks() throws IOException {
+    TableLockManager tableLockManager
+      = TableLockManager.createTableLockManager(zkWatcher.getConfiguration(), zkWatcher, null);
+
+    tableLockManager.reapAllExpiredLocks();
+  }
+
+}