扫小程序码实现网站登陆,提供源代码

使用扫小程序码登陆网站
网络上关于实现本本功能的文章很多，但是给出案列的几乎没有，今天笔者实现用小程序码实现网站登陆，体验地址如下

思路

关键流程

建立场景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||"");
                })
            }
        },  

体验地址

本文所有应用体验地址如下

关于代码

代码获得地址

代码配置

请认真阅读readme.md

推荐阅读