elefant-tools 0.0.2

A library for doing things like pg_dump and pg_restore, with extra features, and probably more bugs.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use crate::storage::data_format::DataFormat;
use crate::Result;
use bytes::Bytes;
use futures::Stream;
use std::future::Future;

/// Data in a table. This data is a stream which can be read from the data source.
///
/// Make sure to call `cleanup` when you have read all the data from the stream.
pub struct TableData<S: Stream<Item = Result<Bytes>> + Send, C: AsyncCleanup> {
    pub data: S,
    pub data_format: DataFormat,
    pub cleanup: C,
}

pub trait AsyncCleanup: Send {
    fn cleanup(self) -> impl Future<Output = Result<()>> + Send;
}

impl AsyncCleanup for () {
    async fn cleanup(self) -> Result<()> {
        Ok(())
    }
}

impl<S: Stream<Item = Result<Bytes>> + Send, C: AsyncCleanup> AsyncCleanup for TableData<S, C> {
    fn cleanup(self) -> impl Future<Output = Result<()>> + Send {
        self.cleanup.cleanup()
    }
}