Merge pull request #4 from OpenTMI/workers_tune

tuning cli util

Author: Jussi Vatjus-Anttila

src/arguments.js

@@ -5,7 +5,8 @@ const yargs = require('yargs');
 
 // app modules
 const {authenticateUser, logoutUser} = require('./cmds/login');
-const {showVersion, updateVersion} = require('./cmds/update');
+const {showVersion, updateVersion} = require('./cmds/version');
+const {showWorkers, reloadWorkers} = require('./cmds/cluster');
 const config = require('./config');
 
 
@@ -14,10 +15,14 @@ yargs.usage('Usage: $0 <command> [options]')
   .command('logout', 'Logout user', () => {}, logoutUser)
   .command('show', 'Show something from server', argv =>
     argv
-      .command('version', 'Show server version', () => {}, showVersion))
+      .command('version', 'Show server version', () => {}, showVersion)
+      .command('workers', 'Show workers status', () => {}, showWorkers))
   .command('update', 'Update something from server', argv =>
     argv
       .command('version <revision>', 'Update server version', () => {}, updateVersion))
+  .command('restart', 'Restart something from server', argv =>
+    argv
+      .command('workers', 'Restart workers', () => {}, reloadWorkers))
   .config(config.getState())
   .demandCommand()
   .alias('h', 'help')

src/cmds/cluster.js

@@ -0,0 +1,44 @@
+const {Transport, Admin, Cluster} = require('opentmi-jsclient');
+
+
+const showWorkers = (argv) => {
+  const transport = new Transport(argv.host);
+  transport.token = argv.token;
+  const cluster = new Cluster(transport);
+  console.log('Start reloading..');
+  cluster.refresh()
+    .then(() => {
+      const {master} = cluster.status;
+      const {
+        requireRestart, pid, coresUsed, uptime, memoryUsage
+      } = master;
+      console.log(`PID: ${pid}, coreUsed: ${coresUsed}`);
+      console.log(`uptime: ${uptime}`);
+      console.log(memoryUsage);
+      console.log(`state: ${requireRestart ? 'require restart workers' : 'OK'}`);
+      const connected = cluster.workers.filter(w => w.isConnected);
+      console.log(`Workers ${connected.length} / ${cluster.workers.length} connected`);
+    })
+    .catch((error) => {
+      console.error(`failed: ${error.message}`);
+    });
+};
+
+const reloadWorkers = (argv) => {
+  const transport = new Transport(argv.host);
+  transport.token = argv.token;
+  const admin = new Admin(transport);
+  console.log('Start reloading..');
+  admin.reloadWorkers()
+    .then(() => {
+      console.log('Workers reloaded');
+    })
+    .catch((error) => {
+      console.error(`failed: ${error.message}`);
+    });
+};
+
+module.exports = {
+  showWorkers,
+  reloadWorkers
+};