internal/driver: close saved profile files (#1022)
* internal/driver: close saved profile files
Signed-off-by: cuishuang <imcusg@gmail.com>
* internal/driver: close saved profile files
Signed-off-by: cuishuang <imcusg@gmail.com>
---------
Signed-off-by: cuishuang <imcusg@gmail.com>
Co-authored-by: Alexey Alexandrov <aalexand@users.noreply.github.com>
diff --git a/internal/driver/fetch.go b/internal/driver/fetch.go
index 0809cd0..cc31971 100644
--- a/internal/driver/fetch.go
+++ b/internal/driver/fetch.go
@@ -110,7 +110,14 @@
tempFile, err := newTempFile(dir, prefix, ".pb.gz")
if err == nil {
- if err = p.Write(tempFile); err == nil {
+ writeErr := p.Write(tempFile)
+ closeErr := tempFile.Close()
+ switch {
+ case writeErr != nil:
+ err = writeErr
+ case closeErr != nil:
+ err = closeErr
+ default:
o.UI.PrintErr("Saved profile in ", tempFile.Name())
}
}