commit
7262164ef0
@ -0,0 +1,67 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"git.mnrva.dev/gabehf/go-project-template/app/middleware"
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/cfg"
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/db"
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/logger"
|
||||
"github.com/go-chi/chi/v5"
|
||||
chimiddleware "github.com/go-chi/chi/v5/middleware"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
func NewServer(logger *zerolog.Logger, cfg *cfg.Config, db db.DB) http.Handler {
|
||||
mux := chi.NewRouter()
|
||||
// bind general middleware to mux
|
||||
mux.Use(middleware.Logger(logger))
|
||||
mux.Use(chimiddleware.RealIP)
|
||||
mux.Use(chimiddleware.Recoverer)
|
||||
// call router binds on mux
|
||||
bindRoutes(mux, logger, cfg, db)
|
||||
return mux
|
||||
}
|
||||
|
||||
func Run(
|
||||
ctx context.Context,
|
||||
getenv func(string) string,
|
||||
stdout io.Writer,
|
||||
db db.DB,
|
||||
) error {
|
||||
cfg, err := cfg.Load(getenv)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to load configuration: %v", err)
|
||||
}
|
||||
logger := logger.Get(cfg).Output(stdout)
|
||||
srv := NewServer(&logger, cfg, db)
|
||||
httpServer := &http.Server{
|
||||
Addr: cfg.ListenAddr,
|
||||
Handler: srv,
|
||||
}
|
||||
logger.Info().Msg("listening on " + cfg.ListenAddr)
|
||||
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
return err
|
||||
}
|
||||
// Wait for interrupt signal to gracefully shutdown the server with a timeout of 10 seconds.
|
||||
// Use a buffered channel to avoid missing signals as recommended for signal.Notify
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-quit
|
||||
logger.Info().Msg("recieved server shutdown notice")
|
||||
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
|
||||
defer cancel()
|
||||
logger.Info().Msg("waiting for all processes to finish...")
|
||||
if err := httpServer.Shutdown(ctx); err != nil {
|
||||
return err
|
||||
}
|
||||
logger.Info().Msg("shutdown successful")
|
||||
return nil
|
||||
}
|
||||
@ -0,0 +1,2 @@
|
||||
// package handlers implements route handlers
|
||||
package handlers
|
||||
@ -0,0 +1,49 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"runtime/debug"
|
||||
"time"
|
||||
|
||||
"github.com/go-chi/chi/v5/middleware"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
func Logger(logger *zerolog.Logger) func(next http.Handler) http.Handler {
|
||||
return func(next http.Handler) http.Handler {
|
||||
fn := func(w http.ResponseWriter, r *http.Request) {
|
||||
log := logger.With().Logger()
|
||||
ww := middleware.NewWrapResponseWriter(w, r.ProtoMajor)
|
||||
t1 := time.Now()
|
||||
defer func() {
|
||||
t2 := time.Now()
|
||||
if rec := recover(); rec != nil {
|
||||
log.Error().
|
||||
Str("type", "error").
|
||||
Timestamp().
|
||||
Interface("recover_info", rec).
|
||||
Bytes("debug_stack", debug.Stack()).
|
||||
Msg("log system error")
|
||||
http.Error(ww, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
|
||||
}
|
||||
log.Info().
|
||||
Str("type", "access").
|
||||
Timestamp().
|
||||
Fields(map[string]interface{}{
|
||||
"remote_ip": r.RemoteAddr,
|
||||
"url": r.URL.Path,
|
||||
"proto": r.Proto,
|
||||
"method": r.Method,
|
||||
"user_agent": r.Header.Get("User-Agent"),
|
||||
"status": ww.Status(),
|
||||
"latency_ms": float64(t2.Sub(t1).Nanoseconds()) / 1000000.0,
|
||||
"bytes_in": r.Header.Get("Content-Length"),
|
||||
"bytes_out": ww.BytesWritten(),
|
||||
}).
|
||||
Msg("incoming_request")
|
||||
}()
|
||||
next.ServeHTTP(ww, r)
|
||||
}
|
||||
return http.HandlerFunc(fn)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,14 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/cfg"
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/db"
|
||||
"github.com/go-chi/chi/v5"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
func bindRoutes(r *chi.Mux, logger *zerolog.Logger, cfg *cfg.Config, db db.DB) {
|
||||
r.Route("/api/v1", func(r chi.Router) {
|
||||
// define routes here
|
||||
})
|
||||
}
|
||||
@ -0,0 +1,21 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"git.mnrva.dev/gabehf/go-project-template/app"
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/cfg"
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/db/memorydb"
|
||||
)
|
||||
|
||||
func main() {
|
||||
ctx := context.Background()
|
||||
store := memorydb.New(&cfg.Config{})
|
||||
defer store.Close()
|
||||
if err := app.Run(ctx, os.Getenv, os.Stdout, store); err != nil {
|
||||
fmt.Fprintf(os.Stderr, "%s\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,14 @@
|
||||
module git.mnrva.dev/gabehf/go-project-template
|
||||
|
||||
go 1.23.7
|
||||
|
||||
require (
|
||||
github.com/go-chi/chi/v5 v5.2.1
|
||||
github.com/rs/zerolog v1.33.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/mattn/go-colorable v0.1.13 // indirect
|
||||
github.com/mattn/go-isatty v0.0.19 // indirect
|
||||
golang.org/x/sys v0.12.0 // indirect
|
||||
)
|
||||
@ -0,0 +1,17 @@
|
||||
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
||||
github.com/go-chi/chi/v5 v5.2.1 h1:KOIHODQj58PmL80G2Eak4WdvUzjSJSm0vG72crDCqb8=
|
||||
github.com/go-chi/chi/v5 v5.2.1/go.mod h1:L2yAIGWB3H+phAw1NxKwWM+7eUH/lU8pOMm5hHcoops=
|
||||
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
|
||||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
|
||||
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg=
|
||||
github.com/rs/zerolog v1.33.0 h1:1cU2KZkvPxNyfgEmhHAz/1A9Bz+llsdYzklWFzgp0r8=
|
||||
github.com/rs/zerolog v1.33.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss=
|
||||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.12.0 h1:CM0HF96J0hcLAwsHPJZjfdNzs0gftsLfgKt57wWHJ0o=
|
||||
golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
@ -0,0 +1,28 @@
|
||||
package cfg
|
||||
|
||||
const (
|
||||
BASE_URL_ENV = "go-project-template_BASE_URL"
|
||||
DATABASE_URL_ENV = "go-project-template_DATABASE_URL"
|
||||
LISTEN_ADDR_ENV = "go-project-template_LISTEN_ADDR"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
ListenAddr string
|
||||
BaseUrl string
|
||||
DatabaseUrl string
|
||||
LogLevel int
|
||||
}
|
||||
|
||||
func Load(getenv func(string) string) (*Config, error) {
|
||||
cfg := new(Config)
|
||||
cfg.BaseUrl = getenv(BASE_URL_ENV)
|
||||
if cfg.BaseUrl == "" {
|
||||
cfg.BaseUrl = "http://localhost"
|
||||
}
|
||||
cfg.DatabaseUrl = getenv(DATABASE_URL_ENV)
|
||||
cfg.ListenAddr = getenv(LISTEN_ADDR_ENV)
|
||||
if cfg.ListenAddr == "" {
|
||||
cfg.ListenAddr = ":3000"
|
||||
}
|
||||
return cfg, nil
|
||||
}
|
||||
@ -0,0 +1,6 @@
|
||||
// package db defines the database interface
|
||||
package db
|
||||
|
||||
type DB interface {
|
||||
Close()
|
||||
}
|
||||
@ -0,0 +1,20 @@
|
||||
// memorydb implements the db interface with an in-memory db for testing
|
||||
package memorydb
|
||||
|
||||
import (
|
||||
"sync"
|
||||
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/cfg"
|
||||
)
|
||||
|
||||
type MemoryDB struct {
|
||||
m map[string]interface{}
|
||||
mut sync.RWMutex
|
||||
}
|
||||
|
||||
func New(cfg *cfg.Config) *MemoryDB {
|
||||
return new(MemoryDB)
|
||||
}
|
||||
|
||||
func (d *MemoryDB) Close() {
|
||||
}
|
||||
@ -0,0 +1,29 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"git.mnrva.dev/gabehf/go-project-template/internal/cfg"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
var once sync.Once
|
||||
|
||||
var log zerolog.Logger
|
||||
|
||||
func Get(cfg *cfg.Config) zerolog.Logger {
|
||||
once.Do(func() {
|
||||
zerolog.TimeFieldFormat = zerolog.TimeFormatUnixMs
|
||||
|
||||
logLevel := cfg.LogLevel
|
||||
|
||||
log = zerolog.New(os.Stdout).
|
||||
Level(zerolog.Level(logLevel)).
|
||||
With().
|
||||
Timestamp().
|
||||
Logger()
|
||||
})
|
||||
|
||||
return log
|
||||
}
|
||||
@ -0,0 +1,2 @@
|
||||
// package models defines the schema of objects in the database
|
||||
package models
|
||||
Loading…
Reference in new issue