func (cr *connectionRun) handleMsgFromServer(msg *msgs.Message) error {
if cr.currentState != cr {
return nil
}
cr.missingBeats = 0
switch which := msg.Which(); which {
case msgs.MESSAGE_HEARTBEAT:
// do nothing
case msgs.MESSAGE_CLIENTTXNOUTCOME:
outcome := msg.ClientTxnOutcome()
return cr.handleTxnOutcome(&outcome)
default:
return fmt.Errorf("Received unexpected message from server: %v", which)
}
return nil
}
func (cr *connectionRun) handleMsgFromPeer(msg *msgs.Message) error {
if cr.currentState != cr {
// probably just draining the queue from the reader after a restart
return nil
}
cr.missingBeats = 0
switch which := msg.Which(); which {
case msgs.MESSAGE_HEARTBEAT:
// do nothing
case msgs.MESSAGE_CLIENTTXNSUBMISSION:
ctxn := msg.ClientTxnSubmission()
origTxnId := common.MakeTxnId(ctxn.Id())
cr.submitter.SubmitClientTransaction(&ctxn, func(clientOutcome *msgs.ClientTxnOutcome, err error) {
switch {
case err != nil:
cr.clientTxnError(&ctxn, err, origTxnId)
case clientOutcome == nil: // shutdown
return
default:
seg := capn.NewBuffer(nil)
msg := msgs.NewRootMessage(seg)
msg.SetClientTxnOutcome(*clientOutcome)
cr.sendMessage(server.SegToBytes(msg.Segment))
}
})
default:
cr.connectionManager.Dispatchers.DispatchMessage(cr.remoteRMId, which, msg)
}
return nil
}
func (d *Dispatchers) DispatchMessage(sender common.RMId, msgType msgs.Message_Which, msg *msgs.Message) {
switch msgType {
case msgs.MESSAGE_TXNSUBMISSION:
txn := msg.TxnSubmission()
d.ProposerDispatcher.TxnReceived(sender, &txn)
case msgs.MESSAGE_SUBMISSIONOUTCOME:
outcome := msg.SubmissionOutcome()
txnId := common.MakeTxnId(outcome.Txn().Id())
connNumber := binary.BigEndian.Uint32(txnId[8:12])
bootNumber := binary.BigEndian.Uint32(txnId[12:16])
if conn := d.connectionManager.GetClient(bootNumber, connNumber); conn == nil {
NewOneShotSender(MakeTxnSubmissionCompleteMsg(txnId), d.connectionManager, sender)
} else {
conn.SubmissionOutcomeReceived(sender, txnId, &outcome)
return
}
case msgs.MESSAGE_SUBMISSIONCOMPLETE:
tsc := msg.SubmissionComplete()
d.AcceptorDispatcher.TxnSubmissionCompleteReceived(sender, &tsc)
case msgs.MESSAGE_SUBMISSIONABORT:
tsa := msg.SubmissionAbort()
d.ProposerDispatcher.TxnSubmissionAbortReceived(sender, &tsa)
case msgs.MESSAGE_ONEATXNVOTES:
oneATxnVotes := msg.OneATxnVotes()
d.AcceptorDispatcher.OneATxnVotesReceived(sender, &oneATxnVotes)
case msgs.MESSAGE_ONEBTXNVOTES:
oneBTxnVotes := msg.OneBTxnVotes()
d.ProposerDispatcher.OneBTxnVotesReceived(sender, &oneBTxnVotes)
case msgs.MESSAGE_TWOATXNVOTES:
twoATxnVotes := msg.TwoATxnVotes()
d.AcceptorDispatcher.TwoATxnVotesReceived(sender, &twoATxnVotes)
case msgs.MESSAGE_TWOBTXNVOTES:
twoBTxnVotes := msg.TwoBTxnVotes()
d.ProposerDispatcher.TwoBTxnVotesReceived(sender, &twoBTxnVotes)
case msgs.MESSAGE_TXNLOCALLYCOMPLETE:
tlc := msg.TxnLocallyComplete()
d.AcceptorDispatcher.TxnLocallyCompleteReceived(sender, &tlc)
case msgs.MESSAGE_TXNGLOBALLYCOMPLETE:
tgc := msg.TxnGloballyComplete()
d.ProposerDispatcher.TxnGloballyCompleteReceived(sender, &tgc)
default:
panic(fmt.Sprintf("Unexpected message received from %v", sender))
}
}