php使用swoole实现TCP服务

目录

• 一：swoole配置TCP
• 二：swoole实现TCP服务基类
• 三：swoole操作类(继承swoole基类)
• 四：操作TCP服务

'swoole' => [
    // 日志文件路径
    'log_file' => '@console/log/swoole.log',
    // 设置swoole_server错误日志打印的等级，范围是0-5。低于log_level设置的日志信息不会抛出
    'log_level' => 1,
    // 进程的PID存储文件
    'pid_file' => '@console/log/swoole.server.pid',

    // HTTP协议配置
    'http' => [
        'host' => '0.0.0.0',
        'port' => '8889',

        // 异步任务的工作进程数量
        'task_worker_num' => 4,
    ],
    // TCP协议配置
    'tcp' => [
        'host' => '0.0.0.0',
        'port' => '14000',

        // 异步任务的工作进程数量
        'task_worker_num' => 4,

        // 启用TCP-Keepalive死连接检测
        'open_tcp_keepalive' => 1,
        // 单位秒，连接在n秒内没有数据请求，将开始对此连接进行探测
        'tcp_keepidle' => 5 * 60,
        // 探测的次数，超过次数后将close此连接
        'tcp_keepcount' => 3,
        // 探测的间隔时间，单位秒
        'tcp_keepinterval' => 60,

        // 心跳检测，此选项表示每隔多久轮循一次，单位为秒
        'heartbeat_check_interval' => 2 * 60,
        // 心跳检测，连接最大允许空闲的时间
        'heartbeat_idle_time' => 5 * 60,
    ]
],

<?php
/**
 * @link http://www.u-bo.com
 * @copyright 南京友博网络科技有限公司
 * @license http://www.u-bo.com/license/
 */

namespace console\swoole;

use Yii;
use yii\helpers\Console;
use yii\helpers\ArrayHelper;

/*
 * Swoole Server基类
 *
 * @author wangjian
 * @since 0.1
 */
abstract class BaseServer
{
    /**
     * @var Swoole\Server
     */
    public $swoole;
    /**
     * @var boolean DEBUG
     */
    public $debug = false;

    /**
     * __construct
     */
    public function __construct($httpConfig, $tcpConfig, $config = [])
    {
        $httpHost = ArrayHelper::remove($httpConfig, 'host');
        $httpPort = ArrayHelper::remove($httpConfig, 'port');
        $this->swoole = new \swoole_http_server($httpHost, $httpPort);
        $this->swoole->set(ArrayHelper::merge($config, $httpConfig));

        $this->swoole->on('start', [$this, 'onStart']);
        $this->swoole->on('request', [$this, 'onRequest']);
        $this->swoole->on('WorkerStart', [$this, 'onWorkerStart']);
        $this->swoole->on('WorkerStop', [$this, 'onWorkerStop']);
        $this->swoole->on('task', [$this, 'onTask']);
        $this->swoole->on('finish', [$this, 'onTaskFinish']);

        $this->swoole->on('shutdown', [$this, 'onShutdown']);

        $tcpHost = ArrayHelper::remove($tcpConfig, 'host');
        $tcpPort = ArrayHelper::remove($tcpConfig, 'port');
        $tcpServer = $this->swoole->listen($tcpHost, $tcpPort, SWOOLE_SOCK_TCP);
        $tcpServer->set($tcpConfig);
        $tcpServer->on('connect', [$this, 'onConnect']);
        $tcpServer->on('receive', [$this, 'onReceive']);
        $tcpServer->on('close', [$this, 'onClose']);
    }

    /*
     * 启动server
     */
    public function run()
    {
        $this->swoole->start();
    }

    /**
     * Server启动在主进程的主线程时的回调事件处理
     *
     * @param swoole_server $server
     */
    public function onStart(\swoole_server $server)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("**Server Start**\n", Console::FG_GREEN);
        $this->stdout("master_pid: ");
        $this->stdout("{$server->master_pid}\n", Console::FG_BLUE);

        $this->onStartHandle($server);

