forked from mirror/websocket
Moved Upgrader variable to package level.
This commit is contained in:
parent
b03dcbad2a
commit
8827bd1a23
|
@ -26,15 +26,17 @@ import (
|
||||||
"unicode/utf8"
|
"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.
|
// echoCopy echoes messages from the client using io.Copy.
|
||||||
func echoCopy(w http.ResponseWriter, r *http.Request, writerOnly bool) {
|
func echoCopy(w http.ResponseWriter, r *http.Request, writerOnly bool) {
|
||||||
u := websocket.Upgrader{
|
conn, err := upgrader.Upgrade(w, r, nil)
|
||||||
ReadBufferSize: 4096,
|
|
||||||
WriteBufferSize: 4096,
|
|
||||||
CheckOrigin: func(r *http.Request) bool {
|
|
||||||
return true
|
|
||||||
}}
|
|
||||||
conn, err := u.Upgrade(w, r, nil)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("Upgrade:", err)
|
log.Println("Upgrade:", err)
|
||||||
http.Error(w, "Bad request", 400)
|
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
|
// echoReadAll echoes messages from the client by reading the entire message
|
||||||
// with ioutil.ReadAll.
|
// with ioutil.ReadAll.
|
||||||
func echoReadAll(w http.ResponseWriter, r *http.Request, writeMessage bool) {
|
func echoReadAll(w http.ResponseWriter, r *http.Request, writeMessage bool) {
|
||||||
u := websocket.Upgrader{
|
conn, err := upgrader.Upgrade(w, r, nil)
|
||||||
ReadBufferSize: 4096,
|
|
||||||
WriteBufferSize: 4096,
|
|
||||||
CheckOrigin: func(r *http.Request) bool {
|
|
||||||
return true
|
|
||||||
}}
|
|
||||||
conn, err := u.Upgrade(w, r, nil)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("Upgrade:", err)
|
log.Println("Upgrade:", err)
|
||||||
return
|
return
|
||||||
|
|
|
@ -25,6 +25,11 @@ const (
|
||||||
maxMessageSize = 512
|
maxMessageSize = 512
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var upgrader = websocket.Upgrader{
|
||||||
|
ReadBufferSize: 1024,
|
||||||
|
WriteBufferSize: 1024,
|
||||||
|
}
|
||||||
|
|
||||||
// connection is an middleman between the websocket connection and the hub.
|
// connection is an middleman between the websocket connection and the hub.
|
||||||
type connection struct {
|
type connection struct {
|
||||||
// The websocket connection.
|
// The websocket connection.
|
||||||
|
@ -89,10 +94,7 @@ func serveWs(w http.ResponseWriter, r *http.Request) {
|
||||||
http.Error(w, "Method not allowed", 405)
|
http.Error(w, "Method not allowed", 405)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
u := websocket.Upgrader{
|
ws, err := upgrader.Upgrade(w, r, nil)
|
||||||
ReadBufferSize: 1024,
|
|
||||||
WriteBufferSize: 1024}
|
|
||||||
ws, err := u.Upgrade(w, r, nil)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if _, ok := err.(websocket.HandshakeError); !ok {
|
if _, ok := err.(websocket.HandshakeError); !ok {
|
||||||
log.Println(err)
|
log.Println(err)
|
||||||
|
|
|
@ -35,6 +35,10 @@ var (
|
||||||
addr = flag.String("addr", ":8080", "http service address")
|
addr = flag.String("addr", ":8080", "http service address")
|
||||||
homeTempl = template.Must(template.New("").Parse(homeHTML))
|
homeTempl = template.Must(template.New("").Parse(homeHTML))
|
||||||
filename string
|
filename string
|
||||||
|
upgrader = websocket.Upgrader{
|
||||||
|
ReadBufferSize: 1024,
|
||||||
|
WriteBufferSize: 1024,
|
||||||
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
func readFileIfModified(lastMod time.Time) ([]byte, time.Time, error) {
|
func readFileIfModified(lastMod time.Time) ([]byte, time.Time, error) {
|
||||||
|
@ -107,10 +111,7 @@ func writer(ws *websocket.Conn, lastMod time.Time) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func serveWs(w http.ResponseWriter, r *http.Request) {
|
func serveWs(w http.ResponseWriter, r *http.Request) {
|
||||||
u := websocket.Upgrader{
|
ws, err := upgrader.Upgrade(w, r, nil)
|
||||||
ReadBufferSize: 1024,
|
|
||||||
WriteBufferSize: 1024}
|
|
||||||
ws, err := u.Upgrade(w, r, nil)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if _, ok := err.(websocket.HandshakeError); !ok {
|
if _, ok := err.(websocket.HandshakeError); !ok {
|
||||||
log.Println(err)
|
log.Println(err)
|
||||||
|
|
Loading…
Reference in New Issue