How to get client's socket status when grpc is on server-side streaming?

[yang282441848]

I want to manage grpc-clients, but cannot find any way to remove clients when they shutdown the grpc socket.

It looks like that only tx.send 's result or tx.isClosed() can get this state. But I want clear the memory not by polling when client is disconnected .

use std::{pin::Pin, time::Duration};

use protobuf::{
    greeter_server::Greeter, inner_message::Data, Auth, ChatMessage, HelloReply, HelloRequest,
    InnerMessage,
};
use tokio::sync::{mpsc, Mutex};
use tokio_stream::{wrappers::ReceiverStream, Stream};
use tonic::{Request, Response, Status};

use crate::{client::AppClient, hub::ClientHub};

pub struct Server {
    pub hub: Mutex<ClientHub>,
}

#[tonic::async_trait]
impl Greeter for Server {
    type MessageStream = ReceiverStream<Result<InnerMessage, Status>>;

    async fn say_hello(
        &self,
        request: Request<HelloRequest>,
    ) -> Result<Response<HelloReply>, Status> {
        Ok(Response::new(HelloReply {
            message: format!("hello {}", request.into_inner().name),
        }))
    }
    async fn message(
        &self,
        request: tonic::Request<Auth>,
    ) -> Result<Response<Self::MessageStream>, Status> {
        let (tx, rx) = mpsc::channel(128);
        let client = AppClient {
            tx,
            addr: request.remote_addr(),
        };

        /// TODO:  How to remove this client if it shutdown this socket.
        self.hub.lock().await.insert(client);
        let output_stream = ReceiverStream::new(rx);
        Ok(Response::new(output_stream))
    }
}

[yang282441848]

#1302