mirror of
https://github.com/openziti/zrok.git
synced 2024-12-22 23:02:52 +01:00
80 lines
2.6 KiB
Go
80 lines
2.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/go-openapi/runtime/middleware"
|
|
"github.com/openziti-test-kitchen/zrok/rest_model_zrok"
|
|
"github.com/openziti-test-kitchen/zrok/rest_server_zrok/operations/metadata"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type environmentDetailHandler struct{}
|
|
|
|
func newEnvironmentDetailHandler() *environmentDetailHandler {
|
|
return &environmentDetailHandler{}
|
|
}
|
|
|
|
func (h *environmentDetailHandler) Handle(params metadata.GetEnvironmentDetailParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
|
tx, err := str.Begin()
|
|
if err != nil {
|
|
logrus.Errorf("error starting transaction: %v", err)
|
|
return metadata.NewGetEnvironmentDetailInternalServerError()
|
|
}
|
|
defer func() { _ = tx.Rollback() }()
|
|
senv, err := str.FindEnvironmentForAccount(params.EnvZID, int(principal.ID), tx)
|
|
if err != nil {
|
|
logrus.Errorf("environment '%v' not found for account '%v': %v", params.EnvZID, principal.Email, err)
|
|
return metadata.NewGetEnvironmentDetailNotFound()
|
|
}
|
|
es := &rest_model_zrok.EnvironmentServices{
|
|
Environment: &rest_model_zrok.Environment{
|
|
Address: senv.Address,
|
|
CreatedAt: senv.CreatedAt.UnixMilli(),
|
|
Description: senv.Description,
|
|
Host: senv.Host,
|
|
UpdatedAt: senv.UpdatedAt.UnixMilli(),
|
|
ZID: senv.ZId,
|
|
},
|
|
}
|
|
svcs, err := str.FindServicesForEnvironment(senv.Id, tx)
|
|
if err != nil {
|
|
logrus.Errorf("error finding services for environment '%v': %v", senv.ZId, err)
|
|
return metadata.NewGetEnvironmentDetailInternalServerError()
|
|
}
|
|
var sparkData map[string][]int64
|
|
if cfg.Influx != nil {
|
|
sparkData, err = sparkDataForServices(svcs)
|
|
if err != nil {
|
|
logrus.Errorf("error querying spark data for services: %v", err)
|
|
return metadata.NewGetEnvironmentDetailInternalServerError()
|
|
}
|
|
}
|
|
for _, svc := range svcs {
|
|
feEndpoint := ""
|
|
if svc.FrontendEndpoint != nil {
|
|
feEndpoint = *svc.FrontendEndpoint
|
|
}
|
|
feSelection := ""
|
|
if svc.FrontendSelection != nil {
|
|
feSelection = *svc.FrontendSelection
|
|
}
|
|
beProxyEndpoint := ""
|
|
if svc.BackendProxyEndpoint != nil {
|
|
beProxyEndpoint = *svc.BackendProxyEndpoint
|
|
}
|
|
es.Services = append(es.Services, &rest_model_zrok.Service{
|
|
Token: svc.Token,
|
|
ZID: svc.ZId,
|
|
ShareMode: svc.ShareMode,
|
|
BackendMode: svc.BackendMode,
|
|
FrontendSelection: feSelection,
|
|
FrontendEndpoint: feEndpoint,
|
|
BackendProxyEndpoint: beProxyEndpoint,
|
|
Reserved: svc.Reserved,
|
|
Metrics: sparkData[svc.Token],
|
|
CreatedAt: svc.CreatedAt.UnixMilli(),
|
|
UpdatedAt: svc.UpdatedAt.UnixMilli(),
|
|
})
|
|
}
|
|
return metadata.NewGetEnvironmentDetailOK().WithPayload(es)
|
|
}
|