這是一個建立於 的文章,其中的資訊可能已經有所發展或是發生改變。
HTML部分
在網頁上放置一個按鈕和一個文字框,按鈕用來向伺服器發送資訊,文字框用來顯示從伺服器傳回的訊息。
<!DOCTYPE html><html lang="en" xmlns="http://www.w3.org/1999/xhtml"><head> <meta charset="utf-8" /> <title></title> <script type="text/javascript" src="js/loli.js"></script></head><body> <p><input id="btn1" type="button" value="按鈕" /></p> <p><input id="txt1" type="text" /></p></body></html>
WebSocket部分
window.onload = main;var ws;function main() { var oBtn = document.getElementById('btn1'); oBtn.onclick = OnButton1;}function OnButton1() { ws = new WebSocket('ws://localhost/ws'); ws.onopen = OnOpen; ws.onmessage = OnMessage;}function OnOpen(event) { ws.send('hello websocket');}function OnMessage(event) { var oTxt = document.getElementById('txt1'); oTxt.value = event.data;};
後端部分
要想處理WebSocket請求,必須有一個
func (*websocket.Conn)
類型的處理函數,例如:
func OnWebSocket(ws *websocket.Conn) {defer ws.Close()var err errorvar str stringfor {if err = websocket.Message.Receive(ws, &str); err != nil {break} else {fmt.Println("從用戶端收到:", str)}str = "hello, I'm server."if err = websocket.Message.Send(ws, str); err != nil {break} else {fmt.Println("向用戶端發送:", str)}}}
然後和一個網址對應起來:
http.Handle("/ws", websocket.Handler(OnWebSocket))
完整代碼如下:
package mainimport ("bytes""errors""fmt""html/template""net/http""runtime""golang.org/x/net/websocket")// 連接埠const (HTTP_PORT string = "80"HTTPS_PORT string = "443")// 目錄const (CSS_CLIENT_PATH = "/css/"DART_CLIENT_PATH = "/js/"IMAGE_CLIENT_PATH = "/image/"CSS_SVR_PATH = "web"DART_SVR_PATH = "web"IMAGE_SVR_PATH = "web")func init() {runtime.GOMAXPROCS(runtime.NumCPU())}func main() {// 先把css和指令碼服務上去http.Handle(CSS_CLIENT_PATH, http.FileServer(http.Dir(CSS_SVR_PATH)))http.Handle(DART_CLIENT_PATH, http.FileServer(http.Dir(DART_SVR_PATH)))// 網址與處理邏輯對應起來http.HandleFunc("/", HomePage)http.Handle("/ws", websocket.Handler(OnWebSocket))// 開始服務err := http.ListenAndServe(":"+HTTP_PORT, nil)if err != nil {fmt.Println("服務失敗 /// ", err)}}func WriteTemplateToHttpResponse(res http.ResponseWriter, t *template.Template) error {if t == nil || res == nil {return errors.New("WriteTemplateToHttpResponse: t must not be nil.")}var buf bytes.Buffererr := t.Execute(&buf, nil)if err != nil {return err}res.Header().Set("Content-Type", "text/html; charset=utf-8")_, err = res.Write(buf.Bytes())return err}func HomePage(res http.ResponseWriter, req *http.Request) {t, err := template.ParseFiles("web/loli.html")if err != nil {fmt.Println(err)return}err = WriteTemplateToHttpResponse(res, t)if err != nil {fmt.Println(err)return}}func OnWebSocket(ws *websocket.Conn) {defer ws.Close()var err errorvar str stringfor {if err = websocket.Message.Receive(ws, &str); err != nil {break} else {fmt.Println("從用戶端收到:", str)}str = "hello, I'm server."if err = websocket.Message.Send(ws, str); err != nil {break} else {fmt.Println("向用戶端發送:", str)}}}