82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/jritsema/gotoolbox"
|
|
"github.com/jritsema/gotoolbox/web"
|
|
"github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var (
|
|
//go:embed all:templates/*
|
|
templateFS embed.FS
|
|
|
|
//go:embed css/output.css
|
|
css embed.FS
|
|
|
|
//parsed templates
|
|
html *template.Template
|
|
)
|
|
|
|
func main() {
|
|
|
|
//exit process immediately upon sigterm
|
|
handleSigTerms()
|
|
|
|
//parse templates
|
|
var err error
|
|
html, err = web.TemplateParseFSRecursive(templateFS, ".html", true, nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
//add routes
|
|
router := http.NewServeMux()
|
|
// router.Handle("/css/output.css", http.FileServer(http.FS(css)))
|
|
|
|
router.Handle("/company/add", web.Action(companyAdd))
|
|
router.Handle("/company/add/", web.Action(companyAdd))
|
|
|
|
router.Handle("/company/edit", web.Action(companyEdit))
|
|
router.Handle("/company/edit/", web.Action(companyEdit))
|
|
|
|
router.Handle("/company", web.Action(companies))
|
|
router.Handle("/company/", web.Action(companies))
|
|
|
|
router.Handle("/", web.Action(index))
|
|
router.Handle("/index.html", web.Action(index))
|
|
|
|
//logging/tracing
|
|
nextRequestID := func() string {
|
|
return fmt.Sprintf("%d", time.Now().UnixNano())
|
|
}
|
|
logger := log.New(os.Stdout, "http: ", log.LstdFlags)
|
|
middleware := tracing(nextRequestID)(logging(logger)(router))
|
|
|
|
port := gotoolbox.GetEnvWithDefault("PORT", "8080")
|
|
logger.Println("listening on http://localhost:" + port)
|
|
if err := http.ListenAndServe(":"+port, middleware); err != nil {
|
|
logger.Println("http.ListenAndServe():", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func handleSigTerms() {
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
go func() {
|
|
<-c
|
|
fmt.Println("received SIGTERM, exiting")
|
|
os.Exit(1)
|
|
}()
|
|
}
|