io.Closer will close any closable connection

Author: BrandonRoehl

dump.go

@@ -44,7 +44,7 @@ type metaData struct {
 	CompleteTime  string
 }
 
-const version = "0.3.3"
+const version = "0.3.4"
 
 const headerTmpl = `-- Go SQL Dump {{ .DumpVersion }}
 --

mysqldump.go

@@ -53,16 +53,16 @@ func Dump(db *sql.DB, out io.Writer) error {
 }
 
 // Close the dumper.
-// Will also close the database the dumper is connected to as well as the out stream if it is a *os.File.
+// Will also close the database the dumper is connected to as well as the out stream if it has a Close method.
 //
 // Not required.
 func (d *Data) Close() error {
 	defer func() {
 		d.Connection = nil
 		d.Out = nil
 	}()
-	if file, ok := d.Out.(*os.File); ok {
-		file.Close()
+	if out, ok := d.Out.(io.Closer); ok {
+		out.Close()
 	}
 	return d.Connection.Close()
 }