Chemmy's Blog

package main

import (
	"fmt"
	"log"
	"net/http"

	"code.google.com/p/go.net/websocket"
)

func echo(ws *websocket.Conn) {
	var err error
	var i int
	for {
		var reply string

		//建立连接后接收来自客户端的信息reply
		if err == websocket.Message.Receive(ws, &reply); err != nil {
			fmt.Println("error, can't receive message ...")
			break
		}

		fmt.Println("recevied from client: " + reply)
		i++

		// 把收到的信息进行处理，可以做过滤也可以返回国定信息
		msg := "received: " + reply
		fmt.Println("send to client: " + msg)
		fmt.Println(i)

		// 返回消息给客户端
		if err = websocket.Message.Send(ws, msg); err != nil {
			fmt.Println("error,can't send message ...")
			break
		}
	}
}

func main() {
	http.Handle("/", websocket.Handle(echo))
	// 访问服务器地址，ws://127.0.0.1:8088
	if err := http.ListenAndServe(":8088", nil); err != nil {
		log.Fatal("listen and serve: ", err)
	}
}

package main

import (
	"log"
	"net/http"

	"github.com/gorilla/websocket"
)

var (
	upgrader = websocket.Upgrader{
		// 读取缓冲区空间大小
		ReadBufferSize: 1024,
		// 写入缓冲区空间大小
		WriteBufferSize: 1024,
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
	}
)

func wsHandler(w http.ResponseWriter, r *http.Request) {
	//完成握手升级为websocket长连接，使用conn发送和接收消息
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println("upgrade: ", err)
		return
	}
	defer conn.Close()

	for {
		messageType, msg, err := conn.ReadMessage()
		if err != nil {
			log.Println("reading error ...", err)
			return
		}
		log.Printf("read from client msg: %s \n", msg)

		if err := conn.WriteMessage(messageType, msg); err != nil {
			log.Println("writing error ...", err)
			return
		}
		log.Printf("write msg to client: %s \n", msg)
	}
}

func main() {
	// 监听地址 ws://127.0.0.1:8088
	http.HandleFunc("/", wsHandler)
	err := http.ListenAndServe(":8088", nil)
	if err != nil {
		log.Fatal("listen and serve ", err.Error())
	}
}