worker

Documentation

Index

• func Generate(g graph.Graph) map[int]float32
• func Peer2PeerSend(client pb.WorkerClient, message []*pb.MessageStruct, id int, ...)
• func Peer2PeerSync(client pb.WorkerClient, message []*pb.SyncStruct, id int)
• func RunCCWorker(id, partitionNum int, is_rep bool)
• func RunPRWorker(id, partitionNum int, is_rep bool)
• func RunSSSPWorker(id, partitionNum int, is_rep bool)
• type CCWorker
  • func (w *CCWorker) Assemble(ctx context.Context, args *pb.EmptyRequest) (*pb.AssembleResponse, error)
  • func (w *CCWorker) BeatHeart(ctx context.Context, args *pb.EmptyRequest) (*pb.EmptyResponse, error)
  • func (w *CCWorker) CCMasterSync()
  • func (w *CCWorker) CCMessageSend(messages map[int][]*algorithm.CCPair, calculateStep bool) []*pb.WorkerCommunicationSize
  • func (w *CCWorker) ExchangeMessage(ctx context.Context, args *pb.EmptyRequest) (*pb.ExchangeResponse, error)
  • func (w *CCWorker) IncEval(ctx context.Context, args *pb.EmptyRequest) (*pb.IncEvalResponse, error)
  • func (w *CCWorker) Lock()
  • func (w *CCWorker) MessageSend(ctx context.Context, args *pb.MessageRequest) (*pb.EmptyResponse, error)
  • func (w *CCWorker) PEval(ctx context.Context, args *pb.EmptyRequest) (*pb.PEvalResponse, error)
  • func (w *CCWorker) Recovery(ctx context.Context, args *pb.RecoveryRequest) (*pb.EmptyResponse, error)
  • func (w *CCWorker) ShutDown(ctx context.Context, args *pb.EmptyRequest) (*pb.ShutDownResponse, error)
  • func (w *CCWorker) SyncVal(ctx context.Context, args *pb.SyncRequest) (*pb.EmptyResponse, error)
  • func (w *CCWorker) UnLock()
• type PRWorker
  • func (w *PRWorker) Assemble(ctx context.Context, args *pb.EmptyRequest) (*pb.AssembleResponse, error)
  • func (w *PRWorker) BeatHeart(ctx context.Context, args *pb.EmptyRequest) (*pb.EmptyResponse, error)
  • func (w *PRWorker) ExchangeMessage(ctx context.Context, args *pb.EmptyRequest) (*pb.ExchangeResponse, error)
  • func (w *PRWorker) IncEval(ctx context.Context, args *pb.EmptyRequest) (*pb.IncEvalResponse, error)
  • func (w *PRWorker) Lock()
  • func (w *PRWorker) MessageSend(ctx context.Context, args *pb.MessageRequest) (*pb.EmptyResponse, error)
  • func (w *PRWorker) PEval(ctx context.Context, args *pb.EmptyRequest) (*pb.PEvalResponse, error)
  • func (w *PRWorker) PRMasterSync()
  • func (w *PRWorker) PRMessageSend(messages map[int][]*algorithm.PRPair, calculateStep bool) []*pb.WorkerCommunicationSize
  • func (w *PRWorker) Recovery(ctx context.Context, args *pb.RecoveryRequest) (*pb.EmptyResponse, error)
  • func (w *PRWorker) ShutDown(ctx context.Context, args *pb.EmptyRequest) (*pb.ShutDownResponse, error)
  • func (w *PRWorker) SyncVal(ctx context.Context, args *pb.SyncRequest) (*pb.EmptyResponse, error)
  • func (w *PRWorker) UnLock()
