improved net/http example

caarlos0 · caarlos0 · commit 78cf7062ea82 · 2016-06-06T20:54:34.000-03:00
diff --git a/examples/nethttp/server.go b/examples/nethttp/server.go
@@ -1,16 +1,23 @@
 package main
 
 import (
-	"github.com/thoas/stats"
+	"encoding/json"
 	"net/http"
+
+	"github.com/thoas/stats"
 )
 
 func main() {
-	h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+	middleware := stats.New()
+	mux := http.NewServeMux()
+	mux.HandleFunc("/hello", func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "application/json")
 		w.Write([]byte("{\"hello\": \"world\"}"))
 	})
-
-	handler := stats.New().Handler(h)
-	http.ListenAndServe(":8080", handler)
+	mux.HandleFunc("/stats", func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "application/json")
+		b, _ := json.Marshal(middleware.Data())
+		w.Write(b)
+	})
+	http.ListenAndServe(":8080", middleware.Handler(mux))
 }
