Kwkm\MkLiveStatusClient\Client

Complexity

Total Complexity

25

Size/Duplication

Total Lines	211
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Test Coverage

Coverage

0%

Importance

Changes	9
Bugs	2	Features	0

11 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	execute()	0	14	2
A	executeRequest()	0	17	1
A	verifyStatusCode()	0	11	2
A	reset()	0	4	1
A	openSocket()	0	16	4
A	openUnixSocket()	0	15	3
A	openTcpSocket()	0	17	3
A	setSocketTimeout()	0	7	2
A	closeSocket()	0	7	2
A	readSocket()	0	21	4

<?php
namespace Kwkm\MkLiveStatusClient;

use \RuntimeException;

/**
 * Class Client
 *
 * @package Kwkm\MkLiveStatusClient
 * @author Takehiro Kawakami <take@kwkm.org>
 * @license MIT
 */
class Client
{
    /**
     * 設定
     * @var Configuration
     * @link http://php.net/manual/en/function.socket-set-option.php
     */
    protected $config;

    /**
     * ソケットのリソース
     * @var resource
     */
    protected $socket = null;

    /**
     * コンストラクタ
     *
     * @param Configuration $conf
     * @throw \BadFunctionCallException
     * @throw \InvalidArgumentException
     */
    public function __construct(Configuration $conf)
    {
        $this->config = $conf;

        $this->reset();
    }

    /**
     * Lqlの実行
     *
     * @param \Kwkm\MkLiveStatusClient\LqlAbstract $lql
     * @return array
     * @throw \RuntimeException
     */
    public function execute(LqlAbstract $lql)
    {
        $result = $this->executeRequest($lql->build());

        $this->verifyStatusCode($result);

        $response = utf8_encode($result['response']);

        if (is_null($response)) {
            throw new RuntimeException("The response was invalid.");
        }

        return $response;
    }

    /**
     * リクエストの発行
     * @param $query
     * @return array
     */
    private function executeRequest($query)
    {
        $this->openSocket();
        socket_write($this->socket, $query);
        // Read 16 bytes to get the status code and body size
        $header = $this->readSocket(16);
        $status = substr($header, 0, 3);
        $length = intval(trim(substr($header, 4, 11)));
        $response = $this->readSocket($length);
        $this->closeSocket();

        return array(
            'status' => $status,
            'length' => $length,
            'response' => $response,
        );
    }

    /**
     * ステータスコードの確認
     * @param $response
     */
    private function verifyStatusCode($response)
    {
        // Check for errors. A 200 response means request was OK.
        // Any other response is a failure.
        if ($response['status'] != "200") {
            throw new RuntimeException(
                "Error response from Nagios MK Livestatus: " . $response['status'],
                (int) $response['status']
            );
        }
    }

    /**
     * リセット処理
     */
    public function reset()
    {
        $this->closeSocket();
    }

    /**
     * 接続の開始
     *
     * @throw \RuntimeException
     */
    protected function openSocket()
    {
        if (!is_null($this->socket)) {
            // Assume socket still good and continue
            return;
        }

        switch ($this->config->socketType) {
            case 'unix':
                $this->openUnixSocket();
                break;
            case 'tcp':
                $this->openTcpSocket();
                break;
        }
    }

    /**
     * Unixソケットの接続を開始
     */
    private function openUnixSocket()
    {
        $this->socket = socket_create(AF_UNIX, SOCK_STREAM, 0);
        if (!$this->socket) {
            $this->socket = null;
            throw new RuntimeException("Could not create socket.");
        }

        if (false === socket_connect($this->socket, $this->config->socketPath)) {
            $this->closeSocket();
            throw new RuntimeException("Unable to connect to socket.");
        }

        $this->setSocketTimeout();
    }

    /**
     * Tcpソケットの接続を開始
     */
    private function openTcpSocket()
    {
        $this->socket = socket_create(AF_INET, SOCK_STREAM, SOL_TCP);
        if (!$this->socket) {
            $this->socket = null;
            throw new RuntimeException("Could not create socket.");
        }

        if (false === socket_connect($this->socket, $this->config->socketAddress, $this->config->socketPort)) {
            $this->closeSocket();
            throw new RuntimeException("Unable to connect to socket.");
        }

        socket_set_option($this->socket, SOL_TCP, TCP_NODELAY, 1);

        $this->setSocketTimeout();
    }

    /**
     * ソケットタイムアウトの設定
     */
    private function setSocketTimeout()
    {
        if (count($this->config->socketTimeout) !== 0) {
            socket_set_option($this->socket, SOCK_STREAM, SO_RCVTIMEO, $this->config->socketTimeout);
            socket_set_option($this->socket, SOCK_STREAM, SO_SNDTIMEO, $this->config->socketTimeout);
        }
    }

    /**
     * 接続の切断
     */
    protected function closeSocket()
    {
        if (is_resource($this->socket)) {
            socket_close($this->socket);
        }
        $this->socket = null;
    }

    /**
     * ソケットの読み出し
     *
     * @param integer $length
     * @return string
     * @throw \RuntimeException
     */
    protected function readSocket($length)
    {
        $offset = 0;
        $socketData = "";
        while ($offset < $length) {
            if (false === ($data = socket_read($this->socket, $length - $offset))) {
                throw new RuntimeException(
                    "Problem reading from socket: "
                    . socket_strerror(socket_last_error($this->socket))
                );
            }
            $dataLen = strlen($data);
            $offset += $dataLen;
            $socketData .= $data;
            if ($dataLen == 0) {
                break;
            }
        }

        return $socketData;
    }
}