Handle error if writing to temp file fails

jwilder · jwilder · commit ecdad5c60b94 · 2015-09-22T08:53:25.000-06:00
diff --git a/template.go b/template.go
@@ -397,7 +397,9 @@ func generateFile(config Config, containers Context) bool {
 			log.Fatalf("unable to create temp file: %s\n", err)
 		}
 
-		dest.Write(contents)
+		if n, err := dest.Write(contents); n != len(contents) || err != nil {
+			log.Fatalf("failed to write to temp file: wrote %d, exp %d, err=%v", n, len(contents), err)
+		}
 
 		oldContents := []byte{}
 		if fi, err := os.Stat(config.Dest); err == nil {

Original file line number	Diff line number	Diff line change
`@@ -397,7 +397,9 @@ func generateFile(config Config, containers Context) bool {`
`397`	`397`	`log.Fatalf("unable to create temp file: %s\n", err)`
`398`	`398`	`}`
`399`	`399`
`400`		`- dest.Write(contents)`
	`400`	`+ if n, err := dest.Write(contents); n != len(contents) \|\| err != nil {`
	`401`	`+ log.Fatalf("failed to write to temp file: wrote %d, exp %d, err=%v", n, len(contents), err)`
	`402`	`+ }`
`401`	`403`
`402`	`404`	`oldContents := []byte{}`
`403`	`405`	`if fi, err := os.Stat(config.Dest); err == nil {`