From 82138df48e178fa76d8e788e1373c5d336e187e2 Mon Sep 17 00:00:00 2001 From: Holger Obermaier <40787752+ho-ob@users.noreply.github.com> Date: Thu, 10 Feb 2022 09:28:06 +0100 Subject: [PATCH] Refactor: Replace readOneLine() by ioutil.ReadFile() --- collectors/cpufreqMetric.go | 42 +++++++++++----------------------- collectors/infinibandMetric.go | 16 +++++++++---- 2 files changed, 24 insertions(+), 34 deletions(-) diff --git a/collectors/cpufreqMetric.go b/collectors/cpufreqMetric.go index 5f47ce5..da9f2d7 100644 --- a/collectors/cpufreqMetric.go +++ b/collectors/cpufreqMetric.go @@ -1,10 +1,9 @@ package collectors import ( - "bufio" "encoding/json" "fmt" - "os" + "io/ioutil" "path/filepath" "strconv" "strings" @@ -15,23 +14,6 @@ import ( "golang.org/x/sys/unix" ) -// -// readOneLine reads one line from a file. -// It returns ok when file was successfully read. -// In this case text contains the first line of the files contents. -// -func readOneLine(filename string) (text string, ok bool) { - file, err := os.Open(filename) - if err != nil { - return - } - defer file.Close() - scanner := bufio.NewScanner(file) - ok = scanner.Scan() - text = scanner.Text() - return -} - type CPUFreqCollectorTopology struct { processor string // logical processor number (continuous, starting at 0) coreID string // socket local core ID @@ -105,10 +87,11 @@ func (m *CPUFreqCollector) Init(config json.RawMessage) error { // Read package ID physicalPackageIDFile := filepath.Join(cpuDir, "topology", "physical_package_id") - physicalPackageID, ok := readOneLine(physicalPackageIDFile) - if !ok { - return fmt.Errorf("Unable to read physical package ID from file '%s'", physicalPackageIDFile) + line, err := ioutil.ReadFile(physicalPackageIDFile) + if err != nil { + return fmt.Errorf("Unable to read physical package ID from file '%s': %v", physicalPackageIDFile, err) } + physicalPackageID := strings.TrimSpace(string(line)) physicalPackageID_int, err := strconv.ParseInt(physicalPackageID, 10, 64) if err != nil { return fmt.Errorf("Unable to convert packageID '%s' to int64: %v", physicalPackageID, err) @@ -116,10 +99,11 @@ func (m *CPUFreqCollector) Init(config json.RawMessage) error { // Read core ID coreIDFile := filepath.Join(cpuDir, "topology", "core_id") - coreID, ok := readOneLine(coreIDFile) - if !ok { - return fmt.Errorf("Unable to read core ID from file '%s'", coreIDFile) + line, err = ioutil.ReadFile(coreIDFile) + if err != nil { + return fmt.Errorf("Unable to read core ID from file '%s': %v", coreIDFile, err) } + coreID := strings.TrimSpace(string(line)) coreID_int, err := strconv.ParseInt(coreID, 10, 64) if err != nil { return fmt.Errorf("Unable to convert coreID '%s' to int64: %v", coreID, err) @@ -205,14 +189,14 @@ func (m *CPUFreqCollector) Read(interval time.Duration, output chan lp.CCMetric) } // Read current frequency - line, ok := readOneLine(t.scalingCurFreqFile) - if !ok { + line, err := ioutil.ReadFile(t.scalingCurFreqFile) + if err != nil { cclog.ComponentError( m.name, - fmt.Sprintf("Read(): Failed to read one line from file '%s'", t.scalingCurFreqFile)) + fmt.Sprintf("Read(): Failed to read file '%s': %v", t.scalingCurFreqFile, err)) continue } - cpuFreq, err := strconv.ParseInt(line, 10, 64) + cpuFreq, err := strconv.ParseInt(strings.TrimSpace(string(line)), 10, 64) if err != nil { cclog.ComponentError( m.name, diff --git a/collectors/infinibandMetric.go b/collectors/infinibandMetric.go index 4c7615b..e5197de 100644 --- a/collectors/infinibandMetric.go +++ b/collectors/infinibandMetric.go @@ -2,6 +2,7 @@ package collectors import ( "fmt" + "io/ioutil" "os" cclog "github.com/ClusterCockpit/cc-metric-collector/internal/ccLogger" @@ -68,8 +69,12 @@ func (m *InfinibandCollector) Init(config json.RawMessage) error { for _, path := range ibDirs { // Skip, when no LID is assigned - LID, ok := readOneLine(path + "/lid") - if !ok || LID == "0x0" { + line, err := ioutil.ReadFile(filepath.Join(path, "lid")) + if err != nil { + continue + } + LID := strings.TrimSpace(string(line)) + if LID == "0x0" { continue } @@ -142,13 +147,14 @@ func (m *InfinibandCollector) Read(interval time.Duration, output chan lp.CCMetr // device info info := &m.info[i] for counterName, counterFile := range info.portCounterFiles { - data, ok := readOneLine(counterFile) - if !ok { + line, err := ioutil.ReadFile(counterFile) + if err != nil { cclog.ComponentError( m.name, - fmt.Sprintf("Read(): Failed to read one line from file '%s'", counterFile)) + fmt.Sprintf("Read(): Failed to read from file '%s': %v", counterFile, err)) continue } + data := strings.TrimSpace(string(line)) v, err := strconv.ParseInt(data, 10, 64) if err != nil { cclog.ComponentError(