adding lifecycle hook for test exit
This commit is contained in:
@@ -28,7 +28,9 @@ func addMiddleware(mux *chi.Mux) {
|
||||
|
||||
func addRoutes(mux *chi.Mux) {
|
||||
// Add a simple resource
|
||||
mux.Mount("/api/id", AppResource{}.Routes())
|
||||
mux.Mount("/api/id", AppResource{}.Routes())
|
||||
// lifecycle management hooks
|
||||
mux.Mount("/.service", AppLifecycleResource{}.Routes())
|
||||
// Live a healthy life!
|
||||
mux.Get("/", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Write([]byte("yolo"))
|
||||
|
||||
26
api/lifecycle.go
Normal file
26
api/lifecycle.go
Normal file
@@ -0,0 +1,26 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"github.com/go-chi/chi"
|
||||
"github.com/go-chi/render"
|
||||
"net/http"
|
||||
"os"
|
||||
)
|
||||
|
||||
type AppLifecycleResource struct{}
|
||||
|
||||
// Routes creates a REST router for the todos resource
|
||||
func (self AppLifecycleResource) Routes() chi.Router {
|
||||
r := chi.NewRouter()
|
||||
|
||||
r.Route("/lifecycle", func(r chi.Router) {
|
||||
r.Post("/stop", self.stop)
|
||||
})
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
func (self AppLifecycleResource) stop(w http.ResponseWriter, r *http.Request) {
|
||||
render.Status(r, http.StatusOK)
|
||||
os.Exit(0)
|
||||
}
|
||||
Reference in New Issue
Block a user