direct

type AgentBase struct {
	IP      string            `json:"ip"`
	Port    int               `json:"port"`
	Message string            `json:"message"`
	Cluster string            `json:"cluster"`
	Labels  map[string]string `json:"labels"`
}

type AgentInfo struct {
	Base      AgentBase        `json:"base"`
	Total     Resource         `json:"total"`
	Free      Resource         `json:"free"`
	Allocated []*AllocatedInfo `json:"allocated"`
}

type AgentResource struct {
	Cluster     string    `gorm:"column:cluster;primary_key" sql:"type:varchar(64)" json:"cluster"`
	IP          string    `gorm:"column:ip;primary_key" sql:"type:varchar(32)" json:"ip"`
	UpdatedAt   time.Time `gorm:"column:updated_at" json:"-"`
	TotalCPU    float32   `gorm:"column:total_cpu" json:"total_cpu"`
	TotalMemory float32   `gorm:"column:total_memory" json:"total_memory"`
	TotalDisk   float32   `gorm:"column:total_disk" json:"total_disk"`
	FreeCPU     float32   `gorm:"column:free_cpu" json:"free_cpu"`
	FreeMemory  float32   `gorm:"column:free_memory" json:"free_memory"`
	FreeDisk    float32   `gorm:"column:free_disk" json:"free_disk"`
}

type AgentResourceExternal struct {
	Base     AgentBase `json:"base"`
	Resource Resource  `json:"free"`
}

type AllocatedInfo struct {
	AllocatedResource Resource `json:"allocated_resource"`
	// 资源使用者id
	UserID string `json:"user_id"`
	// 资源分配id（分配资源时生成的id）
	ResBatchID string         `json:"res_batch_id"`
	Commands   []*CommandInfo `json:"commands"`
}

type AllocatedResource struct {
	UpdatedAt       time.Time `gorm:"column:updated_at" json:"-"`
	UserID          string    `gorm:"column:user_id;primary_key" sql:"type:varchar(64)" json:"user_id"`
	ResourceBatchID string    `gorm:"column:resource_batch_id;primary_key" sql:"type:varchar(128)" json:"resource_batch_id"`
	Released        int32     `gorm:"column:released;index" json:"released"`
	AllocatedTime   int64     `gorm:"column:allocated_time" json:"allocated_time"`
	ReleasedTime    int64     `gorm:"column:released_time" json:"released_time"`
	AllocatedAgent  string    `gorm:"column:allocated_agent" sql:"type:text" json:"allocated_agent"`
	Message         string    `gorm:"column:message" sql:"type:text" json:"message"`
}

type CallBack4Command func(usedRes []*CommandResultInfo) error

type CallBackSelector func(free []*AgentResourceExternal, condition interface{}) ([]*AgentResourceExternal, error)

type CmdType string

type Command struct {
	// 命令的工作目录
	Dir string `json:"dir"`
	// 可执行文件路径，如果是相对路径，则相对于Dir
	Path       string            `json:"path"`
	Cmd        string            `json:"cmd"`
	Parameters []string          `json:"parameters"`
	Env        map[string]string `json:"env"`
	Additional map[string]string `json:"additional"`
	CmdType    CmdType           `json:"cmd_type"`
	// 用户自定义ID，具体含义由资源使用者自己解释
	UserDefineID string `json:"user_define_id"`
	// 保存关联的命令字和id，比如执行释放命令时，需要带上启动命令和进程id，便于agent侧执行相应的释放
	ReferCmd string `json:"refer_cmd"`
	ReferID  string `json:"refer_id"`
}

type CommandInfo struct {
	// 远端关联的id，比如拉起的进程id
	ID   string `json:"id"`
	Cmd  string `json:"cmd"`
	Port int    `json:"port"`
	// 状态
	Status CommandStatusType `json:"status"`
	// 用户自定义ID，具体含义由资源使用者自己解释
	UserDefineID string `json:"user_define_id"`
}

type CommandResultInfo struct {
	IP string `json:"ip"`
	// 远端关联的id，比如拉起的进程id
	ID  string `json:"id"`
	Cmd string `json:"cmd"`
	// 状态
	Status CommandStatusType `json:"status"`
	// 用户自定义ID，具体含义由资源使用者自己解释
	UserDefineID string `json:"user_define_id"`
}

type CommandStatusType int

type HandleWithUser interface {
	// 根据传入的查询条件 和注册的 选择资源回调函数选择资源；记录针对该批资源的通知回调函数
	// 返回值： 资源列表和错误信息
	GetFreeResource(
		resBatchID string,
		condition interface{},
		callbackSelector CallBackSelector,
		callback4UsedRes CallBack4Command) ([]*AgentResourceExternal, error)
	// 释放resourceID标记的相关资源
	ReleaseResource(resBatchID string) error
	// 返回资源id关联的资源列表
	ListResource(resBatchID string) ([]*AgentResourceExternal, error)

	// 在远端agent上执行命令，错误返回值表示当前操作是否正常，不代表远端命令已经执行成功
	ExecuteCommand(ip string, resBatchID string, cmd *Command) error
	// 返回资源id关联的command列表
	ListCommands(resBatchID string) ([]*CommandResultInfo, error)

	// p2p资源不是独占的，逻辑不一样
	GetFreeP2PResource(
		resBatchID string,
		condition interface{},
		callbackSelector CallBackSelector,
		groupKey string,
		platform string) ([]*AgentResourceExternal, error)

	ReleaseP2PResource(resBatchID, platform, groupKey string) error
}

type MySQL interface {
	PutAgentResource(ar *AgentResource) error

	ListAllocateResource(opts commonMySQL.ListOptions) ([]*AllocatedResource, int64, error)
	GetAllocatedResource(userID, resourceBatchID string) (*AllocatedResource, error)
	PutAllocatedResource(ar *AllocatedResource) error
}

type MySQLConf struct {
	MySQLStorage     string
	MySQLDatabase    string
	MySQLUser        string
	MySQLPwd         string
	Charset          string
	MysqlTableOption string
}

type NotifyAgentData struct {
	// 资源使用者的ID
	UserID     string `json:"user_id"`
	ResBatchID string `json:"res_batch_id"`
	// 用户自定义ID，具体含义由资源使用者自己解释
	UserDefineID string `json:"user_define_id"`
	// 命令的工作目录
	Dir string `json:"dir"`
	// 可执行文件路径，如果是相对路径，则相对于agent运行目录
	Path       string            `json:"path"`
	Cmd        string            `json:"cmd"`
	Parameters []string          `json:"parameters"`
	Env        map[string]string `json:"env"`
	Additional map[string]string `json:"additional"`
	CmdType    string            `json:"cmd_type"`
	// 保存关联的命令字和id，比如执行释放命令时，需要带上启动命令和进程id，便于agent侧执行相应的释放
	ReferCmd string `json:"refer_cmd"`
	ReferID  string `json:"refer_id"`
}

type ReportAgentResource struct {
	AgentInfo
}

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

type Resource struct {
	CPU  float64 `json:"cpu"`
	Mem  float64 `json:"mem"`
	Disk float64 `json:"disk"`
}

type ResourceManager interface {
	// 注册用户信息，返回用户相关的handle
	RegisterUser(userID string, releaseCmd *Command) (HandleWithUser, error)

	Run() error
}

type RestResponse struct {
	Resp     *restful.Response
	HTTPCode int

	Data    interface{}
	ErrCode commonTypes.ServerErrCode
	//ErrCode int
	Message string
	Extra   map[string]interface{}

	WrapFunc func([]byte) []byte
}