Match on both quotes

elixir-ecto · Apr 7, 2023 · 366089a · 366089a
1 parent 4e0764b
commit 366089a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/ecto/adapters/myxql/connection.ex b/lib/ecto/adapters/myxql/connection.ex
@@ -44,10 +44,12 @@ if Code.ensure_loaded?(MyXQL) do
       MyXQL.stream(conn, sql, params, opts)
     end
 
+    @quotes ~w(" ')
+
     @impl true
     def to_constraints(%MyXQL.Error{mysql: %{name: :ER_DUP_ENTRY}, message: message}, opts) do
       with [_, quoted] <- :binary.split(message, " for key "),
-           [_, index | _] <- :binary.split(quoted, "'", [:global]) do
+           [_, index | _] <- :binary.split(quoted, @quotes, [:global]) do
         [unique: strip_source(index, opts[:source])]
       else
         _ -> []
@@ -57,7 +59,7 @@ if Code.ensure_loaded?(MyXQL) do
     def to_constraints(%MyXQL.Error{mysql: %{name: name}, message: message}, _opts)
         when name in [:ER_ROW_IS_REFERENCED_2, :ER_NO_REFERENCED_ROW_2] do
       with [_, quoted] <- :binary.split(message, [" CONSTRAINT ", " FOREIGN KEY "]),
-           [_, index | _] <- :binary.split(quoted, "'", [:global]) do
+           [_, index | _] <- :binary.split(quoted, @quotes, [:global]) do
         [foreign_key: index]
       else
         _ -> []