mirror of
https://github.com/openziti/zrok.git
synced 2025-08-04 06:20:50 +02:00
66 lines
2.4 KiB
Go
66 lines
2.4 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-openapi/runtime/middleware"
|
|
"github.com/openziti/zrok/agent/agentGrpc"
|
|
"github.com/openziti/zrok/controller/agentController"
|
|
"github.com/openziti/zrok/rest_model_zrok"
|
|
"github.com/openziti/zrok/rest_server_zrok/operations/agent"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type agentShareHttpHealthcheckHandler struct{}
|
|
|
|
func newAgentShareHttpHealthcheckHandler() *agentShareHttpHealthcheckHandler {
|
|
return &agentShareHttpHealthcheckHandler{}
|
|
}
|
|
|
|
func (h *agentShareHttpHealthcheckHandler) Handle(params agent.ShareHTTPHealthcheckParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
|
trx, err := str.Begin()
|
|
if err != nil {
|
|
logrus.Errorf("error starting transaction for '%v': %v", principal.Email, err)
|
|
return agent.NewShareHTTPHealthcheckInternalServerError()
|
|
}
|
|
defer trx.Rollback()
|
|
|
|
env, err := str.FindEnvironmentForAccount(params.Body.EnvZID, int(principal.ID), trx)
|
|
if err != nil {
|
|
logrus.Errorf("error finding environment '%v' for '%v': %v", params.Body.EnvZID, principal.Email, err)
|
|
return agent.NewShareHTTPHealthcheckUnauthorized()
|
|
}
|
|
|
|
ae, err := str.FindAgentEnrollmentForEnvironment(env.Id, trx)
|
|
if err != nil {
|
|
logrus.Errorf("error finding agent enrollment for environment '%v' (%v): %v", params.Body.EnvZID, principal.Email, err)
|
|
return agent.NewShareHTTPHealthcheckBadGateway()
|
|
}
|
|
_ = trx.Rollback() // ...or will block share trx on sqlite
|
|
|
|
acli, aconn, err := agentController.NewAgentClient(ae.Token, cfg.AgentController)
|
|
if err != nil {
|
|
logrus.Errorf("error creating agent client for '%v' (%v): %v", params.Body.EnvZID, principal.Email, err)
|
|
return agent.NewShareHTTPHealthcheckInternalServerError()
|
|
}
|
|
defer aconn.Close()
|
|
|
|
req := &agentGrpc.ShareHttpHealthcheckRequest{
|
|
Token: params.Body.ShareToken,
|
|
HttpVerb: params.Body.HTTPVerb,
|
|
Endpoint: params.Body.Endpoint,
|
|
ExpectedHttpResponse: uint32(params.Body.ExpectedHTTPResponse),
|
|
TimeoutMs: uint64(params.Body.TimeoutMs),
|
|
}
|
|
resp, err := acli.ShareHttpHealthcheck(context.Background(), req)
|
|
if err != nil {
|
|
logrus.Infof("error invoking remoted share '%v' http healthcheck for '%v': %v", params.Body.ShareToken, params.Body.EnvZID, err)
|
|
return agent.NewShareHTTPHealthcheckBadGateway()
|
|
}
|
|
|
|
return agent.NewShareHTTPHealthcheckOK().WithPayload(&agent.ShareHTTPHealthcheckOKBody{
|
|
Healthy: resp.Healthy,
|
|
Error: resp.Error,
|
|
})
|
|
}
|