• type SSSPWorker
  • func (w *SSSPWorker) Assemble(ctx context.Context, args *pb.EmptyRequest) (*pb.AssembleResponse, error)
  • func (w *SSSPWorker) BeatHeart(ctx context.Context, args *pb.EmptyRequest) (*pb.EmptyResponse, error)
  • func (w *SSSPWorker) ExchangeMessage(ctx context.Context, args *pb.EmptyRequest) (*pb.ExchangeResponse, error)
  • func (w *SSSPWorker) IncEval(ctx context.Context, args *pb.EmptyRequest) (*pb.IncEvalResponse, error)
  • func (w *SSSPWorker) Lock()
  • func (w *SSSPWorker) MessageSend(ctx context.Context, args *pb.MessageRequest) (*pb.EmptyResponse, error)
  • func (w *SSSPWorker) PEval(ctx context.Context, args *pb.EmptyRequest) (*pb.PEvalResponse, error)
  • func (w *SSSPWorker) Recovery(ctx context.Context, args *pb.RecoveryRequest) (*pb.EmptyResponse, error)
  • func (w *SSSPWorker) SSSPMasterSync()
  • func (w *SSSPWorker) SSSPMessageSend(messages map[int][]*algorithm.Pair, calculateStep bool) []*pb.WorkerCommunicationSize
  • func (w *SSSPWorker) ShutDown(ctx context.Context, args *pb.EmptyRequest) (*pb.ShutDownResponse, error)
  • func (w *SSSPWorker) SyncVal(ctx context.Context, args *pb.SyncRequest) (*pb.EmptyResponse, error)
  • func (w *SSSPWorker) UnLock()

Functions

func Generate

func Generate(g graph.Graph) map[int]float32

func Peer2PeerSend

func Peer2PeerSend(client pb.WorkerClient, message []*pb.MessageStruct, id int, calculateStep bool)

rpc send has max size limit, so we spilt our transfer into many small block

func Peer2PeerSync

func Peer2PeerSync(client pb.WorkerClient, message []*pb.SyncStruct, id int)

func RunCCWorker

func RunCCWorker(id, partitionNum int, is_rep bool)

func RunPRWorker

func RunPRWorker(id, partitionNum int, is_rep bool)

func RunSSSPWorker

func RunSSSPWorker(id, partitionNum int, is_rep bool)

Types

type CCWorker

type CCWorker struct {
	CCValue map[int]int
	// contains filtered or unexported fields
}

func (*CCWorker) Assemble

func (w *CCWorker) Assemble(ctx context.Context, args *pb.EmptyRequest) (*pb.AssembleResponse, error)

func (*CCWorker) BeatHeart

func (w *CCWorker) BeatHeart(ctx context.Context, args *pb.EmptyRequest) (*pb.EmptyResponse, error)

func (*CCWorker) CCMasterSync

func (w *CCWorker) CCMasterSync()

func (*CCWorker) CCMessageSend

func (w *CCWorker) CCMessageSend(messages map[int][]*algorithm.CCPair, calculateStep bool) []*pb.WorkerCommunicationSize

func (*CCWorker) ExchangeMessage

func (w *CCWorker) ExchangeMessage(ctx context.Context, args *pb.EmptyRequest) (*pb.ExchangeResponse, error)

func (*CCWorker) IncEval

func (w *CCWorker) IncEval(ctx context.Context, args *pb.EmptyRequest) (*pb.IncEvalResponse, error)

func (*CCWorker) Lock

func (w *CCWorker) Lock()

func (*CCWorker) MessageSend

func (w *CCWorker) MessageSend(ctx context.Context, args *pb.MessageRequest) (*pb.EmptyResponse, error)

func (*CCWorker) PEval

func (w *CCWorker) PEval(ctx context.Context, args *pb.EmptyRequest) (*pb.PEvalResponse, error)

func (*CCWorker) Recovery

func (w *CCWorker) Recovery(ctx context.Context, args *pb.RecoveryRequest) (*pb.EmptyResponse, error)

func (*CCWorker) ShutDown

func (w *CCWorker) ShutDown(ctx context.Context, args *pb.EmptyRequest) (*pb.ShutDownResponse, error)

func (*CCWorker) SyncVal

func (w *CCWorker) SyncVal(ctx context.Context, args *pb.SyncRequest) (*pb.EmptyResponse, error)

func (*CCWorker) UnLock

func (w *CCWorker) UnLock()

type PRWorker

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

func (*PRWorker) Assemble

