utils.go

package main

import (
	"github.com/shirou/gopsutil/cpu"
	"log"
	"os"
	"os/exec"
	"os/signal"
	"os/user"
	"runtime"
	"syscall"
	"time"
)

func memstats() {
	var memstats runtime.MemStats
	if os.Getenv("DEBUG_MEMSTATS") != "1" {
		return
	}
	go func() {
		for {
			runtime.ReadMemStats(&memstats)
			log.Printf("memstats: %+v", memstats)
			time.Sleep(10 * time.Second)
		}
	}()
}

const RLIMIT_MEMLOCK = 8 // nolint

func rlimit() {
	var rLimit syscall.Rlimit
	err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit)
	if err != nil {
		log.Fatal("error getting RLIMIT_NOFILE: ", err)
	}
	log.Printf("NOFILE: %+v", rLimit)
	err = syscall.Getrlimit(RLIMIT_MEMLOCK, &rLimit)
	if err != nil {
		log.Fatal("error getting RLIMIT_MEMLOCK: ", err)
	}
	log.Printf("MEMLOCK: %+v", rLimit)
}

func whoami() {
	u, _ := user.Current()
	log.Printf("Who am I: %+v", u)
}

func top() {

	time.Sleep(30 * time.Second)

	var gracefulStop = make(chan os.Signal)

	signal.Notify(gracefulStop, syscall.SIGTERM)
	signal.Notify(gracefulStop, syscall.SIGINT)

	cmd := exec.Command("top", "-b", "-d30")
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	cmd.Start()

	<-gracefulStop

	cmd.Process.Kill()

	cmd.Wait()

	os.Exit(0)

}

func cpuinfo() {
	info, err := cpu.Info()
	if err != nil {
		log.Print("can't get cpu information: ", err)
		return
	}
	log.Printf("Running on %s %.1fMhz", info[0].ModelName, info[0].Mhz)
}

func swapon() {
	log.Print("# swapon -s:")
	cmd := exec.Command("swapon", "-s")
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	cmd.Start()
	cmd.Wait()
}