eswyft/php-jobd-client
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use socket_* family of functions

Browse Source
This commit is contained in:
Evgeny Zinoviev 2021-03-03 17:46:32 +03:00
parent 0336bc61bf
commit 3bd1d7b536
2 changed files with 51 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
composer.json
View File

@ -16,6 +16,7 @@
},
"require": {
"php": ">=7.0",
"ext-json": "*"
"ext-json": "*",
"ext-sockets": "*"
}
}

61
src/Client.php
View File

@ -31,12 +31,15 @@ class Client {
$this->host = $host;
$this->password = $password;
$this->sock = fsockopen($this->host, $this->port);
if (!$this->sock)
throw new Exception("Failed to connect to {$this->host}:{$this->port}");
if (($socket = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)) === false)
throw new Exception("socket_create() failed: ".$this->getSocketError());
// 0 is reserved
$this->lastOutgoingRequestNo = mt_rand(1, self::REQUEST_NO_LIMIT);
$this->sock = $socket;
if ((socket_connect($socket, $host, $port)) === false)
throw new Exception("socket_connect() failed: ".$this->getSocketError());
$this->lastOutgoingRequestNo = mt_rand(1 /* 0 is reserved */, self::REQUEST_NO_LIMIT);
}
/**
@ -110,11 +113,22 @@ class Client {
/**
* @param Message $message
* @throws Exception
*/
public function send(Message $message)
{
$serialized = $message->serialize();
fwrite($this->sock, $serialized . self::EOT);
$data = $message->serialize() . self::EOT;
$remained = strlen($data);
while ($remained > 0) {
$result = socket_write($this->sock, $data);
if ($result === false)
throw new Exception(__METHOD__ . ": socket_write() failed: ".$this->getSocketError());
$remained -= $result;
if ($remained > 0)
$data = substr($data, $result);
}
}
/**
@ -124,15 +138,27 @@ class Client {
*/
public function recv(int $request_no = -1)
{
$messages = [];
$recv_buf = '';
$buf = '';
while (!feof($this->sock)) {
$buf .= fread($this->sock, 1024);
$buflen = 0;
while (true) {
$result = socket_recv($this->sock, $recv_buf, 1024, 0);
if ($result === false)
throw new Exception(__METHOD__ . ": socket_recv() failed: " . $this->getSocketError());
// peer disconnected
if ($result === 0)
break;
$buf .= $recv_buf;
$buflen = strlen($buf);
if ($buflen > 0 && $buf[$buflen-1] == self::EOT)
if ($buf[$buflen-1] == self::EOT)
break;
}
$messages = [];
$offset = 0;
$eot_pos = 0;
do {
@ -340,8 +366,19 @@ class Client {
if (!$this->sock)
return;
fclose($this->sock);
socket_close($this->sock);
$this->sock = null;
}
/**
* @return string
*/
protected function getSocketError()
{
$sle_args = [];
if ($this->sock !== null)
$sle_args[] = $this->sock;
return socket_strerror(socket_last_error(...$sle_args));
}
}