ocf: add encoder Flush

pedromss · Aug 25, 2020 · 05b56a2 · 05b56a2
1 parent a06921f
commit 05b56a2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/ocf/example_test.go b/ocf/example_test.go
@@ -72,7 +72,11 @@ func ExampleNewEncoder() {
 		log.Fatal(err)
 	}
 
-	if err := enc.Close(); err != nil {
+	if err := enc.Flush(); err != nil {
+		log.Fatal(err)
+	}
+
+	if err := f.Sync(); err != nil {
 		log.Fatal(err)
 	}
 }
diff --git a/ocf/ocf.go b/ocf/ocf.go
@@ -259,8 +259,8 @@ func (e *Encoder) Encode(v interface{}) error {
 	return e.writer.Error
 }
 
-// Close closes the encoder, flushing the writer.
-func (e *Encoder) Close() error {
+// Flush flushes the underlying writer.
+func (e *Encoder) Flush() error {
 	if e.count == 0 {
 		return nil
 	}
@@ -272,6 +272,11 @@ func (e *Encoder) Close() error {
 	return e.writer.Error
 }
 
+// Close closes the encoder, flushing the writer.
+func (e *Encoder) Close() error {
+	return e.Flush()
+}
+
 func (e *Encoder) writerBlock() error {
 	e.writer.WriteLong(int64(e.count))