diff --git a/backend/backend.go b/backend/backend.go index 65c6e10..f647838 100644 --- a/backend/backend.go +++ b/backend/backend.go @@ -20,7 +20,12 @@ func NewBackend(config *shared.Config, lookup *HostLookup) *Backend { } func (b *Backend) Run() error { - r := gin.Default() + r := gin.New() + r.Use(gin.Recovery()) + + if b.config.Verbose { + r.Use(gin.Logger()) + } r.GET("/dnsapi/lookup/:qname/:qtype", func(c *gin.Context) { request := &Request{ @@ -34,7 +39,10 @@ func (b *Backend) Run() error { "result": []*Response{response}, }) } else { - log.Printf("Error during lookup: %v", err) + if b.config.Verbose { + log.Printf("Error during lookup: %v", err) + } + c.JSON(200, gin.H{ "result": false, }) diff --git a/frontend/frontend.go b/frontend/frontend.go index d868e57..c00af83 100644 --- a/frontend/frontend.go +++ b/frontend/frontend.go @@ -24,7 +24,13 @@ func NewFrontend(config *shared.Config, hosts shared.HostBackend) *Frontend { } func (f *Frontend) Run() error { - r := gin.Default() + r := gin.New() + r.Use(gin.Recovery()) + + if f.config.Verbose { + r.Use(gin.Logger()) + } + r.SetHTMLTemplate(buildTemplate()) r.GET("/", func(g *gin.Context) {