handler

type Blockchain interface {
	// executor operation
	RegisterExecutorNode(opt *blockchain.AddNodeOptions) error
	GetExecutorNodeByID(id string) (blockchain.ExecutorNode, error)
	ListExecutorNodes() (blockchain.ExecutorNodes, error)

	// task operation
	ListTask(opt *blockchain.ListFLTaskOptions) (blockchain.FLTasks, error)
	PublishTask(opt *blockchain.PublishFLTaskOptions) error
	GetTaskById(id string) (blockchain.FLTask, error)
	ConfirmTask(opt *blockchain.FLTaskConfirmOptions) error
	RejectTask(opt *blockchain.FLTaskConfirmOptions) error
	ExecuteTask(opt *blockchain.FLTaskExeStatusOptions) error
	FinishTask(opt *blockchain.FLTaskExeStatusOptions) error
	// get file stored in xuperDB by id
	GetFileByID(id string) (xdbchain.File, error)
	// query the list of authorization applications
	ListFileAuthApplications(opt *xdbchain.ListFileAuthOptions) (xdbchain.FileAuthApplications, error)
	// publish sample file's authorization application
	PublishFileAuthApplication(opt *xdbchain.PublishFileAuthOptions) error
	// query the list of storage nodes
	ListNodes() (xdbchain.Nodes, error)

	Close()
}

type FileDownload struct {
	Type           string           // support 'Proxy' and 'Self'
	NodePrivateKey ecdsa.PrivateKey // the executor node's private key

	PrivateKey ecdsa.PrivateKey // key authorized by data owner node, used when the Type is 'Self'
	Host       string           // data owner host address, used when the Type is 'Self'
}

type FileStorage struct {
	ModelStorage      Storage
	EvaluationStorage Storage
	PredictStorage    Storage
}

type FlTask struct {
	// task detail info on chain
	pbTask.FLTask
	// timeout for task execution
	ExpiredTime int64
}

type MpcHandler interface {
	// SaveModel persists a model
	// called by MPC
	SaveModel(*pbCom.TrainTaskResult) error

	// SavePredictOut persists predicting outcomes
	// outcomes will be zero-value if the holder does not have target feature
	// called by MPC
	SavePredictOut(*pbCom.PredictTaskResult) error

	// GetMpcClusterService returns mpc cluster service server
	GetMpcClusterService() *cluster.Service

	// TaskStartPrepare prepares resources needed by task, and adds task to execution pool.
	TaskStartPrepare(task blockchain.FLTask) (*pbCom.StartTaskRequest, error)

	// StartLocalMpcTask executes task
	StartLocalMpcTask(task *pbCom.StartTaskRequest, isSendTaskToOthers bool) error

	// GetAvailableTasksNum returns left number of tasks could be executed
	GetAvailableTasksNum() (int, int)

	// CheckMpcTimeOutTasks checks tasks in execution pool if they're expired,
	// and stops expired tasks
	CheckMpcTimeOutTasks()

	//Close closes all inner services
	Close()
}

type MpcModelHandler struct {
	Config             mpc.Config
	Node               Node          // executor node information
	Storage            FileStorage   // handler for computing results storage
	Download           FileDownload  // handler for file download, 'proxy' or 'self'
	Chain              Blockchain    // handler for blockchain operation
	MpcTaskMaxExecTime time.Duration // maximum execution time for mpc task
	Mpc                mpc.Mpc
	ClusterP2p         *p2p.P2P
	// store execution mpc tasks
	MpcTasks map[string]*FlTask
	sync.RWMutex
}

type Node struct {
	peer.Local
	HttpAddress     string
	PaddleFLAddress string
	PaddleFLRole    int
}

type ParticipantParams struct {
	PaddleFLRole  int
	PaddleFLNodes [3]string
	// contains filtered or unexported fields
}

type Storage interface {
	Write(value io.Reader, key string) (string, error)
	Read(key string) (io.ReadCloser, error)
}