1// Copyright 2024 Google Inc. All rights reserved. 2// 3// Licensed under the Apache License, Version 2.0 (the "License"); 4// you may not use this file except in compliance with the License. 5// You may obtain a copy of the License at 6// 7// http://www.apache.org/licenses/LICENSE-2.0 8// 9// Unless required by applicable law or agreed to in writing, software 10// distributed under the License is distributed on an "AS IS" BASIS, 11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12// See the License for the specific language governing permissions and 13// limitations under the License. 14package metrics 15 16// This file contain code to extract host information on linux from 17// /proc/cpuinfo and /proc/meminfo relevant to machine performance 18 19import ( 20 "io/ioutil" 21 "strings" 22 23 "android/soong/finder/fs" 24) 25 26type fillable interface { 27 fillInfo(key, value string) 28} 29 30func NewCpuInfo(fileSystem fs.FileSystem) (*CpuInfo, error) { 31 c := &CpuInfo{} 32 if err := parseFile(c, "/proc/cpuinfo", true, fileSystem); err != nil { 33 return &CpuInfo{}, err 34 } 35 return c, nil 36} 37 38func NewMemInfo(fileSystem fs.FileSystem) (*MemInfo, error) { 39 m := &MemInfo{} 40 if err := parseFile(m, "/proc/meminfo", false, fileSystem); err != nil { 41 return &MemInfo{}, err 42 } 43 return m, nil 44} 45 46func parseFile(obj fillable, fileName string, endOnBlank bool, fileSystem fs.FileSystem) error { 47 fd, err := fileSystem.Open(fileName) 48 if err != nil { 49 return err 50 } 51 defer fd.Close() 52 53 data, err := ioutil.ReadAll(fd) 54 if err != nil { 55 return err 56 } 57 58 for _, l := range strings.Split(string(data), "\n") { 59 if !strings.Contains(l, ":") { 60 // Terminate after the first blank line. 61 if endOnBlank && strings.TrimSpace(l) == "" { 62 break 63 } 64 // If the line is not of the form "key: values", just skip it. 65 continue 66 } 67 68 kv := strings.SplitN(l, ":", 2) 69 obj.fillInfo(strings.TrimSpace(kv[0]), strings.TrimSpace(kv[1])) 70 } 71 return nil 72} 73