You are viewing a plain text version of this content. The canonical link for it is here.
Posted to common-commits@hadoop.apache.org by st...@apache.org on 2008/01/06 07:12:03 UTC
svn commit: r609256 -
/lucene/hadoop/trunk/src/contrib/hbase/src/java/org/apache/hadoop/hbase/shell/TruncateCommand.java
Author: stack
Date: Sat Jan 5 22:11:55 2008
New Revision: 609256
URL: http://svn.apache.org/viewvc?rev=609256&view=rev
Log:
HADOOP-2240 Truncate for hbase
Added:
lucene/hadoop/trunk/src/contrib/hbase/src/java/org/apache/hadoop/hbase/shell/TruncateCommand.java
Added: lucene/hadoop/trunk/src/contrib/hbase/src/java/org/apache/hadoop/hbase/shell/TruncateCommand.java
URL: http://svn.apache.org/viewvc/lucene/hadoop/trunk/src/contrib/hbase/src/java/org/apache/hadoop/hbase/shell/TruncateCommand.java?rev=609256&view=auto
==============================================================================
--- lucene/hadoop/trunk/src/contrib/hbase/src/java/org/apache/hadoop/hbase/shell/TruncateCommand.java (added)
+++ lucene/hadoop/trunk/src/contrib/hbase/src/java/org/apache/hadoop/hbase/shell/TruncateCommand.java Sat Jan 5 22:11:55 2008
@@ -0,0 +1,86 @@
+/**
+ * Copyright 2007 The Apache Software Foundation
+ *
+ * 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.shell;
+
+import java.io.IOException;
+import java.io.Writer;
+
+import org.apache.hadoop.hbase.HBaseAdmin;
+import org.apache.hadoop.hbase.HBaseConfiguration;
+import org.apache.hadoop.hbase.HColumnDescriptor;
+import org.apache.hadoop.hbase.HConnection;
+import org.apache.hadoop.hbase.HConnectionManager;
+import org.apache.hadoop.hbase.HTableDescriptor;
+import org.apache.hadoop.io.Text;
+
+/**
+ * Truncate table is used to clean all data from a table.
+ */
+public class TruncateCommand extends BasicCommand {
+ private Text tableName;
+
+ public TruncateCommand(Writer o) {
+ super(o);
+ }
+
+ public ReturnMsg execute(final HBaseConfiguration conf) {
+ if (this.tableName == null)
+ return new ReturnMsg(0, "Syntax error : Please check 'Truncate' syntax.");
+
+ try {
+ HConnection conn = HConnectionManager.getConnection(conf);
+ HBaseAdmin admin = new HBaseAdmin(conf);
+
+ if (!conn.tableExists(this.tableName)) {
+ return new ReturnMsg(0, "Table not found.");
+ }
+
+ HTableDescriptor[] tables = conn.listTables();
+ HColumnDescriptor[] columns = null;
+ for (int i = 0; i < tables.length; i++) {
+ if (tables[i].getName().equals(this.tableName)) {
+ columns = tables[i].getFamilies().values().toArray(
+ new HColumnDescriptor[] {});
+ break;
+ }
+ }
+ println("Truncating a '" + tableName + "' table ... Please wait.");
+
+ admin.deleteTable(tableName); // delete the table
+ HTableDescriptor tableDesc = new HTableDescriptor(tableName.toString());
+ for (int i = 0; i < columns.length; i++) {
+ tableDesc.addFamily(columns[i]);
+ }
+ admin.createTable(tableDesc); // re-create the table
+ } catch (IOException e) {
+ return new ReturnMsg(0, "error msg : " + e.toString());
+ }
+ return new ReturnMsg(0, "'" + tableName + "' is successfully truncated.");
+ }
+
+ public void setTableName(String tableName) {
+ this.tableName = new Text(tableName);
+ }
+
+ @Override
+ public CommandType getCommandType() {
+ return CommandType.DDL;
+ }
+}