return document count from builder

This commit is contained in:
marin postma 2021-10-28 10:33:04 +02:00
parent d7943fe225
commit 183d3dada7
No known key found for this signature in database
GPG Key ID: 6088B7721C3E39F9

View File

@ -62,7 +62,7 @@ impl<W: io::Write + io::Seek> DocumentBatchBuilder<W> {
/// This method must be called after the document addition is terminated. It will put the /// This method must be called after the document addition is terminated. It will put the
/// metadata at the end of the file, and write the metadata offset at the beginning on the /// metadata at the end of the file, and write the metadata offset at the beginning on the
/// file. /// file.
pub fn finish(self) -> Result<(), Error> { pub fn finish(self) -> Result<usize, Error> {
let Self { inner: ByteCounter { mut writer, count: offset }, index, count, .. } = self; let Self { inner: ByteCounter { mut writer, count: offset }, index, count, .. } = self;
let meta = DocumentsMetadata { count, index }; let meta = DocumentsMetadata { count, index };
@ -74,7 +74,7 @@ impl<W: io::Write + io::Seek> DocumentBatchBuilder<W> {
writer.flush()?; writer.flush()?;
Ok(()) Ok(count)
} }
/// Extends the builder with json documents from a reader. /// Extends the builder with json documents from a reader.