1 files changed, 8 insertions, 2 deletions
diff --git a/main.go b/main.go
index f68b7b7..8ed57bf 100644
--- a/main.go
+++ b/main.go
@@ -1,11 +1,13 @@
 package main
 
 import (
+	"context"
 	"log"
 	"net/http"
 	"os"
 
 	"github.com/gorilla/handlers"
+	"github.com/jackc/pgx/v5/pgxpool"
 	"github.com/joho/godotenv"
 
 	"gosten/model"
@@ -38,10 +40,14 @@ func main() {
 
 	checkEnv()
 
-	db := openDB(os.Getenv("GOSTEN_DSN"))
-	if err := db.Ping(); err != nil {
+	db, err := pgxpool.New(context.Background(), os.Getenv("GOSTEN_DSN"))
+	if err != nil {
 		log.Fatal(err)
 	}
+	if err := db.Ping(context.Background()); err != nil {
+		log.Fatal(err)
+	}
+	defer db.Close()
 
 	Q = model.New(db)