/
fileutils.go
52 lines (42 loc) · 923 Bytes
/
fileutils.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
package cliftondbserver
import (
"io/ioutil"
"os"
"path"
"strconv"
)
func ensureDirsExist(folderPaths ...string) error {
var err error
for _, p := range folderPaths {
err = os.MkdirAll(p, os.ModePerm)
if err != nil {
return err
}
}
return nil
}
func getPartitionsFromDataDir(dataDirPath string, lockFileName string) (partIds []PartitionId, err error) {
files, err := ioutil.ReadDir(dataDirPath)
if err != nil {
return nil, err
}
partIds = make([]PartitionId, 0, len(files))
for _, file := range files {
if ok, _ := IsFileExists(path.Join(dataDirPath, file.Name(), lockFileName)); !ok {
continue
}
id, err := strconv.Atoi(file.Name())
if err != nil {
continue
}
partIds = append(partIds, PartitionId(id))
}
return
}
func IsFileExists(filePath string) (bool, error) {
_, err := os.Stat(filePath)
if err != nil {
return !os.IsNotExist(err), err
}
return true, nil
}