From 2bb006339d4ceff64a6910c38051728d780c285b Mon Sep 17 00:00:00 2001 From: Claudio La Barbera Date: Wed, 22 Mar 2023 15:29:40 +0100 Subject: [PATCH] added batman in homepage --- main.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/main.go b/main.go index 6da12fb..5977f45 100644 --- a/main.go +++ b/main.go @@ -57,7 +57,19 @@ func handler(w http.ResponseWriter, r *http.Request) { flusher := w.(http.Flusher) vars := mux.Vars(r) + frameSource := vars["frameSource"] + returnFrames(w, r, cn, flusher, frameSource) +} + +func homeHandler(w http.ResponseWriter, r *http.Request) { + cn := w.(http.CloseNotifier) + flusher := w.(http.Flusher) + + returnFrames(w, r, cn, flusher, "batman") +} + +func returnFrames(w http.ResponseWriter, r *http.Request, cn http.CloseNotifier, flusher http.Flusher, frameSource string) { glog.Infof("Frame source %s", frameSource) frames, ok := frames.FrameMap[frameSource] @@ -110,6 +122,7 @@ func main() { flag.Set("logtostderr", "true") r := mux.NewRouter() + r.HandleFunc("/", homeHandler).Methods("GET") r.HandleFunc("/list", listHandler).Methods("GET") r.HandleFunc("/{frameSource}", handler).Methods("GET") r.NotFoundHandler = http.HandlerFunc(notFoundHandler)