log.go

package main

import (
	"log"
	"os"
	"os/user"
	"path"
	"strings"
)

var (
	linit bool = false
	lfd   *os.File
)

func InitLog() error {
	if !opt.debug {
		return nil
	}

	u, err := user.Current()
	if err != nil {
		return err
	}

	f := path.Join(u.HomeDir, ".procstat.log")
	lfd, err := os.OpenFile(f, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
	if err != nil {
		return err
	}

	log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
	log.SetOutput(lfd)

	linit = true
	dbg(strings.Repeat("=", 20))
	dbg(lfd.Name())

	return nil
}

func CleanupLog() {
	if !opt.debug {
		return
	}

	lfd.Close()
	linit = false
}

func dbg(args ...interface{}) {
	if !opt.debug {
		return
	}

	Assert(linit)
	GlobalLock()
	log.Println(args...)
	GlobalUnlock()
}

func dbgf(f string, args ...interface{}) {
	if !opt.debug {
		return
	}

	Assert(linit)
	GlobalLock()
	log.Printf(f, args...)
	GlobalUnlock()
}