Lint fixes 6/n

juanfont · Jun 26, 2022 · fa91ece · fa91ece
1 parent c810b24
commit fa91ece
Showing 1 changed file with 35 additions and 3 deletions.
diff --git a/poll.go b/poll.go
@@ -335,7 +335,18 @@ func (h *Headscale) PollNetMapStream(
 
 				return
 			}
-			writer.(http.Flusher).Flush()
+
+			flusher, ok := writer.(http.Flusher)
+			if !ok {
+				log.Error().
+					Caller().
+					Str("handler", "PollNetMapStream").
+					Str("machine", machine.Hostname).
+					Str("channel", "pollData").
+					Msg("Cannot cast writer to http.Flusher")
+			} else {
+				flusher.Flush()
+			}
 
 			log.Trace().
 				Str("handler", "PollNetMapStream").
@@ -403,7 +414,17 @@ func (h *Headscale) PollNetMapStream(
 
 				return
 			}
-			writer.(http.Flusher).Flush()
+			flusher, ok := writer.(http.Flusher)
+			if !ok {
+				log.Error().
+					Caller().
+					Str("handler", "PollNetMapStream").
+					Str("machine", machine.Hostname).
+					Str("channel", "keepAlive").
+					Msg("Cannot cast writer to http.Flusher")
+			} else {
+				flusher.Flush()
+			}
 
 			log.Trace().
 				Str("handler", "PollNetMapStream").
@@ -492,7 +513,18 @@ func (h *Headscale) PollNetMapStream(
 
 					return
 				}
-				writer.(http.Flusher).Flush()
+
+				flusher, ok := writer.(http.Flusher)
+				if !ok {
+					log.Error().
+						Caller().
+						Str("handler", "PollNetMapStream").
+						Str("machine", machine.Hostname).
+						Str("channel", "update").
+						Msg("Cannot cast writer to http.Flusher")
+				} else {
+					flusher.Flush()
+				}
 
 				log.Trace().
 					Str("handler", "PollNetMapStream").