|
|
@@ -26,15 +26,17 @@ import (
|
|
|
"unicode/utf8"
|
|
|
)
|
|
|
|
|
|
+var upgrader = websocket.Upgrader{
|
|
|
+ ReadBufferSize: 4096,
|
|
|
+ WriteBufferSize: 4096,
|
|
|
+ CheckOrigin: func(r *http.Request) bool {
|
|
|
+ return true
|
|
|
+ },
|
|
|
+}
|
|
|
+
|
|
|
// echoCopy echoes messages from the client using io.Copy.
|
|
|
func echoCopy(w http.ResponseWriter, r *http.Request, writerOnly bool) {
|
|
|
- u := websocket.Upgrader{
|
|
|
- ReadBufferSize: 4096,
|
|
|
- WriteBufferSize: 4096,
|
|
|
- CheckOrigin: func(r *http.Request) bool {
|
|
|
- return true
|
|
|
- }}
|
|
|
- conn, err := u.Upgrade(w, r, nil)
|
|
|
+ conn, err := upgrader.Upgrade(w, r, nil)
|
|
|
if err != nil {
|
|
|
log.Println("Upgrade:", err)
|
|
|
http.Error(w, "Bad request", 400)
|
|
|
@@ -93,13 +95,7 @@ func echoCopyFull(w http.ResponseWriter, r *http.Request) {
|
|
|
// echoReadAll echoes messages from the client by reading the entire message
|
|
|
// with ioutil.ReadAll.
|
|
|
func echoReadAll(w http.ResponseWriter, r *http.Request, writeMessage bool) {
|
|
|
- u := websocket.Upgrader{
|
|
|
- ReadBufferSize: 4096,
|
|
|
- WriteBufferSize: 4096,
|
|
|
- CheckOrigin: func(r *http.Request) bool {
|
|
|
- return true
|
|
|
- }}
|
|
|
- conn, err := u.Upgrade(w, r, nil)
|
|
|
+ conn, err := upgrader.Upgrade(w, r, nil)
|
|
|
if err != nil {
|
|
|
log.Println("Upgrade:", err)
|
|
|
return
|