p17860254
/
categraf
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

code refactor

This commit is contained in:
Ulric Qin 2022-04-14 23:10:14 +08:00
parent 88a454b27a
commit 4d34644cc7
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
inputs/system/system.go
View File

@ -27,6 +27,12 @@ type SystemStats struct {
CollectUserNumber bool
}
func init() {
inputs.Add(InputName, func() inputs.Input {
return &SystemStats{}
})
}
func (s *SystemStats) getInterval() time.Duration {
if s.IntervalSeconds != 0 {
return time.Duration(s.IntervalSeconds) * time.Second
@ -50,13 +56,14 @@ func (s *SystemStats) StartGoroutines(queue chan *types.Sample) {
}
func (s *SystemStats) LoopGather(queue chan *types.Sample) {
interval := s.getInterval()
for {
select {
case <-s.quit:
close(s.quit)
return
default:
time.Sleep(s.getInterval())
time.Sleep(interval)
defer func() {
if r := recover(); r != nil {
if strings.Contains(fmt.Sprint(r), "closed channel") {
@ -119,19 +126,9 @@ func (s *SystemStats) Gather(queue chan *types.Sample) {
if err == nil {
samples = append(samples, inputs.NewSample("n_users", float64(len(users))))
} else if os.IsNotExist(err) {
if config.Config.DebugMode {
log.Println("D! reading os users:", err)
}
log.Println("W! reading os users:", err)
} else if os.IsPermission(err) {
if config.Config.DebugMode {
log.Println("D! reading os users:", err)
}
log.Println("W! reading os users:", err)
}
}
}
func init() {
inputs.Add(InputName, func() inputs.Input {
return &SystemStats{}
})
}