Skip to content

Commit 4eeed3f

Browse files
committed
Simplify initial node.Fetch allocation via estimate.
1 parent 1e11b57 commit 4eeed3f

File tree

1 file changed

+6
-14
lines changed

1 file changed

+6
-14
lines changed

node/root.go

+6-14
Original file line numberDiff line numberDiff line change
@@ -20,32 +20,24 @@ func Fetch(options map[string]bool) []Node {
2020
files, err := ioutil.ReadDir(".")
2121
if err != nil { log.Fatal(err) }
2222

23-
count := 0
24-
for i:=0; i<len(files); i++ {
25-
if !options["all"] && []rune(files[i].Name())[0] == '.' { continue }
26-
count++
27-
}
28-
29-
if options["all"] { count += 2 }
30-
31-
nodes := make([]Node, count)
23+
nodes := make([]Node, len(files)+2)
3224

33-
count = 0
25+
index := 0
3426
if options["all"] {
3527
file, _ := os.Stat(".")
3628
nodes[0] = new(file)
3729
file, _ = os.Stat("..")
3830
nodes[1] = new(file)
39-
count += 2
31+
index += 2
4032
}
4133

4234
for i:=0; i<len(files); i++ {
4335
if !options["all"] && []rune(files[i].Name())[0] == '.' { continue }
44-
nodes[count] = new(files[i])
45-
count++
36+
nodes[index] = new(files[i])
37+
index++
4638
}
4739

48-
return nodes
40+
return nodes[:index]
4941
}
5042

5143
func new(file os.FileInfo) Node {

0 commit comments

Comments
 (0)