Swoole 的微信掃碼登錄

微信應用的便捷，掃碼登錄方式越來越被現在的應用所使用，它因為不用去記住密碼，只要有微信號即可方便快捷登錄，微信的開放平臺原生就有支持掃碼登錄的功能，不過大部分人還是在用公眾平臺，所以掃碼登錄只能自行實作，這里基于微信公眾平臺的帶引數臨時二維碼，并且結合 Swoole 的 WebSocket 服務實作掃碼登錄，大體流程如下：

1. 客戶端打開登錄界面，連接到 WebSocket 服務
2. WebScoket 服務生成帶引數二維碼回傳給客戶端
3. 用戶掃描展示的帶引數二維碼
4. 微信服務器回呼掃碼事件并通知開發者服務器
5. 開發者服務器通知 WebSocket 服務
6. WebSocket 服務通知客戶端登錄成功

連接 WebSocket 服務

安裝完 Swoole 之后，我們需用使用到 WebSocket 服務，新建一個 WebSocket 服務非常簡單：

$server = new swoole_websocket_server("0.0.0.0", 1099);
$server->on('open', function (swoole_websocket_server $server, $request) use ($config){
    echo "server: handshake success with fd{$request->fd}\n";
});
 
$server->on('message', function (swoole_websocket_server $server, $frame) {
 
});
 

這里的 message 回呼其實用不到，因為都是服務端下發訊息的，但是必須設定一個，設定的埠號如果低于 1024 則必須要有 root 權限，服務器記得去防火墻開啟該埠，

生成帶引數二維碼

WebSocket 服務在客戶端連接成功后，需要生成一張微信的帶引數二維碼回傳給客戶端展示：

$server->on('open', function (swoole_websocket_server $server, $request) use ($config){
    $app = Factory::officialAccount($config['wechat']);
    $result = $app->qrcode->temporary($request->fd, 120);
    $url = $app->qrcode->url($result['ticket']);
    $server->push($request->fd, json_encode([
        'message_type'    =>  'qrcode_url',
        'url'       =>  $url
    ]));
});

我們在 open 回呼中，生成一張臨時的二維碼，二維碼的場景值就是客戶端連接的檔案描述符，這樣就可以保證每個客戶端的唯一性.有效時間設定為 120 秒，防止一張二維碼被多次掃碼使用，訊息 push 到客戶端的時候必須要 json，方便客戶端處理，客戶端代碼也很簡單：

const socket = new WebSocket('ws://127.0.0.1:1099');
    socket.addEventListener('message', function (event) {
        var data = https://www.cnblogs.com/wadhf/p/JSON.parse(event.data);
        if (data.message_type == 'qrcode_url'){
            $('#qrcode').attr('src', data.url);
        }
    });

回呼掃碼事件

在客戶端展示二維碼后，需要提示用戶掃碼，對于用戶掃臨時的二維碼，微信會觸發相應的回呼事件，我們需要在該回呼事件中處理用戶的掃碼行為，其中我們需要用到微信傳遞過來的一些引數：

這里要注意一點：微信已關注掃碼推送的 EventKey 是沒有 qrscene_ 前綴的，只有未關注掃碼然后關注才有，

收到微信回呼后我們首先要根據不同的事件型別做不同處理：

if ($message['MsgType'] == 'event'){
    if ($message['Event'] == 'subscribe'){  //關注
        return $this->subscribe($message);
    }
    if ($message['Event'] == 'unsubscribe') {  //取消關注
        return $this->unsubscribe($message);
    }
    if ($message['Event'] == 'SCAN'){   //已關注掃碼
        return $this->scan($message);
    }
}else{
    return "您好！歡迎使用 SwooleWechat 掃描登錄";
}
 

這里只講解一個關注事件的業務邏輯，其他根據需要自行編碼：

public function subscribe($message){
    $eventKey = intval(str_replace('qrscene_', '', $message['EventKey']));
    $openId = $message['FromUserName'];
    $user = $this->app->user->get($openId);
    $this->notify(json_encode([
        'type'  =>  'scan',
        'fd'    =>  $eventKey,
        'nickname'  =>  $user['nickname']
    ]));
    $count = $this->count($openId);
    $msgTemp = "%s，登錄成功！\n這是你第%s次登錄，玩的開心！";
    return sprintf($msgTemp, $user['nickname'], $count);
}

這里的 EventKey實際上就是連接 WebSocket 的客戶端檔案描述符，獲取到掃碼用戶的 OPEN_ID,根據用戶的OPEN_ID獲取用戶資訊，通知 WebSocket 服務，回應文本訊息給微信，這里一個比較麻煩的點就是如何通知 WebSocket 服務，我們知道處理微信回呼的代碼是是不在 WebSocket 服務上的，那么不同 Server 間如何通信呢？Swoole 官方給出的解決方案有兩個：

1. 額外監聽一個UDP埠
2. 使用 swoole_client 作為客戶端訪問 Server

這里我們選擇第二個方案，讓 Server 監聽多個埠，我們在 WebSocket 服務新增監聽一個 TCP 的埠：

$tcp_server = $server->addListener('0.0.0.0', 9999, SWOOLE_SOCK_TCP);
$tcp_server->set([]);
$tcp_server->on('receive', function ($serv, $fd, $threadId, $data) {
    
});
 

主服務器是 WebSocket 或 Http 協議，新監聽的 TCP 埠默認會繼承主 Server 的協議設定，必須單獨呼叫 set 方法設定新的協議才會啟用新協議

然后我們就可以在掃碼回呼的行程中去通知 WebSocket 服務：

public function notify($message){
    $client = new swoole_client(SWOOLE_SOCK_TCP);
    if (!$client->connect('127.0.0.1', $this->config['notify_port'], -1)) {
        return "connect failed. Error: {$client->errCode}\n";
    }
    $ret = $client->send($message);
}

通知登錄成功

在 WebSocket 服務收到登錄成功的通知后，就可以根據需要處理一下用戶資訊，然后把用戶資訊傳遞給客戶端的瀏覽器展示結果，還記得我們剛剛新監聽的 TCP 埠嗎？就可以在 receive 事件中處理：

$tcp_server->on('receive', function ($serv, $fd, $threadId, $data) {
    $data = json_decode($data, true);
    if ($data['type'] == 'scan'){
        $serv->push($data['fd'], json_encode([
            'message_type'    =>  'scan_success',
            'user'  =>  $data['nickname']
        ]));
    }
    $serv->close($fd);
});