in_cluster.go

package impl

import (
	"fmt"

	"github.com/lyft/flyteadmin/pkg/executioncluster"
	"github.com/lyft/flyteadmin/pkg/executioncluster/interfaces"
	"github.com/lyft/flyteadmin/pkg/flytek8s"
	"github.com/lyft/flytestdlib/promutils"
	"github.com/pkg/errors"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

type InCluster struct {
	target executioncluster.ExecutionTarget
}

func (i InCluster) GetTarget(spec *executioncluster.ExecutionTargetSpec) (*executioncluster.ExecutionTarget, error) {
	if spec != nil && spec.TargetID != "" {
		return nil, errors.New(fmt.Sprintf("remote target %s is not supported", spec.TargetID))
	}
	return &i.target, nil
}

func (i InCluster) GetAllValidTargets() []executioncluster.ExecutionTarget {
	return []executioncluster.ExecutionTarget{
		i.target,
	}
}

func NewInCluster(scope promutils.Scope, kubeConfig, master string) (interfaces.ClusterInterface, error) {
	clientConfig, err := flytek8s.GetRestClientConfig(kubeConfig, master, nil)
	if err != nil {
		return nil, err
	}
	flyteClient, err := getRestClientFromKubeConfig(scope, clientConfig)
	if err != nil {
		return nil, err
	}
	client, err := client.New(clientConfig, client.Options{})
	if err != nil {
		return nil, err
	}
	return &InCluster{
		target: executioncluster.ExecutionTarget{
			Client:      client,
			FlyteClient: flyteClient,
		},
	}, nil
}