codysoyland · August 29, 2015 13:56
diff --git a/websockets.go b/websockets.go
 package main

 import (
 	"log"
 	"net/http"

 	"github.com/gorilla/websocket"
 )

 func main() {
 	conns := make([]*websocket.Conn, 0)
 	conn_chan := make(chan *websocket.Conn)
 	color_chan := make(chan string)

 	go func() {
 		for {
 			select {
 			case conn := <-conn_chan:
 				conns = append(conns, conn)
 			case color := <-color_chan:
 				data := map[string]string{"color": color}
 				for _, conn := range conns {
 					conn.WriteJSON(data)
 				}
 			}
 		}
 	}()

 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte(`
 		<html><head><title>Websocket test</title></head><body>
 			<script src="//ajax.googleapis.com/ajax/libs/jquery/1.10.2/jquery.min.js"></script>
 		
 			<input id="color" type="text"/>
 			<button id="submit">submit</button>

 			<script type="text/javascript">
 				var ws = new WebSocket("ws://" + window.location.host + "/ws")
 				ws.onmessage = function(mes) {
 					var color = JSON.parse(mes.data)['color']
 					document.body.style.background = color
 				}
 				$('#submit').click(function() {
 					var value = JSON.stringify({'color': $('#color').val()})
 					ws.send(value)
 				})
 			</script>
 		</body></html>
 		`))
 	})

 	http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
 		ws, err := websocket.Upgrade(w, r, nil, 1024, 1024)
 		if _, ok := err.(websocket.HandshakeError); ok {
 			http.Error(w, "Not a websocket handshake", 400)
 			return
 		} else if err != nil {
 			log.Println(err)
 			http.Error(w, err.Error(), 500)
 			return
 		}
 		conn_chan <- ws
 		for {
 			data := make(map[string]string)
 			err := ws.ReadJSON(&data)
 			if err != nil {
 				log.Println(err)
 				return
 			}
 			color_chan <- data["color"]
 		}
 	})
 	http.ListenAndServe(":9000", nil)
 }
	package main

	import (
	"log"
	"net/http"

	"github.com/gorilla/websocket"
	)

	func main() {
	conns := make([]*websocket.Conn, 0)
	conn_chan := make(chan *websocket.Conn)
	color_chan := make(chan string)

	go func() {
	for {
	select {
	case conn := <-conn_chan:
	conns = append(conns, conn)
	case color := <-color_chan:
	data := map[string]string{"color": color}
	for _, conn := range conns {
	conn.WriteJSON(data)
	}
	}
	}
	}()

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
	w.Write([]byte(`
	<html><head><title>Websocket test</title></head><body>
	<script src="//ajax.googleapis.com/ajax/libs/jquery/1.10.2/jquery.min.js"></script>

	<input id="color" type="text"/>
	<button id="submit">submit</button>

	<script type="text/javascript">
	var ws = new WebSocket("ws://" + window.location.host + "/ws")
	ws.onmessage = function(mes) {
	var color = JSON.parse(mes.data)['color']
	document.body.style.background = color
	}
	$('#submit').click(function() {
	var value = JSON.stringify({'color': $('#color').val()})
	ws.send(value)
	})
	</script>
	</body></html>
	`))
	})

	http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
	ws, err := websocket.Upgrade(w, r, nil, 1024, 1024)
	if _, ok := err.(websocket.HandshakeError); ok {
	http.Error(w, "Not a websocket handshake", 400)
	return
	} else if err != nil {
	log.Println(err)
	http.Error(w, err.Error(), 500)
	return
	}
	conn_chan <- ws
	for {
	data := make(map[string]string)
	err := ws.ReadJSON(&data)
	if err != nil {
	log.Println(err)
	return
	}
	color_chan <- data["color"]
	}
	})
	http.ListenAndServe(":9000", nil)
	}