fix: CloseNotifier deprecated in later version of Go

Fix ported from PR: afex#103 Author: https://github.com/lpxxn
unbxd · Oct 19, 2019 · 1dbda09 · 1dbda09
1 parent 28ca412
commit 1dbda09
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/hystrix/eventstream.go b/hystrix/eventstream.go
@@ -50,7 +50,7 @@ func (sh *StreamHandler) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
  events := sh.register(req)
  defer sh.unregister(req)
 
- notify := rw.(http.CloseNotifier).CloseNotify()
+ notify := req.Context().Done()
 
  rw.Header().Add("Content-Type", "text/event-stream")
  rw.Header().Set("Cache-Control", "no-cache")

diff --git a/loadtest/service/main.go b/loadtest/service/main.go
@@ -53,7 +53,7 @@ func main() {
  if err != nil {
  log.Fatalf("could not initialize statsd client: %v", err)
  }
- metricCollector.Registry.Register(c.NewStatsdCollector)
+ metric.Registry.Register(c.NewStatsdCollector)
 
  hystrix.ConfigureCommand("test", hystrix.CommandConfig{
  Timeout: 50,