115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package process
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/labstack/echo"
|
|
"github.com/labstack/echo/middleware"
|
|
gl "github.com/labstack/gommon/log"
|
|
ep2 "github.com/mayowa/echo-pongo2"
|
|
"github.com/ziflex/lecho"
|
|
|
|
//"github.com/gin-gonic/gin"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// Server is the HTTP webserver object
|
|
type Server struct {
|
|
feta *Feta
|
|
port uint
|
|
e *echo.Echo
|
|
router *gin.Engine
|
|
httpserver *http.Server
|
|
debug bool
|
|
db *gorm.DB
|
|
}
|
|
|
|
// SetFeta tells the http Server where to find the Process object so that it
|
|
// can pull stats and other information for serving via http
|
|
func (a *Server) SetFeta(feta *Feta) {
|
|
a.feta = feta
|
|
}
|
|
|
|
// Serve is the entrypoint for the Server, which should run in its own
|
|
// goroutine (started by the Process)
|
|
func (a *Server) Serve() {
|
|
|
|
if a.feta == nil {
|
|
panic("must have feta app from which to serve stats")
|
|
}
|
|
|
|
if os.Getenv("DEBUG") != "" {
|
|
a.debug = true
|
|
}
|
|
|
|
a.port = 8080
|
|
|
|
if os.Getenv("PORT") != "" {
|
|
s, err := strconv.ParseUint(os.Getenv("PORT"), 10, 64)
|
|
if err != nil {
|
|
a.port = uint(s)
|
|
}
|
|
}
|
|
|
|
a.initRouter()
|
|
a.initServer()
|
|
a.e.Logger.Fatal(a.e.StartServer(a.httpserver))
|
|
}
|
|
|
|
func (s *Server) initRouter() {
|
|
|
|
// Echo instance
|
|
s.e = echo.New()
|
|
|
|
s.e.HideBanner = true
|
|
|
|
lev := gl.INFO
|
|
if os.Getenv("DEBUG") != "" {
|
|
lev = gl.DEBUG
|
|
}
|
|
|
|
logger := lecho.New(
|
|
os.Stdout,
|
|
lecho.WithLevel(lev),
|
|
lecho.WithTimestamp(),
|
|
lecho.WithCaller(),
|
|
)
|
|
s.e.Logger = logger
|
|
s.e.Use(middleware.RequestID())
|
|
|
|
// Middleware
|
|
s.e.Use(middleware.Logger())
|
|
s.e.Use(middleware.Recover())
|
|
|
|
r, err := ep2.NewRenderer("view")
|
|
if err != nil {
|
|
s.e.Logger.Fatal(err)
|
|
}
|
|
s.e.Renderer = r
|
|
|
|
// Routes
|
|
s.e.GET("/", s.indexHandler)
|
|
s.e.GET("/instance/:uuid", s.instanceHandler)
|
|
s.e.GET("/stats.json", s.statsHandler)
|
|
s.e.GET("/.well-known/healthcheck.json", s.healthCheckHandler)
|
|
//a.e.GET("/about", s.aboutHandler)
|
|
|
|
}
|
|
|
|
func (s *Server) initServer() {
|
|
log.Info().Uint("port", s.port).Msg("starting webserver")
|
|
|
|
s.httpserver = &http.Server{
|
|
Addr: fmt.Sprintf(":%d", s.port),
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
MaxHeaderBytes: 1 << 20,
|
|
}
|
|
}
|