chan os.Signal should be buffered

[godlighty.git] / cmd / godlighty / main.go

diff --git a/cmd/godlighty/main.go b/cmd/godlighty/main.go
index 4198753b354edb4411c94c105354fec7196bbfac..2373ed628a8f8b95322810dda56e6932107c11d8 100644 (file)
--- a/cmd/godlighty/main.go
+++ b/cmd/godlighty/main.go
@@ -1,6 +1,6 @@
 /*
 godlighty -- highly-customizable HTTP, HTTP/2, HTTPS server
-Copyright (C) 2021 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2021-2023 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -40,7 +40,10 @@ import (
 
 const MaxConns = 128
 
-var GracefulTime = 10 * time.Second
+var (
+       GracefulTime = 10 * time.Second
+       RWTimeout    = 30 * time.Second
+)
 
 func main() {
        bind := flag.String("bind", "[::]:80", "Address to bind and listen on")
@@ -54,10 +57,10 @@ func main() {
        if *doTLS {
                godlighty.LoadCertificates()
        }
-       shutdown := make(chan os.Signal)
+       shutdown := make(chan os.Signal, 1)
        signal.Notify(shutdown, syscall.SIGTERM, syscall.SIGINT, syscall.SIGHUP)
        exitErr := make(chan error)
-       l, err := net.Listen("tcp", *bind)
+       l, err := godlighty.DeadlinedListen("tcp", *bind, RWTimeout, RWTimeout)
        if err != nil {
                log.Fatalln(err)
        }
@@ -92,8 +95,8 @@ func main() {
                }
        }
 
-       info := make(chan os.Signal)
-       signal.Notify(info, syscall.SIGINFO)
+       info := make(chan os.Signal, 1)
+       signal.Notify(info, InfoSignal)
        go func() {
                for {
                        <-info
@@ -101,7 +104,12 @@ func main() {
                }
        }()
 
-       srv := http.Server{Handler: godlighty.MainHandler}
+       godlighty.BindAddr = *bind
+       srv := http.Server{
+               Handler:           godlighty.MainHandler,
+               ReadHeaderTimeout: RWTimeout,
+               IdleTimeout:       time.Minute,
+       }
        go func() {
                <-shutdown
                log.Println("shutting down")