func (w *PRWorker) Assemble(ctx context.Context, args *pb.EmptyRequest) (*pb.AssembleResponse, error)

func (*PRWorker) BeatHeart

func (w *PRWorker) BeatHeart(ctx context.Context, args *pb.EmptyRequest) (*pb.EmptyResponse, error)

func (*PRWorker) ExchangeMessage

func (w *PRWorker) ExchangeMessage(ctx context.Context, args *pb.EmptyRequest) (*pb.ExchangeResponse, error)

func (*PRWorker) IncEval

func (w *PRWorker) IncEval(ctx context.Context, args *pb.EmptyRequest) (*pb.IncEvalResponse, error)

func (*PRWorker) Lock

func (w *PRWorker) Lock()

func (*PRWorker) MessageSend

func (w *PRWorker) MessageSend(ctx context.Context, args *pb.MessageRequest) (*pb.EmptyResponse, error)

func (*PRWorker) PEval

func (w *PRWorker) PEval(ctx context.Context, args *pb.EmptyRequest) (*pb.PEvalResponse, error)

func (*PRWorker) PRMasterSync

func (w *PRWorker) PRMasterSync()

func (*PRWorker) PRMessageSend

func (w *PRWorker) PRMessageSend(messages map[int][]*algorithm.PRPair, calculateStep bool) []*pb.WorkerCommunicationSize

func (*PRWorker) Recovery

func (w *PRWorker) Recovery(ctx context.Context, args *pb.RecoveryRequest) (*pb.EmptyResponse, error)

func (*PRWorker) ShutDown

func (w *PRWorker) ShutDown(ctx context.Context, args *pb.EmptyRequest) (*pb.ShutDownResponse, error)

func (*PRWorker) SyncVal

func (w *PRWorker) SyncVal(ctx context.Context, args *pb.SyncRequest) (*pb.EmptyResponse, error)

func (*PRWorker) UnLock

func (w *PRWorker) UnLock()

type SSSPWorker

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

func (*SSSPWorker) Assemble

func (w *SSSPWorker) Assemble(ctx context.Context, args *pb.EmptyRequest) (*pb.AssembleResponse, error)

func (*SSSPWorker) BeatHeart

func (w *SSSPWorker) BeatHeart(ctx context.Context, args *pb.EmptyRequest) (*pb.EmptyResponse, error)

func (*SSSPWorker) ExchangeMessage

func (w *SSSPWorker) ExchangeMessage(ctx context.Context, args *pb.EmptyRequest) (*pb.ExchangeResponse, error)

func (*SSSPWorker) IncEval

func (w *SSSPWorker) IncEval(ctx context.Context, args *pb.EmptyRequest) (*pb.IncEvalResponse, error)

func (*SSSPWorker) Lock

func (w *SSSPWorker) Lock()

func (*SSSPWorker) MessageSend

func (w *SSSPWorker) MessageSend(ctx context.Context, args *pb.MessageRequest) (*pb.EmptyResponse, error)

func (*SSSPWorker) PEval

func (w *SSSPWorker) PEval(ctx context.Context, args *pb.EmptyRequest) (*pb.PEvalResponse, error)

func (*SSSPWorker) Recovery

func (w *SSSPWorker) Recovery(ctx context.Context, args *pb.RecoveryRequest) (*pb.EmptyResponse, error)

func (*SSSPWorker) SSSPMasterSync

func (w *SSSPWorker) SSSPMasterSync()

func (*SSSPWorker) SSSPMessageSend

func (w *SSSPWorker) SSSPMessageSend(messages map[int][]*algorithm.Pair, calculateStep bool) []*pb.WorkerCommunicationSize

func (*SSSPWorker) ShutDown

func (w *SSSPWorker) ShutDown(ctx context.Context, args *pb.EmptyRequest) (*pb.ShutDownResponse, error)

func (*SSSPWorker) SyncVal

func (w *SSSPWorker) SyncVal(ctx context.Context, args *pb.SyncRequest) (*pb.EmptyResponse, error)

func (*SSSPWorker) UnLock

func (w *SSSPWorker) UnLock()