cassandra-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From jaa...@apache.org
Subject svn commit: r903933 - /incubator/cassandra/trunk/src/java/org/apache/cassandra/tools/ClusterCmd.java
Date Thu, 28 Jan 2010 01:35:23 GMT
Author: jaakko
Date: Thu Jan 28 01:35:23 2010
New Revision: 903933

URL: http://svn.apache.org/viewvc?rev=903933&view=rev
Log:
add global_snapshot and clear_global_snapshot to clustertool. patch by jaakko, reviewed by
jbellis. CASSANDRA-723

Modified:
    incubator/cassandra/trunk/src/java/org/apache/cassandra/tools/ClusterCmd.java

Modified: incubator/cassandra/trunk/src/java/org/apache/cassandra/tools/ClusterCmd.java
URL: http://svn.apache.org/viewvc/incubator/cassandra/trunk/src/java/org/apache/cassandra/tools/ClusterCmd.java?rev=903933&r1=903932&r2=903933&view=diff
==============================================================================
--- incubator/cassandra/trunk/src/java/org/apache/cassandra/tools/ClusterCmd.java (original)
+++ incubator/cassandra/trunk/src/java/org/apache/cassandra/tools/ClusterCmd.java Thu Jan
28 01:35:23 2010
@@ -19,6 +19,7 @@
 
 import java.io.IOException;
 import java.util.List;
+import java.util.Set;
 import java.net.InetAddress;
 import org.apache.commons.cli.CommandLine;
 import org.apache.commons.cli.CommandLineParser;
@@ -138,7 +139,7 @@
     {
         HelpFormatter hf = new HelpFormatter();
         String header = String.format(
-                "%nAvailable commands: get_endpoints [key]");
+                "%nAvailable commands: get_endpoints [key], global_snapshot [name], clear_global_snapshot");
         String usage = String.format("java %s -host <arg> <command>%n", ClusterCmd.class.getName());
         hf.printHelp(usage, "", options, header);
     }
@@ -150,15 +151,83 @@
         System.out.println(String.format("%-17s: %s", "Endpoints", endpoints));
     }
 
+     /**
+     * Take a snapshot of all tables on all (live) nodes in the cluster
+     *
+     * @param snapshotName name of the snapshot
+     */
+    public void takeGlobalSnapshot(String snapshotName) throws IOException, InterruptedException
+    {
+        Set<String> liveNodes = probe.getLiveNodes();
+        try
+        {
+            probe.takeSnapshot(snapshotName);
+            System.out.println(host + " snapshot taken");
+        }
+        catch (IOException e)
+        {
+            System.out.println(host + " snapshot FAILED: " + e.getMessage());
+        }
+
+        liveNodes.remove(this.host);
+        for (String liveNode : liveNodes)
+        {
+            try
+            {
+                this.host = liveNode;
+                probe = new NodeProbe(host, port);
+                probe.takeSnapshot(snapshotName);
+                System.out.println(host + " snapshot taken");
+            }
+            catch (IOException e)
+            {
+                System.out.println(host + " snapshot FAILED: " + e.getMessage());
+            }
+        }
+    }
+
+    /**
+     * Remove all the existing snapshots from all (live) nodes in the cluster
+     */
+    public void clearGlobalSnapshot() throws IOException, InterruptedException
+    {
+        Set<String> liveNodes = probe.getLiveNodes();
+        try
+        {
+            probe.clearSnapshot();
+            System.out.println(host + " snapshot cleared");
+        }
+        catch (IOException e)
+        {
+            System.out.println(host + " snapshot clear FAILED: " + e.getMessage());
+        }
+
+        liveNodes.remove(this.host);
+        for (String liveNode : liveNodes)
+        {
+            try
+            {
+                this.host = liveNode;
+                probe = new NodeProbe(host, port);
+                probe.clearSnapshot();
+                System.out.println(host + " snapshot cleared");
+            }
+            catch (IOException e)
+            {
+                System.out.println(host + " snapshot clear FAILED: " + e.getMessage());
+            }
+        }
+    }
+
     /**
      * @param args
      */
     public static void main(String[] args) throws IOException, InterruptedException
     {
-        ClusterCmd probe = null;
+        ClusterCmd clusterCmd = null;
         try
         {
-            probe = new ClusterCmd(args);
+            clusterCmd = new ClusterCmd(args);
         }
         catch (ParseException pe)
         {
@@ -173,7 +242,7 @@
             System.exit(3);
         }
 
-        if (probe.getArgs().length < 1)
+        if (clusterCmd.getArgs().length < 1)
         {
             System.err.println("Missing argument for command.");
             ClusterCmd.printUsage();
@@ -181,7 +250,7 @@
         }
 
         // Execute the requested command.
-        String[] arguments = probe.getArgs();
+        String[] arguments = clusterCmd.getArgs();
         String cmdName = arguments[0];
         if (cmdName.equals("get_endpoints"))
         {
@@ -189,7 +258,20 @@
             {
                 System.err.println("missing key argument");
             }
-            probe.printEndPoints(arguments[1]);
+            clusterCmd.printEndPoints(arguments[1]);
+        }
+	else if (cmdName.equals("global_snapshot"))
+	{
+            String snapshotName = "";
+            if (arguments.length > 1)
+            {
+                snapshotName = arguments[1];
+            }
+	    clusterCmd.takeGlobalSnapshot(snapshotName);
+	}
+        else if (cmdName.equals("clear_global_snapshot"))
+        {
+            clusterCmd.clearGlobalSnapshot();
         }
         else
         {



Mime
View raw message