max query size

Author: magnetised

c_src/libpg_query_ex.c

@@ -25,8 +25,11 @@ ERL_NIF_TERM make_binary(ErlNifEnv *env, char *source) {
   return binary;
 }
 
-static ERL_NIF_TERM parse_query(ErlNifEnv *env, int argc,
-                                const ERL_NIF_TERM argv[]) {
+static ERL_NIF_TERM max_query_size(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])  {
+  return enif_make_int64(env, MAX_QUERY_SIZE);
+}
+
+static ERL_NIF_TERM parse_query(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]) {
   ErlNifBinary query;
   ERL_NIF_TERM term;
 
@@ -178,7 +181,8 @@ static ERL_NIF_TERM scan_query(ErlNifEnv *env, int argc,
 static ErlNifFunc funcs[] = {
   {"parse_query", 1, parse_query},
   {"deparse_query", 1, deparse_query},
-  {"scan_query", 1, scan_query}
+  {"scan_query", 1, scan_query},
+  {"max_query_size", 0, max_query_size},
 };
 
 ERL_NIF_INIT(Elixir.PgQuery.Parser, funcs, NULL, NULL, NULL, NULL)

lib/pg_query.ex

@@ -1,5 +1,8 @@
 defmodule PgQuery do
   @type error() :: %{message: String.t(), cursorpos: non_neg_integer()}
+
+  defdelegate max_query_size(), to: PgQuery.Parser
+
   @doc """
   Parses the binary statement `stmt` into a Protobuf AST.
 

lib/pg_query/parser.ex

@@ -71,4 +71,5 @@ defmodule PgQuery.Parser do
   def parse_query(_query), do: :erlang.nif_error(:nif_not_loaded)
   def deparse_query(_encoded_proto), do: :erlang.nif_error(:nif_not_loaded)
   def scan_query(_query), do: :erlang.nif_error(:nif_not_loaded)
+  def max_query_size, do: :erlang.nif_error(:nif_not_loaded)
 end

test/pg_query_test.exs

@@ -75,4 +75,8 @@ defmodule PgQueryTest do
              token.token == :FROM
            end)
   end
+
+  test "max_query_size/0" do
+    assert 65536 = PgQuery.max_query_size()
+  end
 end