使用golang实现websocket编程

第三方库

github.com/gorilla/websocket

Server

package main

import (
	"fmt"
	"net/http"
	"time"

	"github.com/gorilla/websocket"
)

func chat(w http.ResponseWriter, r *http.Request) {
	conn, err := (&websocket.Upgrader{}).Upgrade(w, r, nil)
	if err != nil {
		http.NotFound(w, r)
		fmt.Printf("conn creat err:%s\n", err.Error())
		return
	}
	defer conn.Close()
	for {
		msgtype, msgdata, err := conn.ReadMessage()
		if err != nil {
			fmt.Printf("conn read err:%s\n", err.Error())
			return
		}
		fmt.Printf("recv: %s\n", string(msgdata))
		conn.WriteMessage(msgtype, []byte("hello, client"))
		fmt.Println("send: hello, client")

		time.Sleep(time.Second * 5)
	}
}

func main() {
	fmt.Println("start ws... ...")
	http.HandleFunc("/chat", chat)
	http.ListenAndServe(":9999", nil)
}

Client

package main

import (
	"fmt"
	"net/url"
	"time"

	"github.com/gorilla/websocket"
)

func main() {
	wsUrl := url.URL{Scheme: "ws", Host: "127.0.0.1:9999", Path: "/chat"}
	conn, _, err := (&websocket.Dialer{}).Dial(wsUrl.String(), nil)
	if err != nil {
		fmt.Printf("conn create err:%s\n", err.Error())
		return
	}
	defer conn.Close()
	fmt.Println("conn succ")
	for {
		err = conn.WriteMessage(websocket.TextMessage, []byte("hello, server"))
		if err != nil {
			fmt.Printf("send err:%s\n", err.Error())
			return
		}
		fmt.Println("send: hello, server")

		_, msgdata, err := conn.ReadMessage()
		if err != nil {
			fmt.Printf("conn read err:%s\n", err.Error())
			return
		}
		fmt.Printf("recv: %s\n", string(msgdata))

		time.Sleep(time.Second * 5)
	}
}