learn-go/snippetbox/cmd/web/main.go
tamsin johnson 3deb2022b8 lets-go:5.4
2024-01-25 15:06:28 -08:00

78 lines
1.7 KiB
Go

package main
import (
"database/sql"
"flag"
"html/template"
"log/slog"
"net/http"
"regexp"
"os"
"snippetbox.chaosfem.tw/internal/models"
_ "github.com/go-sql-driver/mysql"
)
// for application wide dependencies
type application struct {
logger *slog.Logger
snippets *models.SnippetModel
templateCache map[string]*template.Template
}
// main it's the snippetbox webapp
func main() {
// configuration
addr := flag.String("addr", ":4000", "HTTP network address")
dsn := flag.String("dsn", "web:dbpass@/snippetbox?parseTime=true", "DB data source name")
logfmt := flag.String("logfmt", "text", "Log output format")
loglevel := flag.String("loglevel", "INFO", "Log level: DEBUG, INFO, WARN, or ERROR")
flag.Parse()
logger := loggerBuilder(logfmt, loglevel)
db, err := openDB(*dsn, logger)
if err != nil {
logger.Error(err.Error())
os.Exit(1)
}
defer db.Close()
templateCache, err := newTemplateCache()
if err != nil {
logger.Error(err.Error())
os.Exit(1)
}
// setup the application
app := &application{
logger: logger,
snippets: &models.SnippetModel{DB: db},
templateCache: templateCache,
}
logger.Info("starting server", slog.String("addr", *addr))
err = http.ListenAndServe(*addr, app.routes())
logger.Error(err.Error())
os.Exit(1)
}
func openDB(dsn string, logger *slog.Logger) (*sql.DB, error) {
db, err := sql.Open("mysql", dsn)
if err != nil {
return nil, err
}
err = db.Ping()
if err != nil {
db.Close()
return nil, err
}
passRegexp := regexp.MustCompile(":\\S+@")
logger.Info("Opened DB connection pool", slog.String("adapter", "mysql"), slog.String("dsn", passRegexp.ReplaceAllString(dsn, ":<REDACTED>@")))
return db, nil
}