LBRYCommunity/tracker
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

improve comments and logging

Browse source
This commit is contained in:
Jimmy Zelinskie 2014-05-08 06:48:32 -04:00
parent 90491dc386
commit cf95b2d94b
2 changed files with 18 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
main.go
View file

@ -30,28 +30,34 @@ func main() {
flag.Parse()
runtime.GOMAXPROCS(runtime.NumCPU())
// Enable the profile if flagged.
if profile {
log.Println("Running with profiling enabled")
log.Println("running with profiling enabled")
f, err := os.Create("chihaya.cpu")
if err != nil {
log.Fatalf("Failed to create profile file: %s\n", err)
log.Fatalf("failed to create profile file: %s\n", err)
}
defer f.Close()
pprof.StartCPUProfile(f)
}
// Load the config file.
if configPath == "" {
log.Fatalf("Must specify a configuration file")
log.Fatalf("must specify a configuration file")
}
conf, err := config.Open(configPath)
if err != nil {
log.Fatalf("Failed to parse configuration file: %s\n", err)
log.Fatalf("failed to parse configuration file: %s\n", err)
}
log.Println("succesfully loaded config")
// Create a new server.
s, err := server.New(conf)
if err != nil {
log.Fatalf("Failed to create server: %s\n", err)
log.Fatalf("failed to create server: %s\n", err)
}
// Spawn a goroutine to handle interrupts and safely shut down.
go func() {
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
@ -61,18 +67,19 @@ func main() {
pprof.StopCPUProfile()
}
log.Println("Caught interrupt, shutting down..")
log.Println("caught interrupt, shutting down.")
err := s.Stop()
if err != nil {
panic("Failed to shutdown cleanly")
panic("failed to shutdown cleanly")
}
log.Println("Shutdown successfully")
log.Println("shutdown successfully")
<-c
os.Exit(0)
}()
// Start the server listening and handling requests.
err = s.ListenAndServe()
if err != nil {
log.Fatalf("Failed to start server: %s\n", err)
log.Fatalf("failed to start server: %s\n", err)
}
}

3
server/server.go
View file

@ -28,7 +28,7 @@ import (
type Server struct {
conf *config.Config
// These are open connections.
// These are open connections/pools.
listener *stoppableListener.StoppableListener
trackerPool tracker.Pool
backendConn backend.Conn
@ -130,6 +130,7 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
func fail(err error, w http.ResponseWriter, r *http.Request) {
errmsg := err.Error()
log.Println("handled failure: " + errmsg)
msg := "d14:failure reason" + strconv.Itoa(len(errmsg)) + ":" + errmsg + "e"
length, _ := io.WriteString(w, msg)
w.Header().Add("Content-Length", string(length))