diff --git a/cmd/server/main.go b/cmd/server/main.go new file mode 100644 index 0000000..f73a33f --- /dev/null +++ b/cmd/server/main.go @@ -0,0 +1,43 @@ +package main + +import ( + "context" + "log" + "net/http" + "os" + "os/signal" + "syscall" + "time" + + "git.jinshen.cn/remilia/push-server/interval/api" + "git.jinshen.cn/remilia/push-server/interval/server" +) + +func main() { + _, serverCancel := context.WithCancel(context.Background()) + defer func() { + serverCancel() + }() + + httpServer := server.NewHTTPServer(":8080", api.NewRouter()) + + go func() { + log.Println("Starting HTTP server on :8080") + if err := httpServer.Start(); err != nil && err != http.ErrServerClosed { + log.Fatalf("HTTP server error: %v", err) + } + }() + + sig := make(chan os.Signal, 1) + signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM) + <-sig + + log.Println("Shutting down server...") + + serverCancel() + + shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), time.Second*10) + defer shutdownCancel() + + httpServer.Shutdown(shutdownCtx) +} diff --git a/interval/api/handler/health.go b/interval/api/handler/health.go new file mode 100644 index 0000000..1460723 --- /dev/null +++ b/interval/api/handler/health.go @@ -0,0 +1,8 @@ +package handler + +import "net/http" + +func Health(w http.ResponseWriter, _ *http.Request) { + w.WriteHeader(http.StatusOK) + w.Write([]byte("OK")) +} diff --git a/interval/api/router.go b/interval/api/router.go new file mode 100644 index 0000000..d0ba6bc --- /dev/null +++ b/interval/api/router.go @@ -0,0 +1,16 @@ +package api + +import ( + "net/http" + + "git.jinshen.cn/remilia/push-server/interval/api/handler" + "github.com/go-chi/chi/v5" +) + +func NewRouter() http.Handler { + r := chi.NewRouter() + + r.Post("/health", handler.Health) + + return r +} diff --git a/interval/server/http.go b/interval/server/http.go new file mode 100644 index 0000000..6b3e73b --- /dev/null +++ b/interval/server/http.go @@ -0,0 +1,27 @@ +package server + +import ( + "context" + "net/http" +) + +type HTTPServer struct { + server *http.Server +} + +func NewHTTPServer(addr string, handler http.Handler) *HTTPServer { + return &HTTPServer{ + server: &http.Server{ + Addr: addr, + Handler: handler, + }, + } +} + +func (s *HTTPServer) Start() error { + return s.server.ListenAndServe() +} + +func (s *HTTPServer) Shutdown(ctx context.Context) error { + return s.server.Shutdown(ctx) +}