Connect to postgresql

1 files changed, 54 insertions, 0 deletions

diff --git a/app/db/connection.go b/app/db/connection.go
new file mode 100644
index 0000000..d494f6c
--- /dev/null
+++ b/app/db/connection.go
@@ -0,0 +1,54 @@
+package db
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+
+	_ "github.com/lib/pq"
+	"gitlab.com/gitlab-org/software-supply-chain-security/authorization/sparkled/pkg/pls"
+)
+
+type Connection struct {
+	db *sql.DB
+}
+
+func NewConnection(databaseURL string) (*Connection, error) {
+	db, err := sql.Open("postgres", databaseURL)
+	if err != nil {
+		return nil, fmt.Errorf("failed to open database connection: %w", err)
+	}
+
+	return &Connection{
+		db: db,
+	}, nil
+}
+
+func (c *Connection) Ping(ctx context.Context) error {
+	if c.db == nil {
+		return fmt.Errorf("database connection not available")
+	}
+
+	return c.db.PingContext(ctx)
+}
+
+func (c *Connection) IsHealthy(ctx context.Context) bool {
+	if c.db == nil {
+		return false
+	}
+
+	err := c.Ping(ctx)
+	if err != nil {
+		pls.LogError(ctx, err)
+		return false
+	}
+
+	return true
+}
+
+func (c *Connection) Close() error {
+	if c.db == nil {
+		return nil
+	}
+	return c.db.Close()
+}