55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
// +build linux
|
|
|
|
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"runtime/debug"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
// Inject madvdontneed=1 as soon as possible.
|
|
|
|
var _ = func() struct{} {
|
|
// If GODEBUG is not empty, then use that instead of trying to exec our own.
|
|
if os.Getenv("GODEBUG") != "" {
|
|
return struct{}{}
|
|
}
|
|
|
|
// Do magic.
|
|
var environ = append(os.Environ(), "GODEBUG=madvdontneed=1")
|
|
|
|
log.Println("execve(2)ing with madvdontneed=1 for aggressive GC.")
|
|
|
|
path, err := exec.LookPath(os.Args[0])
|
|
if err != nil {
|
|
return struct{}{}
|
|
}
|
|
|
|
if err := syscall.Exec(path, os.Args, environ); err != nil {
|
|
log.Println("Error while executing:", err)
|
|
log.Println("Starting up without madvdontneed=1...")
|
|
return struct{}{}
|
|
}
|
|
|
|
os.Exit(0)
|
|
return struct{}{}
|
|
}()
|
|
|
|
func init() {
|
|
// Aggressive memory freeing you asked, so aggressive memory freeing we will
|
|
// deliver.
|
|
if strings.Contains(os.Getenv("GODEBUG"), "madvdontneed=1") {
|
|
go func() {
|
|
log.Println("Now attempting to free memory every 5s... (madvdontneed=1)")
|
|
for range time.Tick(5 * time.Second) {
|
|
debug.FreeOSMemory()
|
|
}
|
|
}()
|
|
}
|
|
}
|