shim: re-impl Events trait for async RemotePublisher

Fixes: https://github.com/containerd/rust-extensions/pull/318/files#r2212454388

Signed-off-by: Tim Zhang <[email protected]>

Author: Tim-Zhang

crates/shim/src/asynchronous/publisher.rs

@@ -16,13 +16,14 @@
 
 use std::os::unix::io::RawFd;
 
+use async_trait::async_trait;
 use containerd_shim_protos::{
-    api::Envelope,
+    api::{Empty, Envelope},
     protobuf::MessageDyn,
     shim::events,
-    shim_async::{Client, EventsClient},
+    shim_async::{Client, Events, EventsClient},
     ttrpc,
-    ttrpc::context::Context,
+    ttrpc::{context::Context, r#async::TtrpcContext},
 };
 use log::{debug, error, warn};
 use tokio::sync::mpsc;
@@ -169,6 +170,29 @@ impl RemotePublisher {
     }
 }
 
+#[async_trait]
+impl Events for RemotePublisher {
+    async fn forward(
+        &self,
+        _ctx: &TtrpcContext,
+        req: events::ForwardRequest,
+    ) -> ttrpc::Result<Empty> {
+        let item = Item {
+            ev: req.envelope().clone(),
+            ctx: Context::default(),
+            count: 0,
+        };
+
+        //if channel is full and send fail ,release it after 3 seconds
+        self.sender
+            .send_timeout(item, tokio::time::Duration::from_secs(3))
+            .await
+            .map_err(|e| error::Error::Ttrpc(ttrpc::error::Error::Others(e.to_string())))?;
+
+        Ok(Empty::default())
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use std::{