mirror of https://github.com/shirou/gopsutil
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
1.7 KiB
Go
106 lines
1.7 KiB
Go
// +build windows
|
|
|
|
package host
|
|
|
|
import (
|
|
"os"
|
|
"time"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/StackExchange/wmi"
|
|
|
|
common "github.com/shirou/gopsutil/common"
|
|
process "github.com/shirou/gopsutil/process"
|
|
)
|
|
|
|
var (
|
|
procGetSystemTimeAsFileTime = common.Modkernel32.NewProc("GetSystemTimeAsFileTime")
|
|
osInfo *Win32_OperatingSystem
|
|
)
|
|
|
|
type Win32_OperatingSystem struct {
|
|
Version string
|
|
Caption string
|
|
ProductType uint32
|
|
LastBootUpTime time.Time
|
|
}
|
|
|
|
func HostInfo() (*HostInfoStat, error) {
|
|
ret := &HostInfoStat{}
|
|
hostname, err := os.Hostname()
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
_, err = GetOSInfo()
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
ret.Hostname = hostname
|
|
ret.Uptime, err = BootTime()
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
// PlatformFamily
|
|
switch osInfo.ProductType {
|
|
case 1:
|
|
ret.PlatformFamily = "Desktop OS"
|
|
case 2:
|
|
ret.PlatformFamily = "Server OS (Domain Controller)"
|
|
case 3:
|
|
ret.PlatformFamily = "Server OS"
|
|
}
|
|
|
|
// Platform
|
|
ret.Platform = strings.Trim(osInfo.Caption, " ")
|
|
|
|
// Platform Version
|
|
ret.PlatformVersion = osInfo.Version
|
|
|
|
ret.OS = runtime.GOOS
|
|
|
|
procs, err := process.Pids()
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
ret.Procs = uint64(len(procs))
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func GetOSInfo() (Win32_OperatingSystem, error) {
|
|
var dst []Win32_OperatingSystem
|
|
q := wmi.CreateQuery(&dst, "")
|
|
err := wmi.Query(q, &dst)
|
|
if err != nil {
|
|
return Win32_OperatingSystem{}, err
|
|
}
|
|
|
|
osInfo = &dst[0]
|
|
|
|
return dst[0], nil
|
|
}
|
|
|
|
func BootTime() (uint64, error) {
|
|
if osInfo == nil {
|
|
_, err := GetOSInfo()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
now := time.Now()
|
|
t := osInfo.LastBootUpTime.Local()
|
|
return uint64(now.Sub(t).Seconds()), nil
|
|
}
|
|
|
|
func Users() ([]UserStat, error) {
|
|
|
|
var ret []UserStat
|
|
|
|
return ret, nil
|
|
}
|