proto

type Action struct {
	Name           string       `json:"name"`
	Exec           *ExecAction  `json:"exec,omitempty"`
	HTTP           *HTTPAction  `json:"http,omitempty"`
	GRPC           *GRPCAction  `json:"grpc,omitempty"`
	TimeoutSeconds int32        `json:"timeoutSeconds,omitempty"`
	RetryPolicy    *RetryPolicy `json:"retryPolicy,omitempty"`
}

type ActionRequest struct {
	Action         string            `json:"action"`
	Parameters     map[string]string `json:"parameters,omitempty"`
	NonBlocking    *bool             `json:"nonBlocking,omitempty"`
	TimeoutSeconds *int32            `json:"timeoutSeconds,omitempty"` // TODO: not implemented
	RetryPolicy    *RetryPolicy      `json:"retryPolicy,omitempty"`    // TODO: not implemented
}

type ActionResponse struct {
	Error   string `json:"error,omitempty"`
	Message string `json:"message,omitempty"`
	Output  []byte `json:"output,omitempty"`
}

type ExecAction struct {
	Commands []string `json:"command,omitempty"`
	Args     []string `json:"args,omitempty"`
}

type GRPCAction struct {
	Port     string            `json:"port"`
	Host     string            `json:"host,omitempty"`
	Service  string            `json:"service"`
	Method   string            `json:"method"`
	Request  map[string]string `json:"request,omitempty"`
	Response GRPCResponse      `json:"response,omitempty"`
}

type GRPCResponse struct {
	Status  string `json:"status,omitempty"`
	Message string `json:"message,omitempty"`
}

type HTTPAction struct {
	Port    string       `json:"port"`
	Host    string       `json:"host,omitempty"`
	Scheme  string       `json:"scheme,omitempty"`
	Path    string       `json:"path,omitempty"`
	Method  string       `json:"method,omitempty"`
	Body    string       `json:"body,omitempty"`
	Headers []HTTPHeader `json:"headers,omitempty"`
}

type HTTPHeader struct {
	Name  string `json:"name"`
	Value string `json:"value"`
}

type NewReplicaTask struct {
	Remote         string            `json:"remote"` // the remote address of the data source
	Port           int32             `json:"port"`
	Replicas       string            `json:"replicas"`                 // replicas to load the data
	Parameters     map[string]string `json:"parameters,omitempty"`     // parameters for data dump and load
	TimeoutSeconds *int32            `json:"timeoutSeconds,omitempty"` // TODO: not implemented
}

type Probe struct {
	Instance            string `json:"instance"`
	Action              string `json:"action"`
	InitialDelaySeconds int32  `json:"initialDelaySeconds,omitempty"`
	PeriodSeconds       int32  `json:"periodSeconds,omitempty"`
	SuccessThreshold    int32  `json:"successThreshold,omitempty"`
	FailureThreshold    int32  `json:"failureThreshold,omitempty"`
	ReportPeriodSeconds int32  `json:"reportPeriodSeconds,omitempty"`
}

type ProbeEvent struct {
	Instance string `json:"instance"`
	Probe    string `json:"probe"`
	Code     int32  `json:"code"`
	Output   []byte `json:"output,omitempty"`  // output of the probe on success, or latest succeed output on failure
	Message  string `json:"message,omitempty"` // message of the probe on failure
}

type RetryPolicy struct {
	MaxRetries    int           `json:"maxRetries,omitempty"`
	RetryInterval time.Duration `json:"retryInterval,omitempty"`
}

type Service struct {
	Kind    string
	Version string
	URI     string
}

type Task struct {
	Instance            string          `json:"instance"`
	Task                string          `json:"task"`
	UID                 string          `json:"UID"`                           // the unique identifier of the task
	Replicas            string          `json:"replicas"`                      // target replicas to run the task
	NotifyAtFinish      bool            `json:"notifyAtFinish,omitempty"`      // whether to notify the controller when the task is finished
	ReportPeriodSeconds int32           `json:"reportPeriodSeconds,omitempty"` // the period to report the progress of the task
	NewReplica          *NewReplicaTask `json:"newReplica,omitempty"`
}

type TaskEvent struct {
	Instance  string    `json:"instance"`
	Task      string    `json:"task"`
	UID       string    `json:"UID"`
	Replica   string    `json:"replica"`
	StartTime time.Time `json:"startTime"`
	EndTime   time.Time `json:"endTime"`
	Code      int32     `json:"code"`
	Output    []byte    `json:"output,omitempty"`  // output of the task on success
	Message   string    `json:"message,omitempty"` // message of the task on failure
}