feat: add a method to check if a scheme can be handled (#749)

Author: Seddryck

DubUrl.Core/Mapping/SchemeMapperBuilder.cs

@@ -15,43 +15,52 @@ namespace DubUrl.Mapping;
 
 public class SchemeMapperBuilder
 {
+    private char[] Separators = ['+', ':'];
+
     private readonly record struct ProviderInfo(string ProviderName, List<string> Aliases, Type DialectType, DriverLocatorFactory? DriverLocatorFactory);
     private bool IsBuilt { get; set; } = false;
 
     private List<MapperInfo> MapperData { get; } = new();
 
-    protected Dictionary<string, IMapper> Mappers { get; set; } = new();
-    private BaseMapperIntrospector[] MapperIntrospectors { get; } = new BaseMapperIntrospector[] { new NativeMapperIntrospector(), new WrapperMapperIntrospector() };
+    protected Dictionary<string, IMapper> Mappers { get; set; } = [];
+    private BaseMapperIntrospector[] MapperIntrospectors { get; } = [new NativeMapperIntrospector(), new WrapperMapperIntrospector()];
     private DialectBuilder DialectBuilder { get; } = new();
 
     public SchemeMapperBuilder()
-     : this(new[] { typeof(SchemeMapperBuilder).Assembly }) { }
+     : this([typeof(SchemeMapperBuilder).Assembly]) { }
 
     public SchemeMapperBuilder(Assembly[] assemblies)
     {
         var asmTypesProbe = new AssemblyTypesProbe(assemblies);
-        MapperIntrospectors = new BaseMapperIntrospector[]
-        {
+        MapperIntrospectors =
+        [
             new NativeMapperIntrospector(asmTypesProbe)
-            , new WrapperMapperIntrospector(asmTypesProbe)
-        };
+            ,
+            new WrapperMapperIntrospector(asmTypesProbe)
+        ];
         Initialize();
     }
 
     protected virtual void Initialize()
     {
-        foreach (var mapperData 
+        foreach (var mapperData
             in MapperIntrospectors.Aggregate(
                 Array.Empty<MapperInfo>(), (data, introspector)
                 => data.Concat(introspector.Locate()).ToArray())
         )
             AddMapping(mapperData);
     }
 
+    public bool CanHandle(string scheme)
+    {
+        var alias = GetAlias(scheme.Split(Separators));
+        return Mappers.ContainsKey(alias);
+    }
+
     public virtual void Build()
     {
         foreach (var mapperData in MapperData)
-            DialectBuilder.AddAliases(mapperData.DialectType, mapperData.Aliases.ToArray());
+            DialectBuilder.AddAliases(mapperData.DialectType, [.. mapperData.Aliases]);
         DialectBuilder.Build();
 
         Mappers.Clear();
@@ -107,7 +116,7 @@ private static string GetAlias(string[] aliases)
     }
 
     public IMapper GetMapper(string alias)
-        => GetMapper(new[] { alias });
+        => GetMapper(alias.Split(Separators));
 
     public virtual IMapper GetMapper(string[] aliases)
     {

DubUrl.Testing/Mapping/SchemeMapperBuilderTest.cs

@@ -65,17 +65,17 @@ public StubRewriter(DbConnectionStringBuilder csb)
     [TestCase("ts", typeof(TimescaleMapper))]
     [TestCase("quest", typeof(QuestDbMapper))]
     [TestCase("tr", typeof(TrinoMapper))]
-    //[TestCase("odbc", typeof(OdbcMapper))]
     [TestCase("odbc+mssql", typeof(OdbcMapper))]
     [TestCase("mssql+odbc", typeof(OdbcMapper))]
-    public void Instantiate_Scheme_CorrectType(string schemeList, Type expected)
+    public void Instantiate_Scheme_CorrectType(string scheme, Type expected)
     {
         var builder = new SchemeMapperBuilder();
         builder.Build();
-        var result = builder.GetMapper(schemeList.Split(new[] { '+', ':' }));
+        var result = builder.GetMapper(scheme.Split(['+', ':']));
 
         Assert.That(result, Is.Not.Null);
         Assert.That(result, Is.TypeOf(expected));
+        Assert.That(builder.CanHandle(scheme), Is.True);
     }
 
     [Test]