Example #1
0
 /**
  * @param resource $socket An open socket client resource
  */
 public function __construct($socket)
 {
     \stream_set_blocking($socket, false);
     $this->state = $state = new \StdClass();
     $readWatcherId = amp\onReadable($socket, static function ($wid, $socket) use($state) {
         $data = @\fread($socket, 8192);
         if ($data != "") {
             $state->bytesRead += \strlen($data);
             $op = \reset($state->readOperations);
             $op->buffer .= $data;
             Client::onRead($state);
         } else {
             Client::onEmptyRead($state);
         }
     }, $options = ["enable" => false]);
     $writeWatcherId = amp\onWritable($socket, static function ($wid, $socket) use($state) {
         $op = \current($state->writeOperations);
         if ($bytes = @\fwrite($socket, $op->buffer)) {
             $state->bytesSent += $bytes;
             Client::onWrite($state, $op, $bytes);
         } else {
             Client::onEmptyWrite($state);
         }
     }, $options = ["enable" => false]);
     $state->readWatcherId = $readWatcherId;
     $state->writeWatcherId = $writeWatcherId;
     $state->socket = $socket;
     $state->isDead = false;
     $state->localName = \stream_socket_get_name($socket, $wantPeer = false);
     $state->remoteName = \stream_socket_get_name($socket, $wantPeer = true);
     $state->readOperations = [];
     $state->writeOperations = [];
     $state->bytesRead = 0;
     $state->bytesSent = 0;
     // We avoid instantiating a closure every time a socket read/write completes
     // without exposing a method in the public API this way ... it may look hacky
     // but it's important for performance.
     if (empty(self::$succeeder)) {
         $callable = (new \ReflectionClass($this))->getMethod("succeed")->getClosure($this);
         self::$succeeder = $callable;
     }
 }
Example #2
0
 private function connect()
 {
     // If we're in the process of connecting already return that same promise
     if ($this->connectPromisor) {
         return $this->connectPromisor->promise();
     }
     // If a read watcher exists we know we're already connected
     if ($this->readWatcher) {
         return new Success($this);
     }
     $this->connectPromisor = new Deferred();
     $socketPromise = connect($this->uri, ["timeout" => $this->timeout]);
     $onWrite = function ($watcherId) {
         if ($this->outputBufferLength === 0) {
             disable($watcherId);
             return;
         }
         $bytes = @fwrite($this->socket, $this->outputBuffer);
         if ($bytes === 0) {
             $this->onError(new ConnectException("Connection went away (write)", $code = 1));
         } else {
             $this->outputBuffer = (string) substr($this->outputBuffer, $bytes);
             $this->outputBufferLength -= $bytes;
         }
     };
     $socketPromise->when(function ($error, $socket) use($onWrite) {
         $connectPromisor = $this->connectPromisor;
         $this->connectPromisor = null;
         if ($error) {
             $connectPromisor->fail(new ConnectException("Connection attempt failed", $code = 0, $error));
             return;
         }
         $this->socket = $socket;
         foreach ($this->handlers["connect"] as $handler) {
             $pipelinedCommand = $handler();
             if (!empty($pipelinedCommand)) {
                 $this->outputBuffer = $pipelinedCommand . $this->outputBuffer;
                 $this->outputBufferLength += strlen($pipelinedCommand);
             }
         }
         $this->readWatcher = onReadable($this->socket, function () {
             $read = fread($this->socket, 8192);
             if ($read != "") {
                 $this->parser->append($read);
             } elseif (!is_resource($this->socket) || @feof($this->socket)) {
                 $this->onError(new ConnectException("Connection went away (read)", $code = 2));
             }
         });
         $this->writeWatcher = onWritable($this->socket, $onWrite, ["enable" => !empty($this->outputBuffer)]);
         $connectPromisor->succeed();
     });
     return $this->connectPromisor->promise();
 }