sql/sqlclient: add helper to check if driver was registered (#3115)

ariga · Sep 12, 2024 · c70602b · c70602b
1 parent 67c6213
commit c70602b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sql/sqlclient/client.go b/sql/sqlclient/client.go
@@ -298,6 +298,12 @@ func Open(ctx context.Context, s string, opts ...OpenOption) (*Client, error) {
 	return OpenURL(ctx, u, opts...)
 }
 
+// HasDriver reports if there is any driver registered with the given scheme.
+func HasDriver(scheme string) bool {
+	_, ok := drivers.Load(scheme)
+	return ok
+}
+
 // OpenURL opens an Atlas client by its provided url.URL.
 func OpenURL(ctx context.Context, u *url.URL, opts ...OpenOption) (*Client, error) {
 	cfg := &openOptions{}