From f500cf945a5aab1e8fdf838510f86b08b5189c95 Mon Sep 17 00:00:00 2001 From: "Alexander \"Arav\" Andreev" Date: Sat, 27 May 2023 20:59:38 +0400 Subject: [PATCH] Updated HttpServer in main. --- cmd/dwelling-home/main.go | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/cmd/dwelling-home/main.go b/cmd/dwelling-home/main.go index e644932..3ac94e4 100644 --- a/cmd/dwelling-home/main.go +++ b/cmd/dwelling-home/main.go @@ -117,20 +117,19 @@ func main() { r.Handler(http.MethodPatch, "/api/mindflow/category/:id", mindflowApi.EditCategory) r.Handler(http.MethodDelete, "/api/mindflow/category/:id", mindflowApi.DeleteCategory) - srv.POST("/api/mindflow-category", mindflowApi.NewCategory) - srv.PATCH("/api/mindflow-category/:id", mindflowApi.EditCategory) - srv.DELETE("/api/mindflow-category/:id", mindflowApi.DeleteCategory) + srv := dwhttp.NewHttpServer(r) if err := srv.Start(network, *listenAddress); err != nil { log.Fatalln(err) } + defer func() { + if err := srv.Stop(); err != nil { + log.Fatalln(err) + } + }() doneSignal := make(chan os.Signal, 1) signal.Notify(doneSignal, os.Interrupt, syscall.SIGINT, syscall.SIGTERM) <-doneSignal - - if err := srv.Stop(); err != nil { - log.Fatalln(err) - } }