Add From support for Round2

pool2win · Nov 25, 2024 · 7c27eeb · 7c27eeb
1 parent 7e82fd0
commit 7c27eeb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/node/protocol.rs b/src/node/protocol.rs
@@ -132,6 +132,12 @@ impl From<dkg::round_one::PackageMessage> for Message {
     }
 }
 
+impl From<dkg::round_two::PackageMessage> for Message {
+    fn from(value: dkg::round_two::PackageMessage) -> Self {
+        Message::Unicast(Unicast::DKGRoundTwoPackage(value))
+    }
+}
+
 impl From<BroadcastProtocol> for Message {
     fn from(value: BroadcastProtocol) -> Self {
         match value {