首頁 > 文章 > 後端開發 > Swoole 實作微信掃碼登入功能

Swoole 實作微信掃碼登入功能

小云云原創: 2018-01-16 13:43:152186瀏覽

掃碼登入方式越來越被現在的應用程式所使用。它因為不用去記住密碼，只要有微信號即可方便快捷登錄.這裡基於微信公眾平台的帶參數臨時二維碼，並且結合 Swoole 的 WebSocket 服務實現掃碼登錄。本文主要和大家分享Swoole 實作微信掃碼登入功能，希望能幫助大家。

隨著微信的普及，掃碼登入方式越來越被現在的應用程式所使用。它因為不用去記住密碼，只要有微訊號即可方便快速登入。微信的開放平台原生就有支援掃碼登入的功能，不過大部分人還是在用公眾平台，所以掃碼登入只能自行實現。這裡基於微信公眾平台的參數臨時二維碼，並且結合 Swoole 的 WebSocket 服務實作掃碼登入。大體流程如下：

客戶端開啟登入介面，連線到WebSocket 服務
WebScoket 服務產生參數二維碼回傳給客戶端
使用者掃描展示的帶參數二維碼
#微信伺服器回調掃碼事件並通知開發者伺服器
開發者伺服器通知WebSocket 服務
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 = JSON.parse(event.data);
    if (data.message_type == 'qrcode_url'){
      $('#qrcode').attr('src', data.url);
    }
  });

回呼掃碼事件

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

FromUserName  发送方帐号（一个OpenID）
MsgType      消息类型，event
Event      事件类型，subscribe
EventKey    事件 KEY 值，qrscene_为前缀，后面为二维码的参数值

這裡要注意一點：微信已關注掃碼推送的 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 官方給的解決方案有兩個：

額外監聽一個UDP埠
使用swoole_client 作為客戶端存取Server

這裡我們選擇第二個方案，Swoole 1.8 版本支援一個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);
});

最後登入的介面：

##總結

##都並不難，主要的兩個難點就是對應連接用戶的掃碼用戶、不同Server 之間的通信，我們的解決辦法就是把連接的文件描述符作為臨時二維碼場景值(這裡也可以採用Redis 來存儲映射關係)、監聽新的TCP 連接埠來接受通知訊息。可以上 http://wechat.sunnyshift.com/index.php 試試看，記得要用電腦開啟。

Swoole 實作微信掃碼登入功能

相關文章