add peer join result to receive

CympleTech · Mar 13, 2020 · 25a049b · 25a049b
1 parent 7055786
commit 25a049b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/message.rs b/src/message.rs
@@ -43,6 +43,9 @@ pub enum GroupReceiveMessage {
  /// when peer join, send from TDN to outside.
  /// params is `peer_id`, `socket_addr` and peer `join_info`.
  PeerJoin(PeerAddr, SocketAddr, Vec<u8>),
+ /// when peer get join result.
+ /// params is `peer_id`, `is_ok` and `result_data`.
+ PeerJoinResult(PeerAddr, bool, Vec<u8>),
  /// when peer leave, send from TDN to outside.
  /// params is `peer_id`.
  PeerLeave(PeerAddr),

diff --git a/src/p2p.rs b/src/p2p.rs
@@ -48,6 +48,11 @@ async fn run_listen<M: GroupMessage>(
  GroupReceiveMessage::PeerJoin(peer_addr, addr, data)
  )).await;
  },
+ ReceiveMessage::PeerJoinResult(peer_addr, is_ok, data) => {
+ out_send.send(M::new_group(
+ GroupReceiveMessage::PeerJoinResult(peer_addr, is_ok, data)
+ )).await;
+ },
  ReceiveMessage::PeerLeave(peer_addr) => {
  out_send.send(M::new_group(
  GroupReceiveMessage::PeerLeave(peer_addr)