小艾的自留地

Stay foolish, Stay hungry

最近一直在学习Swoole,刚好有个老项目的一小部分(一个脚本)有用到了Tcp 协议,借此机会重构一下。

场景描述:

该脚本的作用用一句话就可以概述:将本地数据源推送给另外一台服务器。

原始的处理方式,不合理的地方有以下几点:

  1. 目标服务器需要开放指定端口,这会导致目标服务器向外暴露,不安全。
  2. 如果有多台目标服务器,这会导致频繁需要修改源码,脚本维护起来不方便。

重构

重构需要解决的问题有如下:

  1. 当客户端连接成功后,才会向该客户端推送数据。
  2. 当客户端断开连接时,停止向该客户端推送数据。
  3. 允许多个客户端同时连接。
  4. 因为数据源是不间断的,理论上只要客户端的连接不主动断开,服务端的数据推送就不会主动停止。

最终使用Swoole 的Tcp + Process 实现了以上需求,核心代码如下:

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
<?php
use Swoole\Process;

/**
* 创建Server 对象,监听本地 9501 端口。
*/
$server = new Swoole\Server("0.0.0.0", 9501);

$workers = [];

/**
* 监听连接进入事件
*/
$server->on("Connect", function ($server, $fd) {
global $workers;

// 创建子进程
$process = new swoole_process(function (swoole_process $worker) use ($server, $fd) {
echo "Client Connect" . PHP_EOL;

// todo 业务逻辑
...

// 向客户端推送消息
$server->send($fd, $str);

}, true, 0, false);

// 启动子进程
$pid = $process->start();

array_push($workers, ["pid" => $pid, "fd" => $fd]);
});

/**
* 监听数据接收事件
*/
$server->on("Receive", function ($server, $fd, $from_id, $data){
$server->send($fd, "Server: " . $data);
});

/**
* 监听连接关闭事件
*/
$server->on("Close", function ($server, $fd) {
global $workers;

foreach ($workers as $worker) {
if ($worker['fd'] === $fd){
// 检查子进程是否存在
if (Process::kill($worker['pid'], 0)){
array_shift($worker);
// 通过信号终止子进程
Process::kill($worker['pid'], SIGKILL);
}
}
}
echo "Client Close" . PHP_EOL;
});

// 启动TCP 服务器
$server->start();

其实实现的原理很简单,利用Swoole 的基于事件的 Tcp 异步编程,当有客户端连接时,就创建一个子进程进行推送数据,但客户端连接断开时,就通过信号结束该客户端对应的子进程。

评论