swapus-go/proc.go

71 lines
1.6 KiB
Go
Raw Permalink Normal View History

2019-06-25 05:29:49 +00:00
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package main
import (
"strings"
"regexp"
"io/ioutil"
"strconv"
)
type Proc struct {
Pid int
CmdLine string
Swap int
}
func NewProc(pid int) *Proc {
proc := new(Proc)
proc.Pid = pid
proc.CmdLine = CmdLine(pid)
proc.Swap = Swap(pid)
return proc
}
func CmdLine(pid int) string {
data, err := ioutil.ReadFile("/proc/" + strconv.Itoa(pid) + "/cmdline")
if err != nil {
return ""
}
cmdline := strings.Replace(string(data), "\x00", " ", -1)
return cmdline
}
func Swap(pid int) int {
data, err := ioutil.ReadFile("/proc/" + strconv.Itoa(pid) + "/status")
if err != nil {
return 0
}
r := regexp.MustCompile(`VmSwap:\s+(\d+) kB`)
match := r.FindSubmatch(data)
if len(match) == 0 {
return 0
}
swap, err := strconv.Atoi(string(match[1]))
if err != nil {
return 0
}
return swap
}