Example for negroni

Author: sebest

examples/negroni/main.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"net/http"
+
+	"github.com/codegangsta/negroni"
+	"github.com/gorilla/mux"
+	"github.com/sebest/xff"
+)
+
+func main() {
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Write([]byte("hello from " + r.RemoteAddr + "\n"))
+	})
+
+	mux := mux.NewRouter()
+	mux.Handle("/", handler)
+
+	n := negroni.Classic()
+	n.Use(negroni.HandlerFunc(xff.XFF))
+	n.UseHandler(mux)
+	n.Run(":3000")
+}

examples/server.go renamed to examples/nethttp/main.go

@@ -11,5 +11,5 @@ func main() {
 		w.Write([]byte("hello from " + r.RemoteAddr + "\n"))
 	})
 
-	http.ListenAndServe(":8080", xff.Handler(handler))
+	http.ListenAndServe(":3000", xff.Handler(handler))
 }

xff.go

@@ -65,5 +65,13 @@ func Handler(h http.Handler) http.Handler {
 	})
 }
 
-// XFF equals Handler for backward compatibility.
-var XFF = Handler
+// HandlerFunc is a Martini compatible handler
+func HandlerFunc(w http.ResponseWriter, r *http.Request) {
+	r.RemoteAddr = GetRemoteAddr(r)
+}
+
+// XFF is a Negroni compatible interface
+func XFF(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
+	r.RemoteAddr = GetRemoteAddr(r)
+	next(w, r)
+}