topologyyaml

type BESpec struct {
	ComponentSpec  `yaml:",inline"`
	BeConfig       BeConfig      `yaml:"-"`
	InstallJava    bool          `yaml:"install_java"`
	JavaPackageURL string        `yaml:"java_package_url"`
	JavaDigest     digest.Digest `yaml:"java_digest"`
	PackageURL     string        `yaml:"package_url"`
	Digest         digest.Digest `yaml:"digest"`
}

type BeConfig struct {
	BePort               int    `ini:"be_port"`
	WebServerPort        int    `ini:"webserver_port"`
	HeartbeatServicePort int    `ini:"heartbeat_service_port"`
	StorageRootPath      string `ini:"storage_root_path"`
}

type BeInstance struct {
	DeployUser string
	Topo       *Topology
	BESpec
}

type BeStatus struct {
	Host          string `db:"Host"`
	IP            string `db:"IP"` // doris v1.x use IP, v2.x use Host
	Alive         bool   `db:"Alive"`
	AvailCapacity string `db:"AvailCapacity"`
	TotalCapacity string `db:"TotalCapacity"`
	// UsedUsedPct is the percentage of used capacity
	UsedPct string `db:"UsedPct"`
	Version string `db:"Version"`
}

type ClusterStatus struct {
	FeMasterHealthy bool
	Fes             []FeStatus
	Bes             []BeStatus
}

type ComponentSpec struct {
	Host      string `yaml:"host"`
	SSHPort   int    `yaml:"ssh_port"`
	DeployDir string `yaml:"deploy_dir"`
	Config    string `yaml:"config"`
}

type ComponentType string

type ExecOutput struct {
	Stdout   string
	Stderr   string
	ExitCode int
}

type FESpec struct {
	ComponentSpec  `yaml:",inline"`
	FeConfig       FeConfig      `yaml:"-"`
	IsMaster       bool          `yaml:"is_master"`
	InstallJava    bool          `yaml:"install_java"`
	JavaPackageURL string        `yaml:"java_package_url"`
	JavaDigest     digest.Digest `yaml:"java_digest"`
	PackageURL     string        `yaml:"package_url"`
	Digest         digest.Digest `yaml:"digest"`
}

type FeConfig struct {
	HTTPPort    int `ini:"http_port"`
	RPCPort     int `ini:"rpc_port"`
	QueryPort   int `ini:"query_port"`
	EditLogPort int `ini:"edit_log_port"`
}

type FeInstance struct {
	DeployUser string
	Topo       *Topology
	FESpec
}

type FeStatus struct {
	Host      string `db:"Host"`
	IP        string `db:"IP"` // doris v1.x use IP, v2.x use Host
	IsMaster  bool   `db:"IsMaster"`
	IsHelper  bool   `db:"IsHelper"` // starrocks use "IsHelper"
	QueryPort int    `db:"QueryPort"`
	Alive     bool   `db:"Alive"`
	Version   string `db:"Version"`
}

type GlobalSpec struct {
	DeployUser        string `yaml:"deploy_user"`
	SSHPrivateKeyPath string `yaml:"ssh_private_key_path"`
	UseSystemd        *bool  `yaml:"use_systemd"`
}

type Instance interface {
	EnvironmentVars() map[string]string
	PIDFile() string
	StartupScript() string
	StopScript() string
	SystemdEnvironment() string
	SystemdServiceContent() (string, error)
	SystemdServicePath() string
	SystemdServiceName() string
}

type ManualDeployInfo struct {
	DeployUser string
	SSHPort    int
	SSHKeyPath string

	FeMaster    string
	FeHosts     []string
	BeHosts     []string
	FeDeployDir string
	BeDeployDir string
}

type Topology struct {
	Global GlobalSpec `yaml:"global"`
	// TODO: maybe some common config for all fe/bes, to shorten the topology yaml
	FEs []FESpec `yaml:"fes"`
	BEs []BESpec `yaml:"bes"`
}