From 2953d6d0ed4e134f4a23358c0b14710c7da0d297 Mon Sep 17 00:00:00 2001 From: Andrew Cremins Date: Thu, 11 Dec 2025 14:22:48 -0800 Subject: [PATCH] Init --- pgutils/connector.go | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/pgutils/connector.go b/pgutils/connector.go index 9037fb7..21dce91 100644 --- a/pgutils/connector.go +++ b/pgutils/connector.go @@ -192,3 +192,23 @@ func OpenDB(conn *PostgresqlConnector) *sqlx.DB { sqlDB := sql.OpenDB(conn) return sqlx.NewDb(sqlDB, "postgres") } + +// ConnectDB opens a connection using the connector and verifies it with a ping +func ConnectDB(conn *PostgresqlConnector) (*sqlx.DB, error) { + db := OpenDB(conn) + if err := db.Ping(); err != nil { + db.Close() + return nil, err + } + return db, nil +} + +// MustConnectDB is like ConnectDB but panics on error +func MustConnectDB(conn *PostgresqlConnector) *sqlx.DB { + db, err := ConnectDB(conn) + if err != nil { + panic(err) + } + return db +} +