fixup! feat: sqlconnect library

rudderlabs · Mar 6, 2024 · bd7e747 · bd7e747
1 parent c8aac4e
commit bd7e747
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/sqlconnect/internal/trino/db.go b/sqlconnect/internal/trino/db.go
@@ -1,6 +1,7 @@
 package trino
 
 import (
+	"context"
 	"database/sql"
 	"encoding/json"
 	"fmt"
@@ -72,3 +73,12 @@ func init() {
 type DB struct {
 	*base.DB
 }
+
+func (db *DB) Ping() error {
+	return db.PingContext(context.Background())
+}
+
+func (db *DB) PingContext(ctx context.Context) error {
+	_, err := db.ExecContext(ctx, "select 1")
+	return err
+}