Browse Source

Merge branch 'master' of git.lattuga.net:boyska/circolog

Blallo 5 years ago
parent
commit
6da968177b
2 changed files with 10 additions and 1 deletions
  1. 1 1
      cmd/circolog-tail/main.go
  2. 9 0
      cmd/circologd/http_log.go

+ 1 - 1
cmd/circolog-tail/main.go

@@ -149,7 +149,7 @@ func main() {
 			select {
 			case <-done:
 				log.Println("Successfully close")
-			case <-time.After(1 * time.Second):
+			case <-time.After(5 * time.Second):
 				log.Println("Forced close")
 			}
 			return

+ 9 - 0
cmd/circologd/http_log.go

@@ -153,6 +153,15 @@ func getWSHandler(hub circolog.Hub) http.HandlerFunc {
 				hub.Unregister <- c
 				conn.Close()
 			}()
+			go func() {
+				for {
+					_, _, err := conn.ReadMessage()
+					if err != nil {
+						conn.Close()
+						return
+					}
+				}
+			}()
 			for {
 				select {
 				case message, ok := <-c.Messages: