25 lines
557 B
C++
25 lines
557 B
C++
#pragma once
|
|
|
|
#include <zstd.h>
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <capnp/common.h>
|
|
|
|
class ZstdFileWriter {
|
|
public:
|
|
ZstdFileWriter(const std::string &filename, int compression_level);
|
|
~ZstdFileWriter();
|
|
void write(void* data, size_t size);
|
|
inline void write(kj::ArrayPtr<capnp::byte> array) { write(array.begin(), array.size()); }
|
|
|
|
private:
|
|
void flushCache(bool last_chunk);
|
|
|
|
size_t input_cache_capacity_ = 0;
|
|
std::vector<char> input_cache_;
|
|
std::vector<char> output_buffer_;
|
|
ZSTD_CStream *cstream_;
|
|
FILE* file_ = nullptr;
|
|
};
|