掃小程序碼實現網站登錄,提供源代碼

使用掃小程序碼登錄網站
網絡上關於實現本本功能的文章不少，可是給出案列的幾乎沒有，今天筆者實現用小程序碼實現網站登錄，體驗地址以下
https://idea.techidea8.com/open/login.shtml

思路

關鍵流程

創建場景sceneid和websocket的綁定關係

得到sceneid

場景ID能夠前端生成，也能夠後端生成，只須要保證sceneid的同一時間惟一性便可。
前端生成能夠採用隨機數加時間戳的形式,也能夠用uuid算法

//時間戳
var sceneid ="scend-" +  new Data().getTime() + Math.ceil(Math.random()*888888+1000000);

創建websocket

var ws = new WebSocket("ws://192.168.0.106/websocket?clientid="+sceneid )
ws.onopen=function(env){
    console.log(env)
} 
ws.onmessage=function(env){
    var data = env.data;
    //這個data 就是後端發來的用戶數據
}

後端創建websocket

後端採用go語言github.com/gorilla/websocket包創建websocket.由於golang 很是適合高併發場景。

func (ctrl *PushCtrl) websocket(w http.ResponseWriter, req *http.Request) {

    //fmt.Printf("%+v",request.Header)
    //todo 檢驗接入是否合法
    //checkToken(userId int64,token string)
    query := req.URL.Query()
    clientid := query.Get("clientid")

    conn, err := (&websocket.Upgrader{
        CheckOrigin: func(r *http.Request) bool {
            return true
        },
    }).Upgrade(w, req, nil)

    if err != nil {
        log.Println(err.Error())
        return
    }

    clientMap.Store(clientid, conn)
    go func(clientId string, conn *websocket.Conn) {
        //處理出錯信息
        defer func() {
            conn.Close()
            clientMap.Delete(clientid)
        }()
        for {
            _, _, err := conn.ReadMessage()
            if err != nil {
                log.Println(err.Error())
                return
            }
        }
    }(clientid, conn)
}

咱們採用sync.map創建sceneid和websocket的對應關係

clientMap.Save(sceneid,conn)

對於由於異常斷開的con咱們須要移除

clientMap.Delete(sceneid)

得到小程序二維碼

得到小程序的accesstoken

小程序access 請求接口以下
``
該接口日使用頻率有限制，所以咱們須要將這些數據緩存起來,緩存方案不少，有redis,也有內存，咱們這裏直接用一個變量便可存儲

accesstoken :=""
func GetAccessToken() string{
    return accesstoken 
}
func RefreshAccessToken()string{
    url = ""
    resp := httpget(url)
    //resp 是一個包含accesstoken的json字符串,咱們解析這個json便可
    accesstoken = decodeaccesstokenfromjson(resp)
    return accesstoken 
}

咱們還須要一個滴答計數器用來刷新accesstoken,accesstoken 的有效期是7200秒，咱們4000秒刷新一次accesstoken 便可

func refreshAccessToken(){
    ticker := time.NewTicker(time.Second *4000)
    for{
        select {
            case <-ticker.C:
                RefreshAccessToken()
        }
    }
}

在init.go 中的init方法中啓動協程

go refreshAccessToken()

小程序二維碼編程技巧

小程序二維碼請求接口以下
https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token=ACCESSTOKEN

因爲該連接會返回倆種結果

錯誤返回Json

{
"errcode":400001,
"errmsg":"什麼什麼緣由"
}

正確返回j圖片buffer

這是一個二進制碼流，所以咱們須要對返回結果進行標準化處理
咱們建議一種標準化結果

{
code:0,
data:"",
msg:"結果說明"
}

參數	說明
code	標識成功或者失敗,200爲成功,400爲失敗
data	圖片的base64格式編碼
msg	結果說明,或者出錯提示

代碼示例以下

url := fmt.Sprintf("https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token=%s", token)
ret, err := util.PostJSON(url, arg)
if err != nil {
        util.FailMsg(w, err.Error())
        return
}

jsonstr := string(ret)
if strings.Contains(jsonstr, "errmsg") {
    util.FailMsg(w, jsonstr)
    return
} else {
    base64data := base64.StdEncoding.EncodeToString(ret)
    util.RespOk(w, "data:image/png;base64,"+base64data)
}

其中util是筆者封裝的經常使用工具包

前端請求成功後得到data,能夠利用 $("#qrcode").attr("src",res.data)刷新圖片二維碼

function refreshqrcode(){
    clientId = "scene-"+new Date().getTime();
    var api = restgo.buildapi("miniapp/getwxacodeunlimit")
    restgo.post(api,{"scene":clientId}).then(res=>{
        if(res.code==200){
            $("#qrcode").attr("src",res.data)
        }else{
            alert(res.msg)    
        }
        intiwebsocket()
    },res=>{
        alert(res.msg)
    })
}

小程序處理關鍵點

採用好的框架

咱們編程過程才用了uniapp，這個框架不錯,徹底是vue的語法，
一套代碼能夠生成H5/小程序/android/ios
咱們採用了vue的watch特性,經過監聽userid來肯定是否發送信息

watch:{
     userid:function(a,b){
            
            if(a==0){
                return 
            }
            
            this.loaddata()
            //若是clientid是空的說明不是掃碼進來的    
            if(!this.clientid){
                return 
            }
//若是是掃碼進來的,那麼咱們還須要推送消息到服務器後端
server.PublishMsg(this.clientid,this.userid,this.role,this.avatarUrl,this.nickName).then(res=>{
                tip.error(res.msg)
            },res=>{
                tip.error(res.msg)
            })
            
     }
  }

獲取sceneid

咱們經過onload方法得到sceneid

onLoad(arg) {
        if(!!arg && !!arg.scene){
                    this.clientid = decodeURIComponent(arg.scene)
        }
                
    },

核心代碼以下

經過gotUserInfo得到用戶頭像、暱稱等,經過uni.login得到code

gotUserInfo :function(e) {
             //得到用戶頭像、暱稱、
              uni.login(
              {
                  success:(res) =>{
                            //經過wx.login得到code
                              userInfo.code = res.code
                                                //統一傳遞到後端      
                        this.authwithcode(userInfo)
                  }
              }
              )
            }
          },
        authwithcode:function(userInfo){
                        //code得到openid在後端作,作了後繼續去查User表
            server.AuthWithCode(userInfo).then(res=>{
                    //若是這個openid綁定了用戶                                                                    
                    if(res.data.id>0){
                        //這裏res.data 就是user對象,能夠繼續操做
                        return ;
                    }
                    //若是沒有那麼註冊
                    server.RegisterWithOpenId(res.data.mini_openid,userInfo.avatarUrl,userInfo.nickName)#.then(res=>{
                        //註冊成功則返回
                        //這裏res.data 就是user對象能夠繼續操做
                        
                    },res=>{
                        tip.error(res.msg)
                    })
                },res=>{
                    tip.error(res.msg||"");
                })
            }
        },

體驗地址

本文全部應用體驗地址以下
互聯網行業解決方案吧https://idea.techidea8.com/open/idea.shtml?id=5

關於代碼

代碼得到地址

小程序碼登錄

代碼配置

請認真閱讀readme.md