Added memory check

mda · mda · commit 13adf54550a7 · 2015-05-22T16:33:29.000+02:00
diff --git a/README.md b/README.md
@@ -1,6 +1,6 @@
 ## nagios-check-redis
 
-Simple plugin for nagios to check status of a redis server.
+Simple plugin for nagios to check status or memory usage of a redis server.
 
 ## Requirements
 
@@ -15,29 +15,47 @@ To download the latest version:
 
 ## Usage
 
-	nagios-redis -H localhost -p 6379
+	# check ping
+	check_redis -H localhost -p 6379 -t ping
+	
+	# check ping
+	check_redis -H localhost -p 6379 -t memory -w 1GB -c 2GB
+	
+Note on units: when memory size is needed, it is possible to specify it in the usual form of 1k 5GB 4M and so forth.
 
 Edit your commands.cfg and add the following:
 
 	define command {
-	    command_name    check_redis
-	    command_line    $USER1$/nagios-redis/check_redis.js -H $HOSTADDRESS$ -p $ARG1$
+	    command_name    check_redis_ping
+	    command_line    $USER1$/nagios-redis/check_redis.js -H $HOSTADDRESS$ -p $ARG1$ -t ping
+	}
+
+	define command {
+	    command_name    check_redis_memory
+	    command_line    $USER1$/nagios-redis/check_redis.js -H $HOSTADDRESS$ -p $ARG1$ -t memory -w $ARG2$ -c $ARG3$
 	}
 
 Now you can monitor redis servers by adding:
 
 	define service {
 	    use                 	generic-service
 	    hostgroup_name          Redis servers
-	    service_description     Redis server check
-	    check_command           check_redis!6379
+	    service_description     Redis ping
+	    check_command           check_redis_ping!6379
+	}
+
+	define service {
+	    use                 	generic-service
+	    hostgroup_name          Redis servers
+	    service_description     Redis memory
+	    check_command           check_redis_memory!6379!1GB!2GB
 	}
 
 ## Usage monitoring remote host with NRPE
 
 Add the command to your NRPE configuration on your remote host:
 
-	command[check_redis]=check_redis
+	command[check_redis_ping]=check_redis_ping
 
 Monitor the service on your Nagios host:
 
diff --git a/check_redis.js b/check_redis.js
@@ -2,29 +2,53 @@
 'use strict';
 
 var optimist = require('optimist')
-    .usage('Usage: $0 -H [hostname] -p [port] --help')
+    .usage('Usage: $0 -H [hostname] -p [port] -t [check_type] -w -c --help')
     .default('H', 'localhost')
     .default('p', 6379)
+	.default('t', 'ping')
+	.default('w', '1GB')
+	.default('c', '2GB')
     .describe('H', 'hostname of the redis server')
     .describe('p', 'port of the redis server')
+	.describe('t', 'type to check, ping or memory')
+	.describe('w', 'warning threshold (e.g. 100MB or 1GB, only used when type is memory)')
+	.describe('c', 'critical threshold (e.g. 100MB or 1GB, only used when type is memory)')
     .describe('help', 'prints this help message');
 var argv = optimist.argv;
 var redis = require('redis');
+var bytes = require('bytes');
+
+var exitCode = 0;
 
 if (argv.help) {
     optimist.showHelp();
-    process.exit(0);
+    process.exit(exitCode);
 }
 
 var client = redis.createClient(argv.p, argv.H);
 
 client.on("error", function (err) {
     console.log('CRITICAL - ', err.toString());
-    process.exit(3);
+	exitCode = 2;
 });
 
 client.on('ready', function() {
-    console.log('OK -', 'uptime_in_days:', client.server_info.uptime_in_days);
-    client.end();
-    process.exit(0);
+	if (argv.t === 'ping')
+		console.log('OK -', 'uptime_in_days:', client.server_info.uptime_in_days);
+	if (argv.t === 'memory') {
+		var usage = parseInt(client.server_info.used_memory);
+		var message = bytes(usage) + ' of memory used.';
+
+		if (usage >= bytes(argv.c)) {
+			console.log('CRITICAL -', message);
+			exitCode = 2;
+		} else if (usage >= bytes(argv.w)) {
+			console.log('WARNING -', message);
+			exitCode = 1;
+		} else
+			console.log('OK -', message);
+	}
+
+	client.end();
+	process.exit(exitCode);
 });
diff --git a/package.json b/package.json
@@ -13,6 +13,7 @@
   "author": "Michel Hiemstra <info@michelhiemstra.nl>",
   "license": "none",
   "dependencies": {
+    "bytes": "^2.1.0",
     "hiredis": "^0.3.0",
     "optimist": "^0.6.1",
     "redis": "^0.12.1"
