cluster

type Client struct {
	Namespace    string
	ClusterName  string
	PDMemberFunc func(ns, name string) (string, []string, error)
}

type ClientNode struct {
	Namespace   string // Cluster k8s' namespace
	ClusterName string // Cluster name, use to differentiate different TiDB clusters running on same namespace
	Component   Component
	IP          string
	Port        int32
}

type Cluster interface {
	Apply() error
	Delete() error
	GetNodes() ([]Node, error)
	GetClientNodes() ([]ClientNode, error)
}

type Component string

type K8sProvider struct {
	*tests.TestCli
}

type LocalClusterProvider struct {
	DBs []string
	PDs []string
	KVs []string
}

type Node struct {
	Namespace string    // Cluster k8s' namespace
	Component Component // Node component type
	PodName   string    // Pod's name
	IP        string
	Port      int32
	*Client   `json:"-"`
}

type Provider interface {
	// SetUp sets up cluster, returns err or all nodes info
	SetUp(ctx context.Context, spec Specs) ([]Node, []ClientNode, error)
	// TearDown tears down the cluster
	TearDown(ctx context.Context, spec Specs) error
}

type Specs struct {
	Cluster     Cluster
	NemesisGens []string
	Namespace   string
}