App for fetching and uploading images to cloud bucket storage

server/server.go

@@ -0,0 +1,85 @@
+// Package server provides functionality to easily set up an HTTTP server.
+//
+// Clients:
+//		Database
+package server
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/gorilla/mux"
+	"github.com/kevinmidboe/planetposen-images/clients/gcs"
+	"github.com/kevinmidboe/planetposen-images/config"
+	log "github.com/sirupsen/logrus"
+)
+
+// Server holds the HTTP server, router, config and all clients.
+type Server struct {
+	Config    *config.Config
+	HTTP      *http.Server
+	Router    *mux.Router
+	GCSClient gcs.Client
+}
+
+// Create sets up the HTTP server, router and all clients.
+// Returns an error if an error occurs.
+func (s *Server) Create(ctx context.Context, config *config.Config) error {
+	// metrics.RegisterPrometheusCollectors()
+
+	s.Config = config
+	s.Router = mux.NewRouter()
+	s.HTTP = &http.Server{
+		Addr:    fmt.Sprintf(":%s", s.Config.Port),
+		Handler: s.Router,
+	}
+
+	gcsClient, err := gcs.NewClient(ctx, config.GCSBucket)
+	if err != nil {
+		return err
+	}
+
+	s.GCSClient = gcsClient
+
+	s.setupRoutes()
+
+	return nil
+}
+
+// Serve tells the server to start listening and serve HTTP requests.
+// It also makes sure that the server gracefully shuts down on exit.
+// Returns an error if an error occurs.
+func (s *Server) Serve(ctx context.Context) error {
+	// closer, err := trace.InitGlobalTracer(s.Config)
+
+	// if err != nil {
+	// 	return err
+	// }
+
+	// defer closer.Close()
+
+	go func(ctx context.Context, s *Server) {
+		stop := make(chan os.Signal, 1)
+		signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
+
+		<-stop
+
+		log.Info("Shutdown signal received")
+
+		if err := s.HTTP.Shutdown(ctx); err != nil {
+			log.Error(err.Error())
+		}
+	}(ctx, s)
+
+	log.Infof("Ready at: %s", s.Config.Port)
+
+	if err := s.HTTP.ListenAndServe(); err != http.ErrServerClosed {
+		log.Fatalf(err.Error())
+	}
+
+	return nil
+}