linkerd2/jaeger/cmd/install.go

141 lines
3.1 KiB
Go

package cmd
import (
"context"
"fmt"
"io"
"os"
"github.com/linkerd/linkerd2/jaeger/static"
jaeger "github.com/linkerd/linkerd2/jaeger/values"
"github.com/linkerd/linkerd2/pkg/charts"
"github.com/linkerd/linkerd2/pkg/healthcheck"
"github.com/linkerd/linkerd2/pkg/k8s"
"github.com/spf13/cobra"
"helm.sh/helm/v3/pkg/chart/loader"
"helm.sh/helm/v3/pkg/chartutil"
kerrors "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"sigs.k8s.io/yaml"
)
var (
templatesJaeger = []string{
"templates/namespace.yaml",
"templates/proxy-mutator.yaml",
"templates/rbac.yaml",
"templates/tracing.yaml",
}
)
func newCmdInstall() *cobra.Command {
var skipChecks bool
values, err := jaeger.NewValues()
if err != nil {
fmt.Fprint(os.Stderr, err.Error())
os.Exit(1)
}
cmd := &cobra.Command{
Use: "install [flags]",
Args: cobra.NoArgs,
Short: "Output Kubernetes resources to install jaeger extension",
Long: `Output Kubernetes resources to install jaeger extension.`,
Example: ` # Default install.
linkerd jaeger install | kubectl apply -f -
# Install Jaeger extension into a non-default namespace.
linkerd jaeger install --namespace custom | kubectl apply -f -`,
RunE: func(cmd *cobra.Command, args []string) error {
if !skipChecks {
// Ensure there is a Linkerd installation.
exists, err := checkIfLinkerdExists(cmd.Context())
if err != nil {
return fmt.Errorf("could not check for Linkerd existence: %s", err)
}
if !exists {
return fmt.Errorf("could not find a Linkerd installation")
}
}
return install(os.Stdout, values)
},
}
cmd.Flags().BoolVar(
&skipChecks, "skip-checks", false,
`Skip checks for namespace existence`,
)
// TODO: Add --set flag set and also config
return cmd
}
func install(w io.Writer, values *jaeger.Values) error {
// TODO: Add any validation logic here
return render(w, values)
}
func render(w io.Writer, values *jaeger.Values) error {
// Render raw values and create chart config
rawValues, err := yaml.Marshal(values)
if err != nil {
return err
}
files := []*loader.BufferedFile{
{Name: chartutil.ChartfileName},
}
for _, template := range templatesJaeger {
files = append(files,
&loader.BufferedFile{Name: template},
)
}
chart := &charts.Chart{
Name: "jaeger",
Dir: "jaeger",
Namespace: values.Namespace,
RawValues: rawValues,
Files: files,
Fs: static.Templates,
}
buf, err := chart.Render()
if err != nil {
return err
}
_, err = w.Write(buf.Bytes())
return err
}
func checkIfLinkerdExists(ctx context.Context) (bool, error) {
kubeAPI, err := k8s.NewAPI(kubeconfigPath, kubeContext, impersonate, impersonateGroup, 0)
if err != nil {
return false, err
}
_, err = kubeAPI.CoreV1().Namespaces().Get(ctx, controlPlaneNamespace, metav1.GetOptions{})
if err != nil {
if kerrors.IsNotFound(err) {
return false, nil
}
return false, err
}
_, _, err = healthcheck.FetchCurrentConfiguration(ctx, kubeAPI, controlPlaneNamespace)
if err != nil {
if kerrors.IsNotFound(err) {
return false, nil
}
return false, err
}
return true, nil
}