forked from colinmarc/hdfs
-
Notifications
You must be signed in to change notification settings - Fork 5
/
put.go
85 lines (73 loc) · 1.52 KB
/
put.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package main
import (
"fmt"
"io"
"os"
"path"
"path/filepath"
)
func put(args []string) {
if len(args) != 2 {
printHelp()
}
dests, nn, err := normalizePaths(args[1:])
if err != nil {
fatal(err)
}
dest := dests[0]
source, err := filepath.Abs(args[0])
if err != nil {
fatal(err)
}
client, err := getClient(nn)
if err != nil {
fatal(err)
}
// If the destination is an existing directory, place it inside. Otherwise,
// the destination is really the parent directory, and we need to rename the
// source directory as we copy.
existing, err := client.Stat(dest)
if err == nil {
if existing.IsDir() {
dest = path.Join(dest, filepath.Base(source))
} else {
fatal(&os.PathError{"mkdir", dest, os.ErrExist})
}
} else if !os.IsNotExist(err) {
fatal(err)
}
mode := 0755 | os.ModeDir
err = filepath.Walk(source, func(p string, fi os.FileInfo, err error) error {
if err != nil {
fmt.Fprintln(os.Stderr, err)
return nil
}
rel, err := filepath.Rel(source, p)
if err != nil {
fmt.Fprintln(os.Stderr, err)
return nil
}
fullDest := path.Join(dest, rel)
if fi.IsDir() {
client.Mkdir(fullDest, mode)
} else {
writer, err := client.Create(fullDest)
if err != nil {
fmt.Fprintln(os.Stderr, err)
return nil
}
defer writer.Close()
reader, err := os.Open(p)
if err != nil {
fmt.Fprintln(os.Stderr, err)
return nil
}
defer reader.Close()
_, err = io.Copy(writer, reader)
if err != nil {
fmt.Fprintln(os.Stderr, err)
}
}
return nil
})
}