96 lines
1.3 KiB
Go
96 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"runtime"
|
|
"runtime/pprof"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
logger *GoDNSLogger
|
|
)
|
|
|
|
func main() {
|
|
|
|
initLogger()
|
|
|
|
server := &Server{
|
|
host: settings.Server.Host,
|
|
port: settings.Server.Port,
|
|
rTimeout: 5 * time.Second,
|
|
wTimeout: 5 * time.Second,
|
|
}
|
|
|
|
server.Run()
|
|
|
|
logger.Info("godns %s start", settings.Version)
|
|
|
|
if settings.Debug {
|
|
go profileCPU()
|
|
go profileMEM()
|
|
}
|
|
|
|
sig := make(chan os.Signal)
|
|
signal.Notify(sig, os.Interrupt)
|
|
|
|
forever:
|
|
for {
|
|
select {
|
|
case <-sig:
|
|
logger.Info("signal received, stopping")
|
|
break forever
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func profileCPU() {
|
|
f, err := os.Create("godns.cprof")
|
|
if err != nil {
|
|
logger.Error("%s", err)
|
|
return
|
|
}
|
|
|
|
pprof.StartCPUProfile(f)
|
|
time.AfterFunc(6*time.Minute, func() {
|
|
pprof.StopCPUProfile()
|
|
f.Close()
|
|
|
|
})
|
|
}
|
|
|
|
func profileMEM() {
|
|
f, err := os.Create("godns.mprof")
|
|
if err != nil {
|
|
logger.Error("%s", err)
|
|
return
|
|
}
|
|
|
|
time.AfterFunc(5*time.Minute, func() {
|
|
pprof.WriteHeapProfile(f)
|
|
f.Close()
|
|
})
|
|
|
|
}
|
|
|
|
func initLogger() {
|
|
logger = NewLogger()
|
|
|
|
if settings.Log.Stdout {
|
|
logger.SetLogger("console", nil)
|
|
}
|
|
|
|
if settings.Log.File != "" {
|
|
config := map[string]interface{}{"file": settings.Log.File}
|
|
logger.SetLogger("file", config)
|
|
}
|
|
|
|
logger.SetLevel(settings.Log.LogLevel())
|
|
}
|
|
|
|
func init() {
|
|
runtime.GOMAXPROCS(runtime.NumCPU())
|
|
}
|