add Close methods to smfreader and smfwriter

gomidi · Sep 3, 2018 · 3528d06 · 3528d06
1 parent 83eaf07
commit 3528d06
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/smf/smfreader/reader.go b/smf/smfreader/reader.go
@@ -71,6 +71,14 @@ func New(src io.Reader, opts ...Option) smf.Reader {
 	return rd
 }
 
+// Close closes the internal reader if it is an io.ReadCloser
+func (r *reader) Close() error {
+	if cl, is := r.input.(io.ReadCloser); is {
+		return cl.Close()
+	}
+	return nil
+}
+
 func (r *reader) ReadHeader() error {
 	if r.headerIsRead {
 		return r.error

diff --git a/smf/smfwriter/writer.go b/smf/smfwriter/writer.go
@@ -101,6 +101,13 @@ type writer struct {
 	runningWriter   runningstatus.SMFWriter
 }
 
+func (w *writer) Close() error {
+	if cl, is := w.output.(io.WriteCloser); is {
+		return cl.Close()
+	}
+	return nil
+}
+
 func (w *writer) WriteHeader() error {
 	if w.headerWritten {
 		return w.error