Skip to content

Commit a0dd103

Browse files
committed
Ensure we bail and exit with a nonzero status if an error happens during put
1 parent b585d43 commit a0dd103

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

cmd/hdfs/put.go

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
package main
22

33
import (
4-
"fmt"
54
"io"
65
"os"
76
"path"
@@ -82,14 +81,12 @@ func putFromFile(client *hdfs.Client, source string, dest string) {
8281
mode := 0755 | os.ModeDir
8382
err = filepath.Walk(source, func(p string, fi os.FileInfo, err error) error {
8483
if err != nil {
85-
fmt.Fprintln(os.Stderr, err)
86-
return nil
84+
return err
8785
}
8886

8987
rel, err := filepath.Rel(source, p)
9088
if err != nil {
91-
fmt.Fprintln(os.Stderr, err)
92-
return nil
89+
return err
9390
}
9491

9592
fullDest := path.Join(dest, rel)
@@ -98,24 +95,26 @@ func putFromFile(client *hdfs.Client, source string, dest string) {
9895
} else {
9996
writer, err := client.Create(fullDest)
10097
if err != nil {
101-
fmt.Fprintln(os.Stderr, err)
102-
return nil
98+
return err
10399
}
104100

105101
defer writer.Close()
106102
reader, err := os.Open(p)
107103
if err != nil {
108-
fmt.Fprintln(os.Stderr, err)
109-
return nil
104+
return err
110105
}
111106

112107
defer reader.Close()
113108
_, err = io.Copy(writer, reader)
114109
if err != nil {
115-
fmt.Fprintln(os.Stderr, err)
110+
return err
116111
}
117112
}
118113

119114
return nil
120115
})
116+
117+
if err != nil {
118+
fatal(err)
119+
}
121120
}

0 commit comments

Comments
 (0)