/
cpuinfo.go
77 lines (64 loc) · 1.5 KB
/
cpuinfo.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
package machine
import (
"regexp"
"strconv"
"strings"
)
func NewCpuInfoParserHelper(str string, err error) (*CpuInfo, error) {
if err != nil {
return nil, err
}
return NewCpuInfoParser(str)
}
var cpuinfoRegExp = regexp.MustCompile("([^:]*?)\\s*:\\s*(.*)$")
func NewCpuInfoParser(str string) (*CpuInfo, error) {
lines := strings.Split(str, "\n")
var cpuinfo = &CpuInfo{}
var processor = newProcessorInfo()
for i, line := range lines {
var key string
var value string
if len(line) == 0 && i != len(lines)-1 {
// end of processor
cpuinfo.Processors = append(cpuinfo.Processors, processor)
processor = newProcessorInfo()
continue
} else if i == len(lines)-1 {
continue
}
submatches := cpuinfoRegExp.FindStringSubmatch(line)
key = submatches[1]
value = submatches[2]
switch key {
case "processor":
processor.Id, _ = strconv.ParseInt(value, 10, 64)
case "physical id":
processor.PhysicalId, _ = strconv.ParseInt(value, 10, 64)
case "core id":
processor.CoreId, _ = strconv.ParseInt(value, 10, 64)
}
}
if processor.Id != -1 {
// valid but not append yet
cpuinfo.Processors = append(cpuinfo.Processors, processor)
}
return cpuinfo, nil
}
func newProcessorInfo() *ProcessorInfo {
return &ProcessorInfo{
Id: -1,
CoreId: -1,
PhysicalId: -1,
}
}
type CpuInfo struct {
Processors []*ProcessorInfo
}
func (c *CpuInfo) NumCPUs() int {
return len(c.Processors)
}
type ProcessorInfo struct {
Id int64
PhysicalId int64
CoreId int64
}