massive services -> share boatload (#144)

controller/shareDetail.go

@@ -0,0 +1,78 @@
+package controller
+
+import (
+	"github.com/go-openapi/runtime/middleware"
+	"github.com/openziti-test-kitchen/zrok/controller/store"
+	"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 shareDetailHandler struct{}
+
+func newShareDetailHandler() *shareDetailHandler {
+	return &shareDetailHandler{}
+}
+
+func (h *shareDetailHandler) Handle(params metadata.GetShareDetailParams, principal *rest_model_zrok.Principal) middleware.Responder {
+	tx, err := str.Begin()
+	if err != nil {
+		logrus.Errorf("error starting transaction: %v", err)
+		return metadata.NewGetShareDetailInternalServerError()
+	}
+	defer func() { _ = tx.Rollback() }()
+	shr, err := str.FindShareWithToken(params.ShrToken, tx)
+	if err != nil {
+		logrus.Errorf("error finding share '%v': %v", params.ShrToken, err)
+		return metadata.NewGetShareDetailNotFound()
+	}
+	envs, err := str.FindEnvironmentsForAccount(int(principal.ID), tx)
+	if err != nil {
+		logrus.Errorf("error finding environments for account '%v': %v", principal.Email, err)
+		return metadata.NewGetShareDetailInternalServerError()
+	}
+	found := false
+	for _, env := range envs {
+		if shr.EnvironmentId == env.Id {
+			found = true
+			break
+		}
+	}
+	if !found {
+		logrus.Errorf("environment not matched for service '%v' for account '%v'", params.ShrToken, principal.Email)
+		return metadata.NewGetShareDetailNotFound()
+	}
+	var sparkData map[string][]int64
+	if cfg.Influx != nil {
+		sparkData, err = sparkDataForServices([]*store.Share{shr})
+		if err != nil {
+			logrus.Errorf("error querying spark data for services: %v", err)
+			return metadata.NewGetEnvironmentDetailInternalServerError()
+		}
+	}
+	feEndpoint := ""
+	if shr.FrontendEndpoint != nil {
+		feEndpoint = *shr.FrontendEndpoint
+	}
+	feSelection := ""
+	if shr.FrontendSelection != nil {
+		feSelection = *shr.FrontendSelection
+	}
+	beProxyEndpoint := ""
+	if shr.BackendProxyEndpoint != nil {
+		beProxyEndpoint = *shr.BackendProxyEndpoint
+	}
+	return metadata.NewGetShareDetailOK().WithPayload(&rest_model_zrok.Share{
+		Token:                shr.Token,
+		ZID:                  shr.ZId,
+		ShareMode:            shr.ShareMode,
+		BackendMode:          shr.BackendMode,
+		FrontendSelection:    feSelection,
+		FrontendEndpoint:     feEndpoint,
+		BackendProxyEndpoint: beProxyEndpoint,
+		Reserved:             shr.Reserved,
+		Metrics:              sparkData[shr.Token],
+		CreatedAt:            shr.CreatedAt.UnixMilli(),
+		UpdatedAt:            shr.UpdatedAt.UnixMilli(),
+	})
+}