common/iowrap: Add

3 files changed, 32 insertions, 0 deletions

diff --git a/common/iowrap/doc.go b/common/iowrap/doc.go
new file mode 100644
index 00000000..8e25db16
--- /dev/null
+++ b/common/iowrap/doc.go
@@ -0,0 +1,2 @@
+// Package iowrap provides small public I/O wrapper interfaces and adapters.
+package iowrap
diff --git a/common/iowrap/write_flusher.go b/common/iowrap/write_flusher.go
new file mode 100644
index 00000000..aaac8724
--- /dev/null
+++ b/common/iowrap/write_flusher.go
@@ -0,0 +1,9 @@
+package iowrap
+
+import "io"
+
+// WriteFlusher writes bytes and flushes buffered output state.
+type WriteFlusher interface {
+	io.Writer
+	Flush() error
+}
diff --git a/common/iowrap/write_flusher_nop.go b/common/iowrap/write_flusher_nop.go
new file mode 100644
index 00000000..91d30c3e
--- /dev/null
+++ b/common/iowrap/write_flusher_nop.go
@@ -0,0 +1,21 @@
+package iowrap
+
+import "io"
+
+type nopFlusher struct {
+	io.Writer
+}
+
+// NopFlush adapts writer into a [WriteFlusher] with a no-op Flush.
+func NopFlush(writer io.Writer) WriteFlusher {
+	if writer == nil {
+		return nil
+	}
+
+	return nopFlusher{Writer: writer}
+}
+
+// Flush lets us satisfy [WriteFlusher] but does nothing.
+func (nopFlusher) Flush() error {
+	return nil
+}