Initial commit

app/internal/models/db.go

@@ -0,0 +1,31 @@
+package models
+
+import (
+	"database/sql"
+	"fmt"
+	"log"
+
+	_ "github.com/lib/pq"
+)
+
+var DB *sql.DB
+
+func InitDB() {
+	var err error
+
+	// Example DSN format for PostgreSQL:
+	// "postgres://username:password@host:port/dbname?sslmode=disable"
+	dsn := "postgres://mannpatel:Admin@localhost:5432/poll_database?sslmode=disable"
+
+	DB, err = sql.Open("postgres", dsn)
+	if err != nil {
+		log.Fatalf("Failed to connect to DB: %v", err)
+	}
+
+	err = DB.Ping()
+	if err != nil {
+		log.Fatalf("Failed to ping DB: %v", err)
+	}
+
+	fmt.Println("Database connection successful")
+}