基於 Swoole 的微信掃碼登錄功能實現代碼

隨着微信的普及,掃碼登錄方式越來越被現在的應用所使用。它因爲不用去記住密碼,只要有微信號即可方便快捷登錄。微信的開放平臺原生就有支持掃碼登錄的功能,不過大部分人還是在用公衆平臺,所以掃碼登錄只能自行實現。這裏基於微信公衆平臺的帶參數臨時二維碼,並且結合 Swoole 的 WebSocket 服務實現掃碼登錄。大體流程如下:

客戶端打開登錄界面,連接到 WebSocket 服務
WebScoket 服務生成帶參數二維碼返回給客戶端
用戶掃描展示的帶參數二維碼
微信服務器回調掃碼事件並通知開發者服務器
開發者服務器通知 WebSocket 服務
WebSocket 服務通知客戶端登錄成功

連接 WebSocket 服務

安裝完 Swoole 之後,我們需用使用到 WebSocket 服務。新建一個 WebSocket 服務非常簡單:
1
$server = new swoole_websocket_server("0.0.0.0", 1099);
2
$server->on('open', function (swoole_websocket_server $server, $request) use ($config){
3
echo "server: handshake success with fd{$request->fd}\n";
4
});
5
$server->on('message', function (swoole_websocket_server $server, $frame) {
6
});

這裏的 message 回調其實用不到,因爲都是服務端下發消息的,但是必須設定一個。設定的端口號如果低於 1024 則必須要有 root 權限,服務器記得去防火牆開啓該端口。

生成帶參數二維碼

WebSocket 服務在客戶端連接成功後,需要生成一張微信的帶參數二維碼返回給客戶端展示:
1
$server->on('open', function (swoole_websocket_server $server, $request) use ($config){
2
$app = Factory::officialAccount($config['wechat']);
3
$result = $app->qrcode->temporary($request->fd, 120);
4
$url = $app->qrcode->url($result['ticket']);
5
$server->push($request->fd, json_encode([
6
'message_type' => 'qrcode_url',
7
'url' => $url
8
]));
9
});

我們在 open 回調中,生成一張臨時的二維碼,二維碼的場景值就是客戶端連接的文件描述符,這樣就可以保證每個客戶端的唯一性.有效時間設置爲 120 秒,防止一張二維碼被多次掃碼使用。消息 push 到客戶端的時候必須要 json,方便客戶端處理。客戶端代碼也很簡單:
1
const socket = new WebSocket('ws://127.0.0.1:1099');
2
socket.addEventListener('message', function (event) {
3
var data = JSON.parse(event.data);
4
if (data.message_type == 'qrcode_url'){
5
$('#qrcode').attr('src', data.url);
6
}
7
});

回調掃碼事件

在客戶端展示二維碼後,需要提示用戶掃碼。對於用戶掃臨時的二維碼,微信會觸發相應的回調事件,我們需要在該回調事件中處理用戶的掃碼行爲。其中我們需要用到微信傳遞過來的一些參數:
1
FromUserName 發送方帳號(一個OpenID)
2
MsgType 消息類型,event
3
Event 事件類型,subscribe
4
EventKey 事件 KEY 值,qrscene_爲前綴,後面爲二維碼的參數值

這裏要注意一點:微信已關注掃碼推送的 EventKey 是沒有 qrscene_ 前綴的,只有未關注掃碼然後關注纔有。

收到微信回調後我們首先要根據不同的事件類型做不同處理:
01
if ($message['MsgType'] == 'event'){
02
if ($message['Event'] == 'subscribe'){ //關注
03
return $this->subscribe($message);
04
}
05
if ($message['Event'] == 'unsubscribe') { //取消關注
06
return $this->unsubscribe($message);
07
}
08
if ($message['Event'] == 'SCAN'){ //已關注掃碼
09
return $this->scan($message);
10
}
11
}else{
12
return "您好!歡迎使用 SwooleWechat 掃描登錄";
13
}

這裏只講解一個關注事件的業務邏輯,其他根據需要自行編碼:
01
public function subscribe($message){
02
$eventKey = intval(strreplace('qrscene', '', $message['EventKey']));
03
$openId = $message['FromUserName'];
04
$user = $this->app->user->get($openId);
05
$this->notify(json_encode([
06
'type' => 'scan',
07
'fd' => $eventKey,
08
'nickname' => $user['nickname']
09
]));
10
$count = $this->count($openId);
11
$msgTemp = "%s,登錄成功!\n這是你第%s次登錄,玩的開心!";
12
return sprintf($msgTemp, $user['nickname'], $count);
13
}

這裏的 EventKey 實際上就是連接 WebSocket 的客戶端文件描述符,獲取到掃碼用戶的 OPEN_ID ,根據用戶的 OPEN_ID 獲取用戶信息,通知 WebSocket 服務,響應文本消息給微信。

這裏一個比較麻煩的點就是如何通知 WebSocket 服務,我們知道處理微信回調的代碼是是不在 WebSocket 服務上的,那麼不同 Server 間如何通信呢?Swoole 官方給出的解決方案有兩個:

額外監聽一個UDP端口
使用 swoole_client 作爲客戶端訪問 Server

這裏我們選擇第二個方案,Swoole 1.8 版本支持一個 Server 監聽多個端口,我們在 WebSocket 服務新增監聽一個 TCP 的端口:
1
$tcp_server = $server->addListener('0.0.0.0', 9999, SWOOLE_SOCK_TCP);
2
$tcp_server->set([]);
3
$tcp_server->on('receive', function ($serv, $fd, $threadId, $data) {
4

5
});

主服務器是 WebSocket 或 Http 協議,新監聽的 TCP 端口默認會繼承主 Server 的協議設置,必須單獨調用 set 方法設置新的協議纔會啓用新協議

然後我們就可以在掃碼回調的進程中去通知 WebSocket 服務:
1
public function notify($message){
2
$client = new swoole_client(SWOOLE_SOCK_TCP);
3
if (!$client->connect('127.0.0.1', $this->config['notify_port'], -1)) {
4
return "connect failed. Error: {$client->errCode}\n";
5
}
6
$ret = $client->send($message);
7
}

通知登錄成功

在 WebSocket 服務收到登錄成功的通知後,就可以根據需要處理一下用戶信息,然後把用戶信息傳遞給客戶端的瀏覽器展示結果,還記得我們剛剛新監聽的 TCP 端口嗎?就可以在 receive 事件中處理:
01
$tcp_server->on('receive', function ($serv, $fd, $threadId, $data) {
02
$data = json_decode($data, true);
03
if ($data['type'] == 'scan'){
04
$serv->push($data['fd'], json_encode([
05
'message_type' => 'scan_success',
06
'user' => $data['nickname']
07
]));
08
}
09
$serv->close($fd);
10
});

最後登錄的界面:

整個過程並不難,主要的兩個難點就是對應連接用戶的掃碼用戶、不同 Server 之間的通信,我們的解決辦法就是把連接的文件描述符作爲臨時二維碼場景值(這裏也可以採用 Redis 來存儲映射關係)、監聽新的 TCP 端口來接受通知消息。可以訪問 http://www.niuzhan.com/ 試試看,記得要用電腦打開。

本文轉載於http://news.mkq.online/

以上所述是小編給大家介紹的基於 Swoole 的微信掃碼登錄功能實現代碼,希望對大家有所幫助

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章