mirror of https://github.com/linkerd/linkerd2.git
100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package srv
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/http"
|
|
"regexp"
|
|
|
|
"github.com/julienschmidt/httprouter"
|
|
"github.com/linkerd/linkerd2/pkg/k8s"
|
|
profiles "github.com/linkerd/linkerd2/pkg/profiles"
|
|
vizPb "github.com/linkerd/linkerd2/viz/metrics-api/gen/viz"
|
|
"github.com/patrickmn/go-cache"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var proxyPathRegexp = regexp.MustCompile("/api/v1/namespaces/.*/proxy/")
|
|
|
|
type (
|
|
renderTemplate func(http.ResponseWriter, string, string, interface{}) error
|
|
|
|
handler struct {
|
|
render renderTemplate
|
|
apiClient vizPb.ApiClient
|
|
k8sAPI *k8s.KubernetesAPI
|
|
uuid string
|
|
version string
|
|
controllerNamespace string
|
|
clusterDomain string
|
|
grafana string
|
|
grafanaExternalURL string
|
|
grafanaPrefix string
|
|
jaeger string
|
|
grafanaProxy *reverseProxy
|
|
jaegerProxy *reverseProxy
|
|
hc healthChecker
|
|
statCache *cache.Cache
|
|
}
|
|
)
|
|
|
|
func (h *handler) handleIndex(w http.ResponseWriter, req *http.Request, p httprouter.Params) {
|
|
// when running the dashboard via `linkerd dashboard`, serve the index bundle at the right path
|
|
pathPfx := proxyPathRegexp.FindString(req.URL.Path)
|
|
if pathPfx == "" {
|
|
pathPfx = "/"
|
|
}
|
|
|
|
params := appParams{
|
|
UUID: h.uuid,
|
|
ReleaseVersion: h.version,
|
|
ControllerNamespace: h.controllerNamespace,
|
|
PathPrefix: pathPfx,
|
|
Grafana: h.grafana,
|
|
GrafanaExternalURL: h.grafanaExternalURL,
|
|
GrafanaPrefix: h.grafanaPrefix,
|
|
Jaeger: h.jaeger,
|
|
}
|
|
|
|
err := h.render(w, "app.tmpl.html", "base", params)
|
|
if err != nil {
|
|
log.Error(err)
|
|
}
|
|
}
|
|
|
|
func (h *handler) handleProfileDownload(w http.ResponseWriter, req *http.Request, params httprouter.Params) {
|
|
service := req.FormValue("service")
|
|
namespace := req.FormValue("namespace")
|
|
|
|
if service == "" || namespace == "" {
|
|
err := fmt.Errorf("Service and namespace must be provided to create a new profile")
|
|
log.Error(err)
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
profileYaml := &bytes.Buffer{}
|
|
err := profiles.RenderProfileTemplate(namespace, service, h.clusterDomain, profileYaml)
|
|
|
|
if err != nil {
|
|
log.Error(err)
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
dispositionHeaderVal := fmt.Sprintf("attachment; filename=%s-profile.yml", service)
|
|
|
|
w.Header().Set("Content-Type", "text/yaml")
|
|
w.Header().Set("Content-Disposition", dispositionHeaderVal)
|
|
|
|
w.Write(profileYaml.Bytes())
|
|
}
|
|
|
|
func (h *handler) handleGrafana(w http.ResponseWriter, req *http.Request, p httprouter.Params) {
|
|
h.grafanaProxy.ServeHTTP(w, req)
|
|
}
|
|
|
|
func (h *handler) handleJaeger(w http.ResponseWriter, req *http.Request, p httprouter.Params) {
|
|
h.jaegerProxy.ServeHTTP(w, req)
|
|
}
|