Merge pull request #234 from dolmen-go/WriteTo

Extract WriteTo method (see io.WriterTo) to expose bytes written
formula
xuri 7 years ago committed by GitHub
commit d80d6aa318
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -60,6 +60,12 @@ func (f *File) SaveAs(name string) error {
// Write provides function to write to an io.Writer. // Write provides function to write to an io.Writer.
func (f *File) Write(w io.Writer) error { func (f *File) Write(w io.Writer) error {
_, err := f.WriteTo(w)
return err
}
// WriteTo implements io.WriterTo to write the file.
func (f *File) WriteTo(w io.Writer) (int64, error) {
buf := new(bytes.Buffer) buf := new(bytes.Buffer)
zw := zip.NewWriter(buf) zw := zip.NewWriter(buf)
f.contentTypesWriter() f.contentTypesWriter()
@ -70,21 +76,17 @@ func (f *File) Write(w io.Writer) error {
for path, content := range f.XLSX { for path, content := range f.XLSX {
fi, err := zw.Create(path) fi, err := zw.Create(path)
if err != nil { if err != nil {
return err return 0, err
} }
_, err = fi.Write(content) _, err = fi.Write(content)
if err != nil { if err != nil {
return err return 0, err
} }
} }
err := zw.Close() err := zw.Close()
if err != nil { if err != nil {
return err return 0, err
}
if _, err := buf.WriteTo(w); err != nil {
return err
} }
return nil return buf.WriteTo(w)
} }

Loading…
Cancel
Save