@@ -21,7 +21,10 @@ public static Dictionary<string, object> ConvertHeader(Header header)
2121 { HeaderElement . IncomingEventType , header ? . IncomingEventType } ,
2222 { HeaderElement . PublishTopic , header ? . PublishTopic != null ? encoding . GetBytes ( header . PublishTopic ) : encoding . GetBytes ( string . Empty ) } ,
2323 { HeaderElement . MessageType , header ? . MessageType != null ? encoding . GetBytes ( header . MessageType ) : encoding . GetBytes ( string . Empty ) } ,
24- { HeaderElement . ErrorMessage , header ? . ErrorMessage != null ? encoding . GetBytes ( header . ErrorMessage ) : encoding . GetBytes ( string . Empty ) }
24+ { HeaderElement . ErrorMessage , header ? . ErrorMessage != null ? encoding . GetBytes ( header . ErrorMessage ) : encoding . GetBytes ( string . Empty ) } ,
25+ { HeaderElement . MessageId , header ? . MessageId != null ? encoding . GetBytes ( header . MessageId ) : encoding . GetBytes ( string . Empty ) } ,
26+
27+
2528 } ;
2629 return dico ;
2730 }
@@ -40,7 +43,10 @@ public static Dictionary<string, object> CreateHeaderFromStateMachineRefHeader(S
4043 { HeaderElement . MessageType , stateMachineRefHeader ? . MessageType != null ? encoding . GetBytes ( stateMachineRefHeader . MessageType ) : encoding . GetBytes ( string . Empty ) } ,
4144 { HeaderElement . EventType , eventCode } ,
4245 { HeaderElement . IncomingEventType , ( int ) incomingEventType } ,
43- { HeaderElement . ErrorMessage , stateMachineRefHeader ? . ErrorMessage != null ? encoding . GetBytes ( stateMachineRefHeader . ErrorMessage ) : encoding . GetBytes ( string . Empty ) }
46+ { HeaderElement . ErrorMessage , stateMachineRefHeader ? . ErrorMessage != null ? encoding . GetBytes ( stateMachineRefHeader . ErrorMessage ) : encoding . GetBytes ( string . Empty ) } ,
47+ { HeaderElement . MessageId , stateMachineRefHeader ? . MessageId != null ? encoding . GetBytes ( stateMachineRefHeader . MessageId ) : encoding . GetBytes ( string . Empty ) } ,
48+ { HeaderElement . WorkerId , stateMachineRefHeader ? . WorkerId } ,
49+
4450 } ;
4551 return dico ;
4652 }
@@ -56,6 +62,8 @@ public static StateMachineRefHeader ConvertStateMachineRefHeader(IDictionary<str
5662 var messageType = string . Empty ;
5763 var sessionData = string . Empty ;
5864 var errorMessage = string . Empty ;
65+ var messageId = string . Empty ;
66+ var workerId = - 1 ;
5967
6068 if ( stateMachineRefHeader . ContainsKey ( HeaderElement . StateMachineId ) && stateMachineRefHeader [ HeaderElement . StateMachineId ] != null )
6169 stateMachineId = Encoding . UTF8 . GetString ( ( byte [ ] ) stateMachineRefHeader [ HeaderElement . StateMachineId ] ) ;
@@ -73,6 +81,10 @@ public static StateMachineRefHeader ConvertStateMachineRefHeader(IDictionary<str
7381 sessionData = encoding . GetString ( stateMachineRefHeader [ HeaderElement . SessionData ] as byte [ ] ) ;
7482 if ( stateMachineRefHeader . ContainsKey ( HeaderElement . ErrorMessage ) )
7583 errorMessage = encoding . GetString ( stateMachineRefHeader [ HeaderElement . ErrorMessage ] as byte [ ] ) ;
84+ if ( stateMachineRefHeader . ContainsKey ( HeaderElement . MessageId ) )
85+ messageId = encoding . GetString ( stateMachineRefHeader [ HeaderElement . MessageId ] as byte [ ] ) ;
86+ if ( stateMachineRefHeader . ContainsKey ( HeaderElement . WorkerId ) )
87+ workerId = Convert . ToInt32 ( stateMachineRefHeader [ HeaderElement . WorkerId ] ) ;
7688
7789 return new StateMachineRefHeader ( )
7890 {
@@ -83,7 +95,9 @@ public static StateMachineRefHeader ConvertStateMachineRefHeader(IDictionary<str
8395 MessageType = messageType ,
8496 PrivateTopic = publishTopic ,
8597 SessionData = sessionData ,
86- ErrorMessage = errorMessage
98+ ErrorMessage = errorMessage ,
99+ MessageId = messageId ,
100+ WorkerId = workerId ,
87101 } ;
88102 }
89103 }
0 commit comments