fix unit test.

pingcap · Sep 12, 2023 · 92dcfd9 · 92dcfd9
1 parent cdca12b
commit 92dcfd9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/sink/sink_type.go b/pkg/sink/sink_type.go
@@ -13,6 +13,11 @@
 
 package sink
 
+import (
+	"net/url"
+	"strings"
+)
+
 // Type is the type of sink.
 type Type int
 
@@ -91,3 +96,8 @@ func IsStorageScheme(scheme string) bool {
 func IsPulsarScheme(scheme string) bool {
 	return scheme == PulsarScheme || scheme == PulsarSSLScheme
 }
+
+// GetScheme returns the scheme of the url.
+func GetScheme(url *url.URL) string {
+	return strings.ToLower(url.Scheme)
+}