Setup bare http server

Author: simondanielsson

cmd/internal/config/config.go

@@ -1,7 +1,19 @@
 package config
 
+import "time"
+
 // Load loads a config.
-func Load() (Config, error) {
+func Load(getenv func(string) string) (Config, error) {
 	// TODO: load environment variables into the config using viper
-	return Config{}, nil
+	return Config{
+		Server: ServerConfig{
+			AppEnv:       "local",
+			Addr:         "8999",
+			ReadTimeout:  30 * time.Second,
+			WriteTimeout: 30 * time.Second,
+		},
+		JWT: JWTConfig{
+			Secret: "123",
+		},
+	}, nil
 }

cmd/internal/config/models.go

@@ -1,9 +1,13 @@
 package config
 
+import "time"
+
 // ServerConfig represents the configuration for the server.
 type ServerConfig struct {
-	BindAddr string `mapstructure:"bind_address"`
-	AppEnv   string `mapstructure:"app_env"`
+	AppEnv       string        `mapstructure:"app_env"`
+	Addr         string        `mapstructure:"address"`
+	ReadTimeout  time.Duration `mapstructure:"read_timeout"`
+	WriteTimeout time.Duration `mapstructure:"write_timeout"`
 }
 
 // JWTConfig represents the configuration of the JWT secret

cmd/internal/logging/logging.go

@@ -0,0 +1,18 @@
+package logging
+
+import (
+	"io"
+	"log"
+	"net/http"
+)
+
+func NewLogger(w io.Writer) *log.Logger {
+	logger := log.Logger{}
+	logger.SetOutput(w)
+	logger.SetPrefix("HELLO PREFIX | ")
+	return &logger
+}
+
+func AddLoggingMiddleware(handler http.Handler, logger *log.Logger) http.Handler {
+	return handler
+}

cmd/internal/server/server.go

@@ -0,0 +1,38 @@
+package server
+
+import (
+	"log"
+	"net/http"
+
+	"github.com/simondanielsson/recite/cmd/internal/config"
+	"github.com/simondanielsson/recite/cmd/internal/logging"
+)
+
+type Server struct {
+	server http.Server
+}
+
+func New(config config.Config, logger *log.Logger) http.Server {
+	mux := http.NewServeMux()
+	// Register routes
+	mux.Handle("/", rootMux(logger))
+
+	var handler http.Handler = mux
+	handler = logging.AddLoggingMiddleware(handler, logger)
+
+	// TODO: might not need this wrapping
+	return http.Server{
+		Addr:         ":" + config.Server.Addr,
+		ReadTimeout:  config.Server.ReadTimeout,
+		WriteTimeout: config.Server.WriteTimeout,
+		Handler:      handler,
+	}
+}
+
+func rootMux(logger *log.Logger) http.Handler {
+	return http.HandlerFunc(
+		func(w http.ResponseWriter, r *http.Request) {
+			logger.Println("CALLED /")
+		},
+	)
+}

cmd/recite/main.go

@@ -1,7 +1,55 @@
 package main
 
-import "os"
+import (
+	"context"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/simondanielsson/recite/cmd/internal/config"
+	"github.com/simondanielsson/recite/cmd/internal/logging"
+	"github.com/simondanielsson/recite/cmd/internal/server"
+)
 
 func main() {
-	os.Exit(0)
+	getenv := os.Getenv
+	ctx := context.Background()
+
+	if err := run(ctx, getenv, os.Stdout, os.Stderr, os.Args); err != nil {
+		fmt.Fprintf(os.Stderr, "%s\n", err)
+		os.Exit(0)
+	}
+}
+
+func run(ctx context.Context, getenv func(string) string, outWriter io.Writer, errWriter io.Writer, args []string) error {
+	config, err := config.Load(getenv)
+	if err != nil {
+		return err
+	}
+
+	logger := logging.NewLogger(outWriter)
+	server := server.New(config, logger)
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, os.Interrupt, syscall.SIGTERM, syscall.SIGINT)
+	go func() {
+		// Note: goroutine waits here until interrupt
+		<-quit
+		logger.Println("shutting down server...")
+
+		if err := server.Shutdown(ctx); err != nil {
+			logger.Fatalf("failed to shutdown server: %v", err)
+		}
+		logger.Println("server stopped gracefully")
+	}()
+
+	logger.Printf("listening on %s\n", server.Addr)
+	if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+		fmt.Fprintf(errWriter, "error listening and serving: %s\n", err)
+		return err
+	}
+	return nil
 }