From c8c6400944a880da424c97c36410767290771644 Mon Sep 17 00:00:00 2001 From: Emile Date: Fri, 18 Oct 2019 23:37:41 +0200 Subject: outsourced the creation of the database connection --- src/main.go | 26 +++++--------------------- 1 file changed, 5 insertions(+), 21 deletions(-) (limited to 'src') diff --git a/src/main.go b/src/main.go index 2c99242..38f254c 100644 --- a/src/main.go +++ b/src/main.go @@ -1,7 +1,6 @@ package main import ( - "database/sql" "flag" "fmt" "log" @@ -22,31 +21,16 @@ func main() { log.Printf("Running HTTP server on port %d", *port) httpServer := setupHTTPServer() + // Set up the DB connection + db := setupDatabase() + name := getName(db) + fmt.Println(name) + // React to system signals signalChannel := make(chan os.Signal, 1) signal.Notify(signalChannel) go cleanup(signalChannel, httpServer) - //////////////////////////////////////////////////////////////////////////////// - - connStr := "user=postgres dbname=postgres sslmode=disable" - db, err := sql.Open("postgres", connStr) - if err != nil { - log.Fatal(err) - } - - // get the current name of the challenges - query := fmt.Sprintf("SELECT name FROM challenges") - var name string - err = db.QueryRow(query).Scan(&name) - if err != nil { - log.Fatalf("[ E ] :", err) - } - - fmt.Println(name) - - //////////////////////////////////////////////////////////////////////////////// - // Start our HTTP server log.Fatalln(httpServer.ListenAndServe()) } -- cgit 1.4.1