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