go实现websocket案例

缺乏、安全感 2022-10-28 14:25 195阅读 0赞

## 一、net/websocket ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNDQ3MzAx_size_16_color_FFFFFF_t_70]

package main
    
    import (
    	"bytes"
    	"fmt"
    	"log"
    	"net/http"
    	"unsafe"
    
    	"golang.org/x/net/websocket"
    )
    
    func ByteToString(p *byte) string { 
    	bufs := bytes.NewBufferString("")
    	for ps := p; *ps != byte(0); ps = (*byte)(unsafe.Pointer(1 + (uintptr)(unsafe.Pointer(ps)))) { 
    		bufs.WriteByte(*ps)
    	}
    	return bufs.String()
    }
    
    func main() { 
    	fmt.Println("websocket地址: ws://127.0.0.1:8080/runtime")
    	http.Handle("/runtime", websocket.Handler(echo))
    	if err := http.ListenAndServe(":8080", nil); err != nil { 
    		log.Fatal(err)
    	}
    }
    
    //测试地址： http://coolaf.com/tool/chattest 首字母大写是公有的 Echo，首字母小写是私有的 echo
    func echo(w *websocket.Conn) { 
    	var error error
    	for { 
    		//只支持string类型
    		var reply string
    		if error = websocket.Message.Receive(w, &reply); error != nil { 
    			log.Println("websocket出现异常", error)
    			break
    		}
    		fmt.Println("收到客户端消息:" + reply)
    		msg := reply + ", 我是服务端"
    		fmt.Println("发送客户端消息:" + msg)
    		if error = websocket.Message.Send(w, msg); error != nil { 
    			log.Println("websocket出现异常", error)
    			break
    		}
    	}
    }

## 二、gobwas/ws ##

package main
    
    import (
    	"net/http"
    
    	"github.com/gobwas/ws"
    	"github.com/gobwas/ws/wsutil"
    )
    
    func main() { 
    	http.ListenAndServe(":8080", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { 
    		conn, _, _, err := ws.UpgradeHTTP(r, w)
    		if err != nil { 
    			// handle error
    		}
    		go func() { 
    			defer conn.Close()
    			for { 
    				msg, op, err := wsutil.ReadClientData(conn)
    				if err != nil { 
    					// handle error
    				}
    				err = wsutil.WriteServerMessage(conn, op, msg)
    				if err != nil { 
    					// handle error
    				}
    			}
    		}()
    	}))
    }

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNDQ3MzAx_size_16_color_FFFFFF_t_70 1]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNDQ3MzAx_size_16_color_FFFFFF_t_70]: /images/20221024/11aa3f830cc8497fab040d15ae13e719.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNDQ3MzAx_size_16_color_FFFFFF_t_70 1]: /images/20221024/17969a8ed3c64cb1a5576594303fc493.png