-
Notifications
You must be signed in to change notification settings - Fork 315
/
common.go
90 lines (75 loc) · 2.24 KB
/
common.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
86
87
88
89
90
/*
Copyright 2022 The Koordinator Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package system
import (
"io"
"io/ioutil"
"os"
"path"
"strings"
"k8s.io/klog/v2"
)
var (
CommonRootDir = "" // for uni-test
)
func CommonFileRead(file string) (string, error) {
file = path.Join(CommonRootDir, file)
klog.V(5).Infof("read %s", file)
data, err := ioutil.ReadFile(file)
return strings.Trim(string(data), "\n"), err
}
func CommonFileWriteIfDifferent(file string, value string) error {
currentValue, err := CommonFileRead(file)
if err != nil {
return err
}
if value == currentValue {
klog.Infof("resource currentValue equal newValue, skip update resource! file:%s, value %s", file, value)
return nil
}
return CommonFileWrite(file, value)
}
func CommonFileWrite(file string, data string) error {
file = path.Join(CommonRootDir, file)
klog.V(5).Infof("write %s [%s]", file, data)
return ioutil.WriteFile(file, []byte(data), 0644)
}
// ReadFileNoStat uses ioutil.ReadAll to read contents of entire file.
// This is similar to ioutil.ReadFile but without the call to os.Stat, because
// many files in /proc and /sys report incorrect file sizes (either 0 or 4096).
// Reads a max file size of 512kB. For files larger than this, a scanner
// should be used.
func ReadFileNoStat(filename string) ([]byte, error) {
const maxBufferSize = 1024 * 512
f, err := os.Open(filename)
if err != nil {
return nil, err
}
defer f.Close()
reader := io.LimitReader(f, maxBufferSize)
return ioutil.ReadAll(reader)
}
func FileExists(path string) bool {
_, err := os.Stat(path)
return !os.IsNotExist(err)
}
func PathExists(path string) (bool, error) {
_, err := os.Stat(path)
if err == nil {
return true, nil
}
if os.IsNotExist(err) {
return false, nil
}
return false, err
}