mirror of
https://github.com/gogrlx/nats-server.git
synced 2026-04-02 11:48:43 -07:00
63 lines
1.0 KiB
Go
63 lines
1.0 KiB
Go
// Copyright 2015 Apcera Inc. All rights reserved.
|
|
|
|
package server
|
|
|
|
/*
|
|
#include <unistd.h>
|
|
*/
|
|
import "C"
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"syscall"
|
|
)
|
|
|
|
var procStatFile string
|
|
var ticks int64
|
|
|
|
const (
|
|
utimePos = 13
|
|
stimePos = 14
|
|
startPos = 21
|
|
vssPos = 22
|
|
rssPos = 23
|
|
)
|
|
|
|
func init() {
|
|
ticks = int64(C.sysconf(C._SC_CLK_TCK))
|
|
procStatFile = fmt.Sprintf("/proc/%d/stat", os.Getpid())
|
|
}
|
|
|
|
func procUsage(pcpu *float64, rss, vss *int64) error {
|
|
contents, err := ioutil.ReadFile(procStatFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fields := bytes.Fields(contents)
|
|
|
|
*rss = (parseInt64(fields[rssPos])) << 12
|
|
*vss = parseInt64(fields[vssPos])
|
|
|
|
startTime := parseInt64(fields[startPos])
|
|
utime := parseInt64(fields[utimePos])
|
|
stime := parseInt64(fields[stimePos])
|
|
totalTime := utime + stime
|
|
|
|
var sysinfo syscall.Sysinfo_t
|
|
if err := syscall.Sysinfo(&sysinfo); err != nil {
|
|
return err
|
|
}
|
|
|
|
seconds := sysinfo.Uptime - (startTime / ticks)
|
|
|
|
if seconds > 0 {
|
|
ipcpu := (totalTime * 1000 / ticks) / seconds
|
|
*pcpu = float64(ipcpu) / 10.0
|
|
}
|
|
|
|
return nil
|
|
}
|