基于 Swoole 的微信掃碼登錄功能實(shí)現(xiàn)代碼

隨著微信的普及，掃碼登錄方式越來(lái)越被現(xiàn)在的應(yīng)用所使用。它因?yàn)椴挥萌ビ涀∶艽a，只要有微信號(hào)即可方便快捷登錄。微信的開(kāi)放平臺(tái)原生就有支持掃碼登錄的功能，不過(guò)大部分人還是在用公眾平臺(tái)，所以掃碼登錄只能自行實(shí)現(xiàn)。這里基于微信公眾平臺(tái)的帶參數(shù)臨時(shí)二維碼，并且結(jié)合 Swoole 的 WebSocket 服務(wù)實(shí)現(xiàn)掃碼登錄。大體流程如下：

1. 客戶端打開(kāi)登錄界面，連接到 WebSocket 服務(wù)
2. WebScoket 服務(wù)生成帶參數(shù)二維碼返回給客戶端
3. 用戶掃描展示的帶參數(shù)二維碼
4. 微信服務(wù)器回調(diào)掃碼事件并通知開(kāi)發(fā)者服務(wù)器
5. 開(kāi)發(fā)者服務(wù)器通知 WebSocket 服務(wù)
6. WebSocket 服務(wù)通知客戶端登錄成功

連接 WebSocket 服務(wù)

安裝完 Swoole 之后，我們需用使用到 WebSocket 服務(wù)。新建一個(gè) WebSocket 服務(wù)非常簡(jiǎn)單：

$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 回調(diào)其實(shí)用不到，因?yàn)槎际欠?wù)端下發(fā)消息的，但是必須設(shè)定一個(gè)。設(shè)定的端口號(hào)如果低于 1024 則必須要有 root 權(quán)限，服務(wù)器記得去防火墻開(kāi)啟該端口。

生成帶參數(shù)二維碼

WebSocket 服務(wù)在客戶端連接成功后，需要生成一張微信的帶參數(shù)二維碼返回給客戶端展示：

$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
  ]));
});

我們?cè)?open 回調(diào)中，生成一張臨時(shí)的二維碼，二維碼的場(chǎng)景值就是客戶端連接的文件描述符，這樣就可以保證每個(gè)客戶端的唯一性.有效時(shí)間設(shè)置為 120 秒，防止一張二維碼被多次掃碼使用。消息 push 到客戶端的時(shí)候必須要 json，方便客戶端處理?？蛻舳舜a也很簡(jiǎn)單：

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

回調(diào)掃碼事件

在客戶端展示二維碼后，需要提示用戶掃碼。對(duì)于用戶掃臨時(shí)的二維碼，微信會(huì)觸發(fā)相應(yīng)的回調(diào)事件，我們需要在該回調(diào)事件中處理用戶的掃碼行為。其中我們需要用到微信傳遞過(guò)來(lái)的一些參數(shù)：

FromUserName  發(fā)送方帳號(hào)（一個(gè)OpenID）
MsgType      消息類型，event
Event      事件類型，subscribe
EventKey    事件 KEY 值，qrscene_為前綴，后面為二維碼的參數(shù)值

這里要注意一點(diǎn)：微信已關(guān)注掃碼推送的 EventKey 是沒(méi)有 qrscene_ 前綴的，只有未關(guān)注掃碼然后關(guān)注才有。

收到微信回調(diào)后我們首先要根據(jù)不同的事件類型做不同處理：

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

這里只講解一個(gè)關(guān)注事件的業(yè)務(wù)邏輯，其他根據(jù)需要自行編碼：

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次登錄，玩的開(kāi)心！";
  return sprintf($msgTemp, $user['nickname'], $count);
}

這里的 EventKey 實(shí)際上就是連接 WebSocket 的客戶端文件描述符，獲取到掃碼用戶的 OPEN_ID ,根據(jù)用戶的 OPEN_ID 獲取用戶信息，通知 WebSocket 服務(wù)，響應(yīng)文本消息給微信。

這里一個(gè)比較麻煩的點(diǎn)就是如何通知 WebSocket 服務(wù)，我們知道處理微信回調(diào)的代碼是是不在 WebSocket 服務(wù)上的，那么不同 Server 間如何通信呢？Swoole 官方給出的解決方案有兩個(gè)：

1. 額外監(jiān)聽(tīng)一個(gè)UDP端口
2. 使用 swoole_client 作為客戶端訪問(wèn) Server
   

這里我們選擇第二個(gè)方案，Swoole 1.8 版本支持一個(gè) Server 監(jiān)聽(tīng)多個(gè)端口，我們?cè)?WebSocket 服務(wù)新增監(jiān)聽(tīng)一個(gè) 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) {
  
});

主服務(wù)器是 WebSocket 或 Http 協(xié)議，新監(jiān)聽(tīng)的 TCP 端口默認(rèn)會(huì)繼承主 Server 的協(xié)議設(shè)置，必須單獨(dú)調(diào)用 set 方法設(shè)置新的協(xié)議才會(huì)啟用新協(xié)議

然后我們就可以在掃碼回調(diào)的進(jìn)程中去通知 WebSocket 服務(wù)：

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 服務(wù)收到登錄成功的通知后，就可以根據(jù)需要處理一下用戶信息，然后把用戶信息傳遞給客戶端的瀏覽器展示結(jié)果，還記得我們剛剛新監(jiān)聽(tīng)的 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);
});

最后登錄的界面：

總結(jié)

整個(gè)過(guò)程并不難，主要的兩個(gè)難點(diǎn)就是對(duì)應(yīng)連接用戶的掃碼用戶、不同 Server 之間的通信，我們的解決辦法就是把連接的文件描述符作為臨時(shí)二維碼場(chǎng)景值(這里也可以采用 Redis 來(lái)存儲(chǔ)映射關(guān)系)、監(jiān)聽(tīng)新的 TCP 端口來(lái)接受通知消息?？梢栽L問(wèn) http://wechat.sunnyshift.com/index.php 試試看，記得要用電腦打開(kāi)。

以上所述是小編給大家介紹的基于 Swoole 的微信掃碼登錄功能實(shí)現(xiàn)代碼，希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持！