From 51079288923076d00a8b36ecec07980a158d742c Mon Sep 17 00:00:00 2001
From: zaddok <zadok@tai.io>
Date: Fri, 9 Aug 2019 09:47:06 +1000
Subject: Fix potential memory leak
Fix potential memory leak where zw is not Close() when an error occurs.
---
file.go | 2 ++
1 file changed, 2 insertions(+)
(limited to 'file.go')
diff --git a/file.go b/file.go
index a9e7eec..a4aa11d 100644
--- a/file.go
+++ b/file.go
@@ -109,10 +109,12 @@ func (f *File) WriteToBuffer() (*bytes.Buffer, error) {
for path, content := range f.XLSX {
fi, err := zw.Create(path)
if err != nil {
+ zw.Close()
return buf, err
}
_, err = fi.Write(content)
if err != nil {
+ zw.Close()
return buf, err
}
}
--
cgit v1.2.1