hmn/src/website/website.go

83 lines
2.1 KiB
Go
Raw Normal View History

2021-03-09 08:05:07 +00:00
package website
import (
"context"
2021-03-11 03:29:32 +00:00
"errors"
2021-03-09 08:05:07 +00:00
"fmt"
"net/http"
2021-03-11 03:29:32 +00:00
"os"
"os/signal"
"time"
2021-03-09 08:05:07 +00:00
2021-03-11 03:39:24 +00:00
"git.handmade.network/hmn/hmn/src/config"
"git.handmade.network/hmn/hmn/src/db"
"git.handmade.network/hmn/hmn/src/logging"
2021-03-09 08:05:07 +00:00
"github.com/julienschmidt/httprouter"
"github.com/spf13/cobra"
)
var WebsiteCommand = &cobra.Command{
Short: "Run the HMN website",
Run: func(cmd *cobra.Command, args []string) {
2021-03-11 05:02:43 +00:00
defer logging.LogPanics()
2021-03-11 03:19:39 +00:00
logging.Info().Msg("Hello, HMN!")
2021-03-09 08:05:07 +00:00
conn := db.NewConnPool(4, 8)
router := httprouter.New()
2021-03-11 05:02:43 +00:00
router.GET("/", WithRequestLogger(func(rw http.ResponseWriter, r *http.Request, p httprouter.Params) {
rw.Write([]byte(fmt.Sprintf("Hello, HMN! The time is: %v\n", time.Now())))
}))
2021-03-09 08:05:07 +00:00
router.GET("/project/:id", func(rw http.ResponseWriter, r *http.Request, p httprouter.Params) {
id := p.ByName("id")
row := conn.QueryRow(context.Background(), "SELECT name FROM handmade_project WHERE id = $1", p.ByName("id"))
var name string
err := row.Scan(&name)
if err != nil {
panic(err)
}
rw.Write([]byte(fmt.Sprintf("(%s) %s\n", id, name)))
})
2021-03-11 03:29:32 +00:00
server := http.Server{
Addr: config.Config.Addr,
Handler: router,
}
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)
}()
2021-03-11 03:19:39 +00:00
logging.Info().Str("addr", config.Config.Addr).Msg("Serving the website")
2021-03-11 03:29:32 +00:00
serverErr := server.ListenAndServe()
if !errors.Is(serverErr, http.ErrServerClosed) {
logging.Error().Err(serverErr).Msg("Server shut down unexpectedly")
}
2021-03-09 08:05:07 +00:00
},
}
2021-03-11 05:02:43 +00:00
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)
}
}