mirror of
https://github.com/openziti/zrok.git
synced 2026-08-24 10:14:56 -05:00
78 lines
2.9 KiB
Go
78 lines
2.9 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/go-openapi/runtime/middleware"
|
|
"github.com/michaelquigley/df/dl"
|
|
"github.com/openziti/zrok/v2/rest_model_zrok"
|
|
"github.com/openziti/zrok/v2/rest_server_zrok/operations/metadata"
|
|
)
|
|
|
|
type environmentDetailHandler struct{}
|
|
|
|
func newEnvironmentDetailHandler() *environmentDetailHandler {
|
|
return &environmentDetailHandler{}
|
|
}
|
|
|
|
func (h *environmentDetailHandler) Handle(params metadata.GetEnvironmentDetailParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
|
trx, err := str.Begin()
|
|
if err != nil {
|
|
dl.Errorf("error starting transaction for user '%v': %v", principal.Email, err)
|
|
return metadata.NewGetEnvironmentDetailInternalServerError()
|
|
}
|
|
defer func() { _ = trx.Rollback() }()
|
|
senv, err := str.FindEnvironmentForAccount(params.EnvZID, int(principal.ID), trx)
|
|
if err != nil {
|
|
dl.Errorf("environment '%v' not found for account '%v': %v", params.EnvZID, principal.Email, err)
|
|
return metadata.NewGetEnvironmentDetailNotFound()
|
|
}
|
|
es := &rest_model_zrok.EnvironmentAndResources{
|
|
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,
|
|
},
|
|
}
|
|
shrs, err := str.FindSharesForEnvironment(senv.Id, trx)
|
|
if err != nil {
|
|
dl.Errorf("error finding shares for environment '%v' for user '%v': %v", senv.ZId, principal.Email, err)
|
|
return metadata.NewGetEnvironmentDetailInternalServerError()
|
|
}
|
|
sparkRx := make(map[string][]int64)
|
|
sparkTx := make(map[string][]int64)
|
|
if cfg.Metrics != nil && cfg.Metrics.Influx != nil {
|
|
sparkRx, sparkTx, err = sparkDataForShares(shrs)
|
|
if err != nil {
|
|
dl.Errorf("error querying spark data for shares for user '%v': %v", principal.Email, err)
|
|
}
|
|
} else {
|
|
dl.Debug("skipping spark data for shares; no influx configuration")
|
|
}
|
|
for _, shr := range shrs {
|
|
frontendEndpoints := buildFrontendEndpointsForShare(shr.Id, shr.Token, shr.FrontendEndpoint, trx)
|
|
target := ""
|
|
if shr.BackendProxyEndpoint != nil {
|
|
target = *shr.BackendProxyEndpoint
|
|
}
|
|
var sparkData []*rest_model_zrok.SparkDataSample
|
|
for i := 0; i < len(sparkRx[shr.Token]) && i < len(sparkTx[shr.Token]); i++ {
|
|
sparkData = append(sparkData, &rest_model_zrok.SparkDataSample{Rx: float64(sparkRx[shr.Token][i]), Tx: float64(sparkTx[shr.Token][i])})
|
|
}
|
|
es.Shares = append(es.Shares, &rest_model_zrok.Share{
|
|
ShareToken: shr.Token,
|
|
ZID: shr.ZId,
|
|
EnvZID: senv.ZId,
|
|
ShareMode: shr.ShareMode,
|
|
BackendMode: shr.BackendMode,
|
|
FrontendEndpoints: frontendEndpoints,
|
|
Target: target,
|
|
Activity: sparkData,
|
|
CreatedAt: shr.CreatedAt.UnixMilli(),
|
|
UpdatedAt: shr.UpdatedAt.UnixMilli(),
|
|
})
|
|
}
|
|
return metadata.NewGetEnvironmentDetailOK().WithPayload(es)
|
|
}
|