message: handle Writer.Close errors in Entity.WriteTo

emersion · Mar 6, 2024 · f7e55c4 · f7e55c4
1 parent dbb628d
commit f7e55c4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/entity.go b/entity.go
@@ -187,9 +187,13 @@ func (e *Entity) WriteTo(w io.Writer) error {
  if err != nil {
  return err
  }
- defer ew.Close()
 
- return e.writeBodyTo(ew)
+ if err := e.writeBodyTo(ew); err != nil {
+ ew.Close()
+ return err
+ }
+
+ return ew.Close()
 }
 
 // WalkFunc is the type of the function called for each part visited by Walk.