        $this->afterExec($startedAt);
    }

    /**
     * 客户端与服务器建立连接后的回调事件处理
     *
     * @param swoole_server $server
     * @param integer $fd
     * @param integer $reactorId
     */
    abstract public function onConnect(\swoole_server $server, int $fd, int $reactorId);

    /**
     * 当服务器收到来自客户端的数据时的回调事件处理
     *
     * @param swoole_server $server
     * @param integer $fd
     * @param integer $reactorId
     * @param string $data
     */
    abstract public function onReceive(\swoole_server $server, int $fd, int $reactorId, string $data);

    /**
     * 当服务器收到来自客户端的HTTP请求时的回调事件处理
     *
     * @param swoole_http_request $request
     * @param swoole_http_response $response
     */
    abstract public function onRequest(\swoole_http_request $request, \swoole_http_response $response);

    /**
     * Worker进程/Task进程启动时发生
     *
     * @param swoole_server $server
     * @param integer $worker_id
     */
    abstract public function onWorkerStart(\swoole_server $server, int $worker_id);

    /**
     * Worker进程/Task进程终止时发生
     *
     * @param swoole_server $server
     * @param integer $worker_id
     */
    abstract public function onWorkerStop(\swoole_server $server, int $worker_id);

    /**
     * 异步任务处理
     *
     * @param swoole_server $server
     * @param integer $taskId
     * @param integer $srcWorkerId
     * @param mixed $data
     */
    abstract public function onTask(\swoole_server $server, int $taskId, int $srcWorkerId, mixed $data);

    /**
     * 异步任务处理完成
     *
     * @param swoole_server $server
     * @param integer $taskId
     * @param mixed $data
     */
    abstract public function onTaskFinish(\swoole_server $server, int $taskId, mixed $data);

    /**
     * 客户端与服务器断开连接后的回调事件处理
     *
     * @param swoole_server $server
     * @param integer $fd
     */
    abstract public function onClose(\swoole_server $server, $fd);

    /**
     * Server正常结束时的回调事件处理
     *
     * @param swoole_server $server
     */
    public function onShutdown(\swoole_server $server)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("**Server Stop**\n", Console::FG_GREEN);
        $this->stdout("master_pid: ");
        $this->stdout("{$server->master_pid}\n", Console::FG_BLUE);

        $this->onShutdownHandle($server);

        $this->afterExec($startedAt);
    }

    /**
     * Server启动在主进程的主线程时的自定义事件处理
     *
     * @param swoole_server $server
     */
    protected function onStartHandle(\swoole_server $server)
    {

    }

    /**
     * Server正常结束时的自定义事件处理
     *
     * @param swoole_server $server
     */
    protected function onShutdownHandle(\swoole_server $server)
    {

    }

    /**
     * 获取请求路由
     *
     * @param swoole_http_request $request
     */
    protected function getRoute(\swoole_http_request $request)
    {
        return ltrim($request->server['request_uri'], '/');
    }

    /**
     * 获取请求的GET参数
     *
     * @param swoole_http_request $request
     */
    protected function getParams(\swoole_http_request $request)
    {
        return $request->get;
    }

    /**
     * 解析收到的数据
     *
     * @param string $data
     */
    protected function decodeData($data)
    {
        return json_decode($data, true);
    }

    /**
     * Before Exec
     */
    protected function beforeExec()
    {
        $startedAt = microtime(true);
        $this->stdout(date('Y-m-d H:i:s') . "\n", Console::FG_YELLOW);
        return $startedAt;
    }

    /**
     * After Exec
     */
    protected function afterExec($startedAt)
    {
        $duration = number_format(round(microtime(true) - $startedAt, 3), 3);
        $this->stdout("{$duration} s\n\n", Console::FG_YELLOW);
    }

    /**
     * Prints a string to STDOUT.
     */
    protected function stdout($string)
    {
        if (Console::streamSupportsAnsiColors(\STDOUT)) {
            $args = func_get_args();
            array_shift($args);
            $string = Console::ansiFormat($string, $args);
        }
        return Console::stdout($string);
    }
}

<?php
/**
 * @link http://www.u-bo.com
 * @copyright 南京友博网络科技有限公司
 * @license http://www.u-bo.com/license/
 */

namespace console\swoole;

use Yii;
use yii\db\Query;
use yii\helpers\Console;
use yii\helpers\VarDumper;
use apps\sqjc\models\WaterLevel;
use apps\sqjc\models\WaterLevelLog;
use common\models\Bayonet;
use common\models\Device;
use common\models\DeviceCategory;

/**
 * Swoole Server测试类
 *
 * @author wangjian
 * @since 1.0
 */
class Server extends BaseServer
{
    /**
     * @inheritdoc
     */
    public function onConnect($server, $fd, $reactorId)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("**Connection Open**\n", Console::FG_GREEN);
        $this->stdout("fd: ");
        $this->stdout("{$fd}\n", Console::FG_BLUE);

        $this->afterExec($startedAt);
    }

    /**
     * @inheritdoc
     */
    public function onReceive($server, $fd, $reactorId, $data)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("**Received Message**\n", Console::FG_GREEN);

        $this->stdout("fd: ");
        $this->stdout("{$fd}\n", Console::FG_BLUE);

        $this->stdout("data: ");//接收的数据
        $this->stdout("{$data}\n", Console::FG_BLUE);



        $result = $server->send($fd, '回复消息');



        $this->afterExec($startedAt);
    }


    /**
     * @inheritdoc
     */
    public function onRequest($request, $response)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("**HTTP Request**\n", Console::FG_GREEN);

        $this->stdout("fd: ");
        $this->stdout("{$request->fd}\n", Console::FG_BLUE);

        $response->status(200);
        $response->end('success');

        $this->afterExec($startedAt);
    }

    /**
     * @inheritdoc
     */
    public function onClose($server, $fd)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("**Connection Close**\n", Console::FG_GREEN);

        $this->stdout("fd: ");
        $this->stdout("{$fd}\n", Console::FG_BLUE);

        $this->afterExec($startedAt);
    }



    /**
     * @inheritdoc
     */
    public function onTask($server, $taskId, $srcWorkerId, $data)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("New AsyncTask: ");
        $this->stdout("{$taskId}\n", Console::FG_BLUE);
        $this->stdout("{$data}\n", Console::FG_BLUE);

        $server->finish($data);

        $this->afterExec($startedAt);
    }

    /**
     * @inheritdoc
     */
    public function onWorkerStop($server, $worker_id)
    {
        // Yii::$app->db->close();
    }
    /**
     * @inheritdoc
     */
    public function onWorkerStart($server, $worker_id)
    {
        // Yii::$app->db->open();
    }


    /**
     * @inheritdoc
     */
    public function onTaskFinish($server, $taskId, $data)
    {
        $startedAt = $this->beforeExec();

        $this->stdout("AsyncTask finished: ");
        $this->stdout("{$taskId}\n", Console::FG_BLUE);

        $this->afterExec($startedAt);
    }
}

<?php
/**
 * @link http://www.u-bo.com
 * @copyright 南京友博网络科技有限公司
 * @license http://www.u-bo.com/license/
 */

namespace console\controllers;

use Yii;
use yii\helpers\Console;
use yii\helpers\FileHelper;
use yii\helpers\ArrayHelper;
use console\swoole\Server;

/**
 * WebSocket Server controller.
 *
 * @see https://github.com/tystudy/yii2-swoole-websocket/blob/master/README.md
 *
 * @author wangjian
 * @since 1.0
 */
class SwooleController extends Controller
{
    /**
     * @var string 监听IP
     */
    public $host;
    /**
     * @var string 监听端口
     */
    public $port;
    /**
     * @var boolean 是否以守护进程方式启动
     */
    public $daemon = false;
    /**
     * @var boolean 是否启动测试类
     */
    public $test = false;

    /**
     * @var array Swoole参数配置项
     */
    private $_params;

    /**
     * @var array Swoole参数配置项(HTTP协议)
     */
    private $_http_params;
    /**
     * @var array Swoole参数配置项(TCP协议)
     */
    private $_tcp_params;

    /**
     * @inheritdoc
     */
    public function beforeAction($action)
    {
        if (parent::beforeAction($action)) {
            //判断是否开启swoole拓展
            if (!extension_loaded('swoole')) {
                return false;
            }

            //获取swoole配置信息
            if (!isset(Yii::$app->params['swoole'])) {
                return false;
            }
            $this->_params = Yii::$app->params['swoole'];
            $this->_http_params = ArrayHelper::remove($this->_params, 'http');
            $this->_tcp_params = ArrayHelper::remove($this->_params, 'tcp');

            foreach ($this->_params as &$param) {
                if (strncmp($param, '@', 1) === 0) {
                    $param = Yii::getAlias($param);
                }
            }

            $this->_params = ArrayHelper::merge($this->_params, [
                'daemonize' => $this->daemon
            ]);

            return true;
        } else {
            return false;
        }
    }
    /**
     * 启动服务
     */
    public function actionStart()
    {
        if ($this->getPid() !== false) {
            $this->stdout("WebSocket Server is already started!\n", Console::FG_RED);
            return self::EXIT_CODE_NORMAL;
        }
        $server = new Server($this->_http_params, $this->_tcp_params, $this->_params);
        $server->run();
    }

    /**
     * 停止服务
     */
    public function actionStop()
    {
        $pid = $this->getPid();
        if ($pid === false) {
            $this->stdout("Tcp Server is already stoped!\n", Console::FG_RED);
            return self::EXIT_CODE_NORMAL;
        }

        \swoole_process::kill($pid);
    }

    /**
     * 清理日志文件
     */
    public function actionClearLog()
    {
        $logFile = Yii::getAlias($this->_params['log_file']);
        FileHelper::unlink($logFile);
    }

    /**
     * 获取进程PID
     *
     * @return false|integer PID
     */
    private function getPid()
    {
        $pidFile = $this->_params['pid_file'];
        if (!file_exists($pidFile)) {
            return false;
        }

        $pid = file_get_contents($pidFile);
        if (empty($pid)) {
            return false;
        }

        $pid = intval($pid);
        if (\swoole_process::kill($pid, 0)) {
            return $pid;
        } else {
            FileHelper::unlink($pidFile);
            return false;
        }
    }

    /**
     * @inheritdoc
     */
    public function options($actionID)
    {
        return ArrayHelper::merge(parent::options($actionID), [
            'daemon',
            'test'
        ]);
    }

    /**
     * @inheritdoc
     */
    public function optionAliases()
    {
        return ArrayHelper::merge(parent::optionAliases(), [
            'd' => 'daemon',
            't' => 'test',
        ]);
    }
}