Merge pull request #115 from chenyukang/yukang-add-reuse

Add reuse address and port for CI issues

Cargo.toml

@@ -67,6 +67,7 @@ ractor = "0.9.7"
 arcode = "0.2.4"
 nom = "7.1.3"
 regex = "1.10.5"
+socket2 = "0.5.7"
 
 [profile.release]
 panic = "abort"

src/rpc/mod.rs

@@ -21,14 +21,41 @@ use tokio::sync::mpsc::Sender;
 
 pub type InvoiceCommandWithReply = (InvoiceCommand, Sender<crate::Result<String>>);
 
+fn build_server(addr: &str) -> Server {
+    #[cfg(not(release))]
+    {
+        // Use socket2 to set reuse address and reuse port,
+        // so that we can restart the server without waiting for the port to be released.
+        // it will avoid the error: "Address already in use" in CI.
+        use socket2::{Domain, Socket, Type};
+        let addr = addr.parse().unwrap();
+        let domain = Domain::for_address(addr);
+        let socket = Socket::new(domain, Type::STREAM, None).unwrap();
+        socket.set_nonblocking(true).unwrap();
+        socket.set_reuse_address(true).expect("set reuse address");
+        socket.set_reuse_port(true).expect("set reuse port");
+
+        socket.bind(&addr.into()).unwrap();
+        socket.listen(4096).unwrap();
+
+        jsonrpsee::server::Server::builder()
+            .build_from_tcp(socket)
+            .unwrap()
+    }
+    #[cfg(release)]
+    {
+        Server::builder().build(addr).await.unwrap()
+    }
+}
+
 pub async fn start_rpc<S: ChannelActorStateStore + InvoiceStore + Clone + Send + Sync + 'static>(
     config: RpcConfig,
     ckb_network_actor: Option<ActorRef<NetworkActorMessage>>,
     cch_command_sender: Option<Sender<CchCommand>>,
     store: S,
 ) -> ServerHandle {
     let listening_addr = config.listening_addr.as_deref().unwrap_or("[::]:0");
-    let server = Server::builder().build(listening_addr).await.unwrap();
+    let server = build_server(listening_addr);
     let mut methods = InvoiceRpcServerImpl::new(store.clone()).into_rpc();
     if let Some(ckb_network_actor) = ckb_network_actor {
         let peer = PeerRpcServerImpl::new(ckb_network_actor.clone());