tom: add continuous backup feature

tombburnell · Sep 9, 2014 · 4910f84 · 4910f84
1 parent cc2d636
commit 4910f84
Showing 1 changed file with 61 additions and 27 deletions.
diff --git a/bin/etcd-dump.js b/bin/etcd-dump.js
@@ -14,38 +14,72 @@ var pkg = require('../package.json');
 
 // General options
 prog
-.version(pkg.version)
-.option('-f, --file [json_file]', 'Path to JSON dump file for dumping/storing', './etcd_dump.json')
-.option('-h, --host [host]', 'Hostname to connect to', 'localhost')
-.option('-p, --port [port]', 'Port to connect to', '4001')
+    .version(pkg.version)
+    .option('-f, --file [json_file]', 'Path to JSON dump file for dumping/storing', 'etcd_dump.json')
+    .option('-h, --host [host]', 'Hostname to connect to', 'localhost')
+    .option('-p, --port [port]', 'Port to connect to', '4001')
+    .option('-m, --minutes [minutes]', 'inteval in minutes for continuous backup', '30')
 ;
 
+function newDumper() {
+    return require('../')(prog.host, prog.port);
+}
+
 // Dump command
 prog
-.command('dump')
-.action(function() {
-    var dumper = require('../')(prog.host, prog.port);
-
-    return dumper.dump()
-    .then(function(data) {
-        // Write file to disk
-        fs.writeFileSync(prog.file, JSON.stringify(data));
-    })
-    .done();
-});
+    .command('dump')
+    .action(function () {
+        var dumper = newDumper();
+
+        return dumper.dump()
+            .then(function (data) {
+                // Write file to disk
+                fs.writeFileSync(prog.file, JSON.stringify(data));
+            })
+            .done();
+    });
+
+prog
+    .command('restore')
+    .action(function () {
+        var dumper = newDumper();
+        var entries = JSON.parse(fs.readFileSync(prog.file));
+
+        return dumper.restore(entries)
+            .then(function () {
+                console.log('Restore succeeded');
+            })
+            .done();
+    });
 
 prog
-.command('restore')
-.action(function() {
-    var dumper = require('../')(prog.host, prog.port);
-    var entries = JSON.parse(fs.readFileSync(prog.file));
-
-    return dumper.restore(entries)
-    .then(function() {
-        console.log('Restore succeeded');
-    })
-    .done();
-});
+    .command('continuous')
+    .action(function () {
+
+        var MINUTES = prog.minutes;
+        var PREFIX = '/var/'
+        var interval = MINUTES * 60 * 1000;
+
+        var dumper = newDumper();
+        function performDump() {
+            var date = new Date().toISOString().replace(/[T]+/, '_').replace(/\..*$/, '');
+            var file = prog.file.replace(/\.([^\.]*)$/, '.' + date + '.$1');
+
+            console.log("Dumping to file", file);
+            return dumper.dump()
+                .then(function (data) {
+                    fs.writeFileSync(file, JSON.stringify(data));
+                })
+                .done();
+        }
+
+        setInterval(function () {
+            performDump();
+        }, interval);
+
+        performDump();
+
+    });
 
 // Parse and fallback to help if no args
-if(_.isEmpty(prog.parse(process.argv).args)) prog.help();
+if (_.isEmpty(prog.parse(process.argv).args)) prog.help();