1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 

<?php

namespace SAREhub\Commons\Zmq;


use SAREhub\Commons\Misc\Dsn;

abstract class ZmqSocketSupport
{

    /**
     * @var Dsn
     */
    private $binded = null;

    /**
     * @var Dsn[]
     */
    private $connections = [];

    /**
     * @var \ZMQSocket
     */
    private $socket;

    public function __construct(\ZMQSocket $socket)
    {
        $this->socket = $socket;
    }

    /**
     * @param Dsn $dsn
     * @return $this
     * @throws \LogicException \ZMQException
     */
    public function bind(Dsn $dsn)
    {
        if ($this->isBinded()) {
            throw new \LogicException("socket can't be bind second time");
        }

        $this->getSocket()->bind((string)$dsn);
        $this->binded = $dsn;

        return $this;
    }

    /**
     * @throws \ZMQException
     */
    public function unbind()
    {
        if ($this->isBinded()) {
            $this->getSocket()->unbind((string)$this->getBinded());
            $this->binded = null;
        }

        return $this;
    }

    /**
     * @return bool
     */
    public function isBinded()
    {
        return $this->binded !== null;
    }

    /**
     * @param Dsn $dsn
     * @return $this
     */
    public function connect(Dsn $dsn)
    {
        if (!$this->isConnectedTo($dsn)) {
            $this->getSocket()->connect((string)$dsn);
            $this->connections[(string)$dsn] = $dsn;
        }

        return $this;
    }

    /**
     * @param Dsn $dsn
     * @return $this
     */
    public function disconnect(Dsn $dsn)
    {
        if ($this->isConnectedTo($dsn)) {
            $this->getSocket()->disconnect((string)$dsn);
            unset($this->connections[(string)$dsn]);
        }

        return $this;
    }

    /**
     * @return $this
     */
    public function disconnectAll()
    {
        foreach ($this->getConnections() as $connection) {
            $this->disconnect($connection);
        }

        return $this;
    }

    public function close()
    {
        $this->disconnectAll();
        $this->unbind();
    }

    /**
     * @return bool
     */
    public function isConnectedToAny()
    {
        return !empty($this->getConnections());
    }

    /**
     * @param Dsn $dsn
     * @return bool
     */
    public function isConnectedTo(Dsn $dsn)
    {
        return isset($this->connections[(string)$dsn]);
    }

    public function isBindedOrConnected()
    {
        return $this->isBinded() || $this->isConnectedToAny();
    }

    /**
     * @return \ZMQSocket
     */
    public function getSocket()
    {
        return $this->socket;
    }

    /**
     * @return Dsn
     */
    public function getBinded()
    {
        return $this->binded;
    }

    /**
     * @return Dsn[]
     */
    public function getConnections()
    {
        return $this->connections;
    }
}