aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/lib/io/record_writer.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/lib/io/record_writer.h')
-rw-r--r--tensorflow/core/lib/io/record_writer.h34
1 files changed, 34 insertions, 0 deletions
diff --git a/tensorflow/core/lib/io/record_writer.h b/tensorflow/core/lib/io/record_writer.h
new file mode 100644
index 0000000000..c7af00e5ae
--- /dev/null
+++ b/tensorflow/core/lib/io/record_writer.h
@@ -0,0 +1,34 @@
+#ifndef TENSORFLOW_LIB_IO_RECORD_WRITER_H_
+#define TENSORFLOW_LIB_IO_RECORD_WRITER_H_
+
+#include "tensorflow/core/lib/core/stringpiece.h"
+#include "tensorflow/core/platform/port.h"
+#include "tensorflow/core/public/status.h"
+
+namespace tensorflow {
+
+class WritableFile;
+
+namespace io {
+
+class RecordWriter {
+ public:
+ // Create a writer that will append data to "*dest".
+ // "*dest" must be initially empty.
+ // "*dest" must remain live while this Writer is in use.
+ explicit RecordWriter(WritableFile* dest);
+
+ ~RecordWriter();
+
+ Status WriteRecord(StringPiece slice);
+
+ private:
+ WritableFile* const dest_;
+
+ TF_DISALLOW_COPY_AND_ASSIGN(RecordWriter);
+};
+
+} // namespace io
+} // namespace tensorflow
+
+#endif // TENSORFLOW_LIB_IO_RECORD_WRITER_H_