Add basic bitwise operators support

elixir-ecto · Nov 25, 2023 · d2ef848 · d2ef848
1 parent d5153e3
commit d2ef848
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/ecto/adapters/postgres/connection.ex b/lib/ecto/adapters/postgres/connection.ex
@@ -428,6 +428,10 @@ if Code.ensure_loaded?(Postgrex) do
       -: " - ",
       *: " * ",
       /: " / ",
+      &&&: " & ",
+      |||: " | ",
+      <<<: " << ",
+      >>>: " >> ",
       and: " AND ",
       or: " OR ",
       ilike: " ILIKE ",

diff --git a/test/ecto/adapters/postgres_test.exs b/test/ecto/adapters/postgres_test.exs
@@ -774,6 +774,18 @@ defmodule Ecto.Adapters.PostgresTest do
 
     query = Schema |> select([r], r.x + 2) |> plan()
     assert all(query) == ~s{SELECT s0."x" + 2 FROM "schema" AS s0}
+
+    query = Schema |> select([r], r.x &&& 2) |> plan()
+    assert all(query) == ~s{SELECT s0."x" & 2 FROM "schema" AS s0}
+
+    query = Schema |> select([r], r.x ||| 2) |> plan()
+    assert all(query) == ~s{SELECT s0."x" | 2 FROM "schema" AS s0}
+
+    query = Schema |> select([r], r.x <<< 2) |> plan()
+    assert all(query) == ~s{SELECT s0."x" << 2 FROM "schema" AS s0}
+
+    query = Schema |> select([r], r.x >>> 2) |> plan()
+    assert all(query) == ~s{SELECT s0."x" >> 2 FROM "schema" AS s0}
   end
 
   test "is_nil" do