71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package website
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"git.handmade.network/hmn/hmn/src/config"
|
|
"git.handmade.network/hmn/hmn/src/db"
|
|
"git.handmade.network/hmn/hmn/src/logging"
|
|
"git.handmade.network/hmn/hmn/src/templates"
|
|
"github.com/julienschmidt/httprouter"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var WebsiteCommand = &cobra.Command{
|
|
Short: "Run the HMN website",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
templates.Init()
|
|
|
|
defer logging.LogPanics()
|
|
|
|
logging.Info().Msg("Hello, HMN!")
|
|
|
|
conn := db.NewConnPool(4, 8)
|
|
|
|
server := http.Server{
|
|
Addr: config.Config.Addr,
|
|
Handler: NewWebsiteRoutes(conn),
|
|
}
|
|
|
|
signals := make(chan os.Signal, 1)
|
|
signal.Notify(signals, os.Interrupt)
|
|
go func() {
|
|
<-signals
|
|
logging.Info().Msg("Shutting down the website")
|
|
timeout, _ := context.WithTimeout(context.Background(), 30*time.Second)
|
|
server.Shutdown(timeout)
|
|
|
|
<-signals
|
|
logging.Warn().Msg("Forcibly killed the website")
|
|
os.Exit(1)
|
|
}()
|
|
|
|
logging.Info().Str("addr", config.Config.Addr).Msg("Serving the website")
|
|
serverErr := server.ListenAndServe()
|
|
if !errors.Is(serverErr, http.ErrServerClosed) {
|
|
logging.Error().Err(serverErr).Msg("Server shut down unexpectedly")
|
|
}
|
|
},
|
|
}
|
|
|
|
func withRequestLogger(h httprouter.Handle) httprouter.Handle {
|
|
return func(rw http.ResponseWriter, r *http.Request, p httprouter.Params) {
|
|
defer logging.LogPanics()
|
|
|
|
start := time.Now()
|
|
defer func() {
|
|
end := time.Now()
|
|
logging.Info().Dur("duration", end.Sub(start)).Msg("Completed request")
|
|
}()
|
|
|
|
h(rw, r, p)
|
|
}
|
|
}
|
|
|
|
// func handleRequestResults
|