From e259b416b2c7fcd04bc2057de32938549e8d06e8 Mon Sep 17 00:00:00 2001 From: ale <ale@incal.net> Date: Thu, 11 Jan 2018 07:53:10 +0000 Subject: [PATCH] Add command-line option to set request timeout --- cmd/auth-server/main.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cmd/auth-server/main.go b/cmd/auth-server/main.go index a0d017ff..c93644c1 100644 --- a/cmd/auth-server/main.go +++ b/cmd/auth-server/main.go @@ -8,6 +8,7 @@ import ( "os" "os/signal" "syscall" + "time" "git.autistici.org/ai3/go-common/unix" "github.com/coreos/go-systemd/daemon" @@ -21,6 +22,7 @@ var ( socketPath = flag.String("socket", "/run/auth/socket", "`path` to the UNIX socket to listen on") systemdSocketActivation = flag.Bool("systemd-socket", false, "use SystemD socket activation") httpAddr = flag.String("http-addr", "", "if not nil, bind an HTTP server to this `addr` for Prometheus metrics") + requestTimeout = flag.Duration("timeout", 5*time.Second, "timeout for incoming requests") ) func usage() { @@ -58,6 +60,7 @@ func main() { } srv := unix.NewLineServer(server.NewSocketServer(authSrv)) + srv.RequestTimeout = *requestTimeout var sockSrv *unix.SocketServer if *systemdSocketActivation { -- GitLab