Last active
May 17, 2016 20:58
-
-
Save luisbebop/c446dd3bb1329606316688bda7846499 to your computer and use it in GitHub Desktop.
Go websocket echo server with permessage-deflate extension enabled
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// Copyright 2015 The Gorilla WebSocket Authors. All rights reserved. | |
// Use of this source code is governed by a BSD-style | |
// license that can be found in the LICENSE file. | |
// +build ignore | |
package main | |
import ( | |
"flag" | |
"fmt" | |
"github.com/euforia/websocket" | |
"html/template" | |
"log" | |
"net/http" | |
"net/http/httputil" | |
) | |
var addr = flag.String("addr", "localhost:8080", "http service address") | |
//var upgrader = websocket.Upgrader{} // use default options | |
var upgrader = websocket.Upgrader{ | |
ReadBufferSize: 2048, | |
WriteBufferSize: 2048, | |
CheckOrigin: func(r *http.Request) bool { | |
dump, _ := httputil.DumpRequest(r, true) | |
fmt.Printf("%q\n\n", dump) | |
return true | |
}, | |
Extensions: []string{websocket.CompressPermessageDeflate}, | |
} | |
func echo(w http.ResponseWriter, r *http.Request) { | |
c, err := upgrader.Upgrade(w, r, nil) | |
if err != nil { | |
log.Print("upgrade:", err) | |
return | |
} | |
defer c.Close() | |
for { | |
mt, message, err := c.ReadMessage() | |
if err != nil { | |
log.Println("read:", err) | |
break | |
} | |
log.Printf("mt: %d len: %d recv: %s", mt, len(message), message) | |
err = c.WriteMessage(mt, message) | |
if err != nil { | |
log.Println("write:", err) | |
break | |
} | |
} | |
} | |
func home(w http.ResponseWriter, r *http.Request) { | |
homeTemplate.Execute(w, "ws://"+r.Host+"/echo") | |
} | |
func main() { | |
flag.Parse() | |
log.SetFlags(0) | |
http.HandleFunc("/echo", echo) | |
http.HandleFunc("/", home) | |
log.Fatal(http.ListenAndServe(*addr, nil)) | |
} | |
var homeTemplate = template.Must(template.New("").Parse(` | |
<!DOCTYPE html> | |
<head> | |
<meta charset="utf-8"> | |
<script> | |
window.addEventListener("load", function(evt) { | |
var output = document.getElementById("output"); | |
var input = document.getElementById("input"); | |
var ws; | |
var print = function(message) { | |
var d = document.createElement("div"); | |
d.innerHTML = message; | |
output.appendChild(d); | |
}; | |
document.getElementById("open").onclick = function(evt) { | |
if (ws) { | |
return false; | |
} | |
ws = new WebSocket("{{.}}"); | |
ws.onopen = function(evt) { | |
print("OPEN"); | |
} | |
ws.onclose = function(evt) { | |
print("CLOSE"); | |
ws = null; | |
} | |
ws.onmessage = function(evt) { | |
print("RESPONSE: " + evt.data); | |
} | |
ws.onerror = function(evt) { | |
print("ERROR: " + evt.data); | |
} | |
return false; | |
}; | |
document.getElementById("send").onclick = function(evt) { | |
if (!ws) { | |
return false; | |
} | |
print("SEND: " + input.value); | |
ws.send(input.value); | |
return false; | |
}; | |
document.getElementById("close").onclick = function(evt) { | |
if (!ws) { | |
return false; | |
} | |
ws.close(); | |
return false; | |
}; | |
}); | |
</script> | |
</head> | |
<body> | |
<table> | |
<tr><td valign="top" width="50%"> | |
<p>Click "Open" to create a connection to the server, | |
"Send" to send a message to the server and "Close" to close the connection. | |
You can change the message and send multiple times. | |
<p> | |
<form> | |
<button id="open">Open</button> | |
<button id="close">Close</button> | |
<p><input id="input" type="text" value="Hello world!"> | |
<button id="send">Send</button> | |
</form> | |
</td><td valign="top" width="50%"> | |
<div id="output"></div> | |
</td></tr></table> | |
</body> | |
</html> | |
`)) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment