61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
type killer struct{}
|
|
|
|
func (k killer) Signal() {}
|
|
func (k killer) String() string {
|
|
return "0"
|
|
}
|
|
|
|
type ShutdownHandler interface {
|
|
WatchForShutdown(server *http.Server)
|
|
ServeHTTP(w http.ResponseWriter, r *http.Request)
|
|
}
|
|
|
|
type shutdownService struct {
|
|
sigCh chan os.Signal
|
|
}
|
|
|
|
func (s shutdownService) WatchForShutdown(server *http.Server) {
|
|
/* Handles shutdown requests */
|
|
<-s.sigCh
|
|
err := server.Shutdown(context.Background())
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Server could not shut down gracefully: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
type ShutdownRequest struct {
|
|
Restart bool `json:"restart"`
|
|
}
|
|
|
|
/* Shuts down the HTTP server and exit the process by signaling to the shutdown channel */
|
|
func (s shutdownService) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
|
|
payload := r.Context().Value(payload("payload")).(*ShutdownRequest)
|
|
|
|
var text = "Shut down server"
|
|
if payload.Restart {
|
|
text = "Restarted server"
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
response := SuccessResponse{Message: text}
|
|
|
|
err := json.NewEncoder(w).Encode(response)
|
|
if err != nil {
|
|
handleError(w, err, "Could not encode response", http.StatusInternalServerError)
|
|
} else {
|
|
s.sigCh <- killer{}
|
|
}
|
|
}
|