client

type Alloc struct {
	PoolID   string `json:"pool_id"`
	AllocID  string `json:"alloc_id"`
	WorkerID string `json:"worker_id"`
}

type Client struct {
	// contains filtered or unexported fields
}

type CompleteAllocInput struct {
	PoolID  string `json:"pool_id"`
	AllocID string `json:"alloc_id"`
}

type CompleteAllocOutput struct{}

type CreatePoolInput struct{}

type CreatePoolOutput struct {
	PoolID string `json:"pool_id"`
}

type DisbandPoolInput struct {
	PoolID string `json:"pool_id"`
}

type DisbandPoolOutput struct{}

type ReceiveAllocsInput struct {
	WorkerQueueURL      string `json:"worker_queue_url"`
	MaxNumberOfMessages int64  `json:"max_number_of_messages"`
	WaitTimeSeconds     int64  `json:"wait_time_seconds"`
}

type ReceiveAllocsOutput struct {
	Allocs []*Alloc `json:"allocs"`
}

type RegisterWorkerInput struct {
	PoolID   string `json:"pool_id"`
	Capacity int    `json:"capacity"`
}

type RegisterWorkerOutput struct {
	PoolID   string `json:"pool_id"`
	WorkerID string `json:"worker_id"`
	QueueURL string `json:"queue_url"`
	Capacity int    `json:"capacity"`
}

type ScheduleEvalInput struct {
	PoolID    string `json:"pool_id"`
	DatasetID string `json:"dataset_id"`
	Size      int    `json:"size"`
}

type ScheduleEvalOutput struct{}

type SendHeartbeatInput struct {
	PoolID   string   `json:"pool_id"`
	WorkerID string   `json:"worker_id"`
	Allocs   []string `json:"allocs"`
	Datasets []string `json:"datasets"`
}

type SendHeartbeatOutput struct {
}