Merge pull request #39 from kbond/redis-rabbitmq-checks

Add RabbitMQ and Redis checks.

Author: Thinkscape

.travis.yml

@@ -1,5 +1,9 @@
 language: php
 
+services:
+  - redis-server
+  - rabbitmq
+
 php:
   - 5.3
   - 5.3.3

README.md

@@ -22,6 +22,8 @@ It currently ships with the following Diagnostic Checks:
  * [PhpVersion](#phpversion) - make sure that PHP version matches constraint,
  * [PhpFlag](#phpflag) - make sure that given PHP flag (feature) is turned on or off.
  * [ProcessRunning](#processrunning) - check if a process with given name or ID is currently running,
+ * [RabbitMQ](#rabbitmq) - Validate that a RabbitMQ service is running,
+ * [Redis](#redis) - Validate that a Redis service is running,
  * [SecurityAdvisory](#securityadvisory) - check installed composer dependencies against SensioLabs SA database,
  * [StreamWrapperExists](#streamwrapperexists) - make sure given stream wrapper is available.
 
@@ -510,6 +512,28 @@ $checkApache = new ProcessRunning('httpd');
 $checkProcess1000 = new ProcessRunning(1000);
 ````
 
+### RabbitMQ
+
+Validate that a RabbitMQ service is running.
+
+```php
+<?php
+use ZendDiagnostics\Check\RabbitMQ;
+
+$rabbitMQCheck = new RabbitMQ('localhost', 5672, 'guest', 'guest', '/');
+```
+
+### Redis
+
+Validate that a Redis service is running.
+
+```php
+<?php
+use ZendDiagnostics\Check\Redis;
+
+$redisCheck = new Redis('localhost', 6379);
+```
+
 ### SecurityAdvisory
 
 Run a security check of libraries locally installed by [Composer](http://getcomposer.org) against

composer.json

@@ -20,15 +20,19 @@
     "require-dev": {
         "zendframework/zend-loader"  : "2.*",
         "sensiolabs/security-checker": "1.3.*@dev",
-        "symfony/yaml": "v2.3.11",
+        "symfony/yaml"               : "v2.3.11",
         "guzzle/http"                : "3.*",
-        "guzzle/plugin-mock"         : "3.*"
+        "guzzle/plugin-mock"         : "3.*",
+        "videlalvaro/php-amqplib"    : "2.*",
+        "predis/predis"              : "0.8.*"
     },
     "suggest" : {
         "ext-bcmath" : "Required by Check\\CpuPerformance",
         "sensiolabs/security-checker": "Required by Check\\SecurityAdvisory",
         "symfony/yaml": "Required by Check\\YamlFile",
-        "guzzle/http": "Required by Check\\GuzzleHttpService"
+        "guzzle/http": "Required by Check\\GuzzleHttpService",
+        "predis/predis": "Required by Check\\Redis",
+        "videlalvaro/php-amqplib": "Required by Check\\RabbitMQ"
     },
     "extra": {
         "branch-alias": {

src/ZendDiagnostics/Check/RabbitMQ.php

@@ -0,0 +1,87 @@
+<?php
+/**
+ * @license   http://framework.zend.com/license/new-bsd New BSD License
+ */
+
+namespace ZendDiagnostics\Check;
+
+use PhpAmqpLib\Connection\AMQPConnection;
+use ZendDiagnostics\Result\Failure;
+use ZendDiagnostics\Result\Success;
+
+/**
+ * Validate that a RabbitMQ service is running
+ */
+class RabbitMQ extends AbstractCheck
+{
+    /**
+     * @var string
+     */
+    protected $host;
+
+    /**
+     * @var integer
+     */
+    protected $port;
+
+    /**
+     * @var string
+     */
+    protected $user;
+
+    /**
+     * @var string
+     */
+    protected $password;
+
+    /**
+     * @var string
+     */
+    protected $vhost;
+
+    /**
+     * @param string  $host
+     * @param integer $port
+     * @param string  $user
+     * @param string  $password
+     * @param string  $vhost
+     */
+    public function __construct(
+        $host = 'localhost',
+        $port = 5672,
+        $user = 'guest',
+        $password = 'guest',
+        $vhost = '/'
+    ) {
+        $this->host     = $host;
+        $this->port     = $port;
+        $this->user     = $user;
+        $this->password = $password;
+        $this->vhost    = $vhost;
+    }
+
+    /**
+     * Perform the check
+     *
+     * @see \ZendDiagnostics\Check\CheckInterface::check()
+     * @return Failure|Success
+     */
+    public function check()
+    {
+        if (!class_exists('PhpAmqpLib\Connection\AMQPConnection')) {
+            return new Failure('PhpAmqpLib is not installed');
+        }
+
+        $conn = new AMQPConnection(
+            $this->host,
+            $this->port,
+            $this->user,
+            $this->password,
+            $this->vhost
+        );
+
+        $conn->channel();
+
+        return new Success();
+    }
+}

src/ZendDiagnostics/Check/Redis.php

@@ -0,0 +1,58 @@
+<?php
+/**
+ * @license   http://framework.zend.com/license/new-bsd New BSD License
+ */
+
+namespace ZendDiagnostics\Check;
+
+use Predis\Client;
+use ZendDiagnostics\Result\Failure;
+use ZendDiagnostics\Result\Success;
+
+/**
+ * Validate that a Redis service is running
+ */
+class Redis extends AbstractCheck
+{
+    /**
+     * @var string
+     */
+    protected $host;
+
+    /**
+     * @var int
+     */
+    protected $port;
+
+    /**
+     * @param string  $host
+     * @param int $port
+     */
+    public function __construct($host = 'localhost', $port = 6379)
+    {
+        $this->host = $host;
+        $this->port = $port;
+    }
+
+    /**
+     * Perform the check
+     *
+     * @see \ZendDiagnostics\Check\CheckInterface::check()
+     * @return Failure|Success
+     */
+    public function check()
+    {
+        if (!class_exists('Predis\Client')) {
+            return new Failure('Predis is not installed');
+        }
+
+        $client = new Client(array(
+            'host' => $this->host,
+            'port' => $this->port,
+        ));
+
+        $client->ping();
+
+        return new Success();
+    }
+}

tests/ZendDiagnosticsTest/ChecksTest.php

@@ -15,6 +15,8 @@
 use ZendDiagnostics\Check\PhpFlag;
 use ZendDiagnostics\Check\PhpVersion;
 use ZendDiagnostics\Check\ProcessRunning;
+use ZendDiagnostics\Check\RabbitMQ;
+use ZendDiagnostics\Check\Redis;
 use ZendDiagnostics\Check\StreamWrapperExists;
 use ZendDiagnostics\Check\XmlFile;
 use ZendDiagnostics\Check\YamlFile;
@@ -43,6 +45,28 @@ public function testCpuPerformance()
         $this->assertInstanceOf('ZendDiagnostics\Result\Failure', $result);
     }
 
+    public function testRabbitMQ()
+    {
+        $check = new RabbitMQ();
+        $result = $check->check();
+        $this->assertInstanceOf('ZendDiagnostics\Result\Success', $result);
+
+        $check = new RabbitMQ('127.0.0.250', 9999);
+        $this->setExpectedException('PhpAmqpLib\Exception\AMQPRuntimeException');
+        $check->check();
+    }
+
+    public function testRedis()
+    {
+        $check = new Redis();
+        $result = $check->check();
+        $this->assertInstanceOf('ZendDiagnostics\Result\Success', $result);
+
+        $check = new Redis('127.0.0.250', 9999);
+        $this->setExpectedException('Predis\Connection\ConnectionException');
+        $check->check();
+    }
+
     public function testClassExists()
     {
         $check = new ClassExists(__CLASS__);