Merge branch 'sparkle-visibility' into 'main'

Connect to postgresql

See merge request gitlab-org/software-supply-chain-security/authorization/sparkled!21

2 files changed, 73 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()
+}
diff --git a/app/db/url.go b/app/db/url.go
new file mode 100644
index 0000000..b17c651
--- /dev/null
+++ b/app/db/url.go
@@ -0,0 +1,19 @@
+package db
+
+import (
+	"fmt"
+
+	"github.com/xlgmokha/x/pkg/env"
+)
+
+func URL() string {
+	if url := env.Fetch("DATABASE_URL", ""); url != "" {
+		return url
+	}
+
+	return fmt.Sprintf(
+		"postgresql://postgres:%s@localhost:5000/%s?sslmode=disable",
+		env.Fetch("RUNWAY_PG_USER_POSTGRES_PASSWORD_SPARKLE", ""),
+		env.Fetch("DATABASE_NAME", "sparkle"),
+	)
+}