From 04a5024064542f27c28a780be1311413a76bbdaa Mon Sep 17 00:00:00 2001 From: DutchEllie Date: Thu, 23 Sep 2021 13:30:30 +0200 Subject: [PATCH] Added reload function --- pepeservice/main.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/pepeservice/main.go b/pepeservice/main.go index 7d2c01c..e253a5c 100644 --- a/pepeservice/main.go +++ b/pepeservice/main.go @@ -45,6 +45,7 @@ func main() { mux := http.NewServeMux() mux.HandleFunc("/pepe", app.sendPepe) + mux.HandleFunc("/reload", app.reloadList) app.infoLog.Printf("Starting server at :4000\n") err = http.ListenAndServe(":4000", mux) @@ -63,3 +64,21 @@ func (app *application) sendPepe(w http.ResponseWriter, r *http.Request) { w.Write([]byte(URL)) } + +func (app *application) reloadList(w http.ResponseWriter, r *http.Request) { + file, err := os.Open(app.pepe_dir) + if err != nil { + app.errorLog.Printf("Error opening pepe directory\n") + return + } + defer file.Close() + pepe_list, err := file.Readdirnames(0) + if err != nil { + app.errorLog.Printf("Error reading pepe directory file names\n") + return + } + app.pepe_list = pepe_list + + w.WriteHeader(http.StatusOK) + w.Write([]byte("200 - Reloaded the list of pepes")) +}