containerd
/
rust-extensions
mirror of https://github.com/containerd/rust-extensions.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #137 from simon-mo/list-request-impl 

Implement the input parameters for snapshotter::list
This commit is contained in:
Maksym Pavlenko 2023-04-23 19:28:03 +02:00 committed by GitHub
parent e3c2364587 9bf2db776b
commit c54a467f16
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
crates/snapshots/examples/snapshotter.rs
View File

@ -100,7 +100,12 @@ impl snapshots::Snapshotter for Example {
} }
type InfoStream = EmptyStream; type InfoStream = EmptyStream;
async fn list(&self) -> Result<Self::InfoStream, Self::Error> { async fn list(
&self,
snapshotter: String,
filters: Vec<String>,
) -> Result<Self::InfoStream, Self::Error> {
info!("List: snapshotter={}, filters={:?}", snapshotter, filters);
// Returns no snapshots. // Returns no snapshots.
Ok(EmptyStream) Ok(EmptyStream)
} }

6
crates/snapshots/src/lib.rs
View File

@ -283,5 +283,9 @@ pub trait Snapshotter: Send + Sync + 'static {
/// }) /// })
/// } /// }
/// ``` /// ```
async fn list(&self) -> Result<Self::InfoStream, Self::Error>; async fn list(
&self,
snapshotter: String,
filters: Vec<String>,
) -> Result<Self::InfoStream, Self::Error>;
} }

5
crates/snapshots/src/wrap.rs
View File

@ -158,11 +158,12 @@ impl<S: Snapshotter> Snapshots for Wrapper<S> {
async fn list( async fn list(
&self, &self,
_request: tonic::Request<ListSnapshotsRequest>, request: tonic::Request<ListSnapshotsRequest>,
) -> Result<tonic::Response<Self::ListStream>, tonic::Status> { ) -> Result<tonic::Response<Self::ListStream>, tonic::Status> {
let request = request.into_inner();
let sn = self.snapshotter.clone(); let sn = self.snapshotter.clone();
let output = async_stream::try_stream! { let output = async_stream::try_stream! {
let walk_stream = sn.list().await?; let walk_stream = sn.list(request.snapshotter, request.filters).await?;
pin_utils::pin_mut!(walk_stream); pin_utils::pin_mut!(walk_stream);
let mut infos = Vec::<Info>::new(); let mut infos = Vec::<Info>::new();
while let Some(info) = walk_stream.next().await { while let Some(info) = walk_stream.next().await {