cluster

type Mpc interface {
	Predict(*pb.PredictRequest) (*pb.PredictResponse, error)
	Train(*pb.TrainRequest) (*pb.TrainResponse, error)
}

type P2P interface {
	GetPeer(address string) (*p2p.Peer, error)
	FreePeer()
}

type PredictHandler interface {
	StepPredict(req *pb.PredictRequest, peerName string) (*pb.PredictResponse, error)

	// StepPredictWithRetry sends prediction message to remote mpc-node
	// retries 2 times at most
	// inteSec indicates the interval between retry requests, in seconds
	StepPredictWithRetry(req *pb.PredictRequest, peerName string, times int, inteSec int64) (*pb.PredictResponse, error)
}

type Rpc interface {
	PredictHandler
	TrainHandler
}

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

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

type TrainHandler interface {
	StepTrain(req *pb.TrainRequest, peerName string) (*pb.TrainResponse, error)

	// StepTrainWithRetry sends training message to remote mpc-node
	// retries 2 times at most
	// inteSec indicates the interval between retry requests, in seconds
	StepTrainWithRetry(req *pb.TrainRequest, peerName string, times int, inteSec int64) (*pb.TrainResponse, error)
}