atcoder
diff --git a/‎server.go
Copy file name to clipboardExpand all lines: server.go
+6Lines changed: 6 additions & 0 deletions b/‎server.go
Copy file name to clipboardExpand all lines: server.go
+6Lines changed: 6 additions & 0 deletions
@@ -25,6 +25,12 @@ func handle(w http.ResponseWriter, r *http.Request) {
 	upgrade := r.Header.Get("Upgrade")
 	if upgrade == "websocket" || upgrade == "Websocket" {
 		websocket.Handler(func(ws *websocket.Conn) {
+			ws.SetDeadline(time.Now().Add(time.Second * 10))
+			var msg = make([]byte, 512)
+			if _, err := ws.Read(msg); err != nil {
+				ERROR.Println(err)
+			}
+			WARN.Println(time.Now(), string(msg))
 			r.Method = "WS"
 			handleInternal(w, r, ws)
 		}).ServeHTTP(w, r)