99 lines
3.8 KiB
Go
99 lines
3.8 KiB
Go
package api
|
|
|
|
import (
|
|
"github.com/gorilla/mux"
|
|
"github.com/openHPI/poseidon/internal/api/auth"
|
|
"github.com/openHPI/poseidon/internal/config"
|
|
"github.com/openHPI/poseidon/internal/environment"
|
|
"github.com/openHPI/poseidon/internal/runner"
|
|
"github.com/openHPI/poseidon/pkg/dto"
|
|
"github.com/openHPI/poseidon/pkg/logging"
|
|
"github.com/openHPI/poseidon/pkg/monitoring"
|
|
"net/http"
|
|
)
|
|
|
|
var log = logging.GetLogger("api")
|
|
|
|
const (
|
|
BasePath = "/api/v1"
|
|
HealthPath = "/health"
|
|
VersionPath = "/version"
|
|
RunnersPath = "/runners"
|
|
EnvironmentsPath = "/execution-environments"
|
|
StatisticsPath = "/statistics"
|
|
)
|
|
|
|
// NewRouter returns a *mux.Router which can be
|
|
// used by the net/http package to serve the routes of our API. It
|
|
// always returns a router for the newest version of our API. We
|
|
// use gorilla/mux because it is more convenient than net/http, e.g.
|
|
// when extracting path parameters.
|
|
func NewRouter(runnerManager runner.Manager, environmentManager environment.ManagerHandler) *mux.Router {
|
|
router := mux.NewRouter()
|
|
// this can later be restricted to a specific host with
|
|
// `router.Host(...)` and to HTTPS with `router.Schemes("https")`
|
|
configureV1Router(router, runnerManager, environmentManager)
|
|
router.Use(logging.HTTPLoggingMiddleware)
|
|
router.Use(monitoring.InfluxDB2Middleware)
|
|
return router
|
|
}
|
|
|
|
// configureV1Router configures a given router with the routes of version 1 of the Poseidon API.
|
|
func configureV1Router(router *mux.Router,
|
|
runnerManager runner.Manager, environmentManager environment.ManagerHandler) {
|
|
router.NotFoundHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
log.WithContext(r.Context()).WithField("request", r).Debug("Not Found Handler")
|
|
w.WriteHeader(http.StatusNotFound)
|
|
})
|
|
v1 := router.PathPrefix(BasePath).Subrouter()
|
|
v1.HandleFunc(HealthPath, Health(environmentManager)).Methods(http.MethodGet).Name(HealthPath)
|
|
v1.HandleFunc(VersionPath, Version).Methods(http.MethodGet).Name(VersionPath)
|
|
|
|
runnerController := &RunnerController{manager: runnerManager}
|
|
environmentController := &EnvironmentController{manager: environmentManager}
|
|
configureRoutes := func(router *mux.Router) {
|
|
runnerController.ConfigureRoutes(router)
|
|
environmentController.ConfigureRoutes(router)
|
|
|
|
// May add a statistics controller if another route joins
|
|
statisticsRouter := router.PathPrefix(StatisticsPath).Subrouter()
|
|
statisticsRouter.
|
|
HandleFunc(EnvironmentsPath, StatisticsExecutionEnvironments(environmentManager)).
|
|
Methods(http.MethodGet).Name(EnvironmentsPath)
|
|
}
|
|
|
|
if auth.InitializeAuthentication() {
|
|
// Create new authenticated subrouter.
|
|
// All routes added to v1 after this require authentication.
|
|
authenticatedV1Router := v1.PathPrefix("").Subrouter()
|
|
authenticatedV1Router.Use(auth.HTTPAuthenticationMiddleware)
|
|
configureRoutes(authenticatedV1Router)
|
|
} else {
|
|
configureRoutes(v1)
|
|
}
|
|
}
|
|
|
|
// Version handles the version route.
|
|
// It responds the release information stored in the configuration.
|
|
func Version(writer http.ResponseWriter, request *http.Request) {
|
|
release := config.Config.Sentry.Release
|
|
if release != "" {
|
|
sendJSON(writer, release, http.StatusOK, request.Context())
|
|
} else {
|
|
writer.WriteHeader(http.StatusNotFound)
|
|
}
|
|
}
|
|
|
|
// StatisticsExecutionEnvironments handles the route for statistics about execution environments.
|
|
// It responds the prewarming pool size and the number of idle runners and used runners.
|
|
func StatisticsExecutionEnvironments(manager environment.Manager) http.HandlerFunc {
|
|
return func(writer http.ResponseWriter, request *http.Request) {
|
|
result := make(map[string]*dto.StatisticalExecutionEnvironmentData)
|
|
environmentsData := manager.Statistics()
|
|
for id, data := range environmentsData {
|
|
result[id.ToString()] = data
|
|
}
|
|
sendJSON(writer, result, http.StatusOK, request.Context())
|
|
}
|
|
}
|