Support simple connection strings without key pairs

FoundatioFx · Oct 31, 2024 · a5c7758 · a5c7758
1 parent 86f19c9
commit a5c7758
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/src/Foundatio/Utility/ConnectionStringParser.cs b/src/Foundatio/Utility/ConnectionStringParser.cs
@@ -29,7 +29,7 @@ public static class ConnectionStringParser
 
     private static readonly Regex _connectionStringRegex = new(ConnectionStringPattern, RegexOptions.ExplicitCapture | RegexOptions.Compiled | RegexOptions.IgnorePatternWhitespace);
 
-    private static Dictionary<string, string> Parse(string connectionString, IDictionary<string, string> synonyms)
+    private static Dictionary<string, string> Parse(string connectionString, IDictionary<string, string> synonyms, string defaultKey = null)
     {
         var parseTable = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
 
@@ -42,7 +42,15 @@ private static Dictionary<string, string> Parse(string connectionString, IDictio
 
         var match = _connectionStringRegex.Match(connectionString);
         if (!match.Success || (match.Length != connectionString.Length))
+        {
+            if (defaultKey != null)
+            {
+                parseTable[defaultKey] = connectionString;
+                return parseTable;
+            }
+
             throw new ArgumentException($"Format of the initialization string does not conform to specification starting at index {match.Length}");
+        }
 
         int indexValue = 0;
         var keyValues = match.Groups[valueIndex].Captures;
@@ -87,9 +95,9 @@ private static bool IsKeyNameValid(string keyName)
         return keyName[0] != ';' && !Char.IsWhiteSpace(keyName[0]) && keyName.IndexOf('\u0000') == -1;
     }
 
-    public static Dictionary<string, string> ParseConnectionString(this string connectionString, IDictionary<string, string> synonyms = null)
+    public static Dictionary<string, string> ParseConnectionString(this string connectionString, IDictionary<string, string> synonyms = null, string defaultKey = null)
     {
-        return Parse(connectionString, synonyms);
+        return Parse(connectionString, synonyms, defaultKey);
     }
 
     public static string BuildConnectionString(this IDictionary<string, string> options, IEnumerable<string> excludedKeys = null)

diff --git a/tests/Foundatio.Tests/Utility/ConnectionStringParserTests.cs b/tests/Foundatio.Tests/Utility/ConnectionStringParserTests.cs
@@ -42,6 +42,17 @@ public void CanParseQuotedConnectionString()
         Assert.Equal(connectionString, data.BuildConnectionString());
     }
 
+    [Fact]
+    public void CanParseSimpleConnectionString()
+    {
+        const string connectionString = "localhost,6379";
+        var data = connectionString.ParseConnectionString(defaultKey: "server");
+        Assert.Single(data);
+        Assert.Equal("localhost,6379", data["server"]);
+
+        Assert.Equal("server=localhost,6379", data.BuildConnectionString());
+    }
+
     [Fact]
     public void CanParseComplexQuotedConnectionString()
     {