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

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

本文采用restgo 框架,相关信息请查阅

https://gitbook.cn/m/mazi/activity/5c77c109a1fc3527823d995d

思路关键流程

建立场景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

github.com/gorilla/websocket

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) }

小程序处理关键点

采用好的框架

H5小程序androidios

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) } },

核心代码如下

gotUserInfouni.login

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