1 files changed, 7 insertions, 7 deletions
diff --git a/main.go b/main.go
index d9c4580..2b29286 100644
--- a/main.go
+++ b/main.go
@@ -23,24 +23,24 @@ var versionTxt string
 var router *mux.Router = mux.NewRouter()
 
 func main() {
-	run()
+	run(router, Addr, WriteTimeout, ReadTimeout, []syscall.Signal{syscall.SIGTERM, syscall.SIGINT})
 }
 
-func run() {
+func run(handler http.Handler, addr string, wTimeout time.Duration, rTimeout time.Duration, signals []syscall.Signal) {
 	log.Println("finserv finance server was started.")
 	defer log.Println("finserv finance server was stopped.")
 
 	server := &http.Server{
-		Handler:      router,
-		Addr:         Addr,
-		WriteTimeout: WriteTimeout,
-		ReadTimeout:  ReadTimeout,
+		Handler:      handler,
+		Addr:         addr,
+		WriteTimeout: wTimeout,
+		ReadTimeout:  rTimeout,
 	}
 
 	go server.ListenAndServe()
 	log.Printf("Serving at http://%s\n", Addr)
 
-	await(syscall.SIGTERM, syscall.SIGINT)
+	await(signals...)
 }
 
 func await(signals ...syscall.Signal) {