Fix :invalid_chunk_size error

Author: warmwaffles

CHANGELOG.md

@@ -1,6 +1,12 @@
 # Changelog
 
 
+## [0.5.3] - 2021-03-23
+
+### Fixed
+- `:invalid_chunk_size` being emitted by the `DBConnection.execute`
+
+
 ## [0.5.2] - 2021-03-23
 
 ### Added

README.md

@@ -31,7 +31,7 @@ Package: https://hex.pm/packages/exqlite
 
 ```elixir
 defp deps do
-  {:exqlite, "~> 0.5.1"}
+  {:exqlite, "~> 0.5.3"}
 end
 ```
 

lib/exqlite/connection.ex

@@ -33,7 +33,8 @@ defmodule Exqlite.Connection do
     :db,
     :path,
     :transaction_status,
-    :status
+    :status,
+    :chunk_size
   ]
 
   @type t() :: %__MODULE__{
@@ -88,13 +89,20 @@ defmodule Exqlite.Connection do
       negative value turns auto-checkpointing off.
     * `:busy_timeout` - Sets the busy timeout in milliseconds for a connection.
       Default is `2000`.
+    * `:chunk_size` - The chunk size for bulk fetching. Defaults to `50`.
 
   For more information about the options above, see [sqlite documenation][1]
 
   [1]: https://www.sqlite.org/pragma.html
   """
   def connect(options) do
     database = Keyword.get(options, :database)
+    options =
+      Keyword.put_new(
+        options,
+        :chunk_size,
+        Application.get_env(:exqlite, :default_chunk_size, 50)
+      )
 
     case database do
       nil ->
@@ -384,7 +392,8 @@ defmodule Exqlite.Connection do
         db: db,
         path: path,
         transaction_status: :idle,
-        status: :idle
+        status: :idle,
+        chunk_size: Keyword.get(options, :chunk_size)
       }
 
       {:ok, state}
@@ -491,7 +500,7 @@ defmodule Exqlite.Connection do
   end
 
   defp get_rows(query, state) do
-    case Sqlite3.fetch_all(state.db, query.ref) do
+    case Sqlite3.fetch_all(state.db, query.ref, state.chunk_size) do
       {:ok, rows} -> {:ok, rows}
       {:error, reason} -> {:error, %Error{message: reason}, state}
     end

lib/exqlite/sqlite3.ex

@@ -118,13 +118,19 @@ defmodule Exqlite.Sqlite3 do
     Sqlite3NIF.execute(conn, String.to_charlist("PRAGMA shrink_memory"))
   end
 
+  @spec fetch_all(db(), statement(), integer()) :: {:ok, [row()]} | {:error, reason()}
+  def fetch_all(conn, statement, chunk_size) do
+    fetch_all(conn, statement, chunk_size, [])
+  end
+
   @spec fetch_all(db(), statement()) :: {:ok, [row()]} | {:error, reason()}
-  def fetch_all(conn, statement, chunk_size \\ 50) do
+  def fetch_all(conn, statement) do
     # TODO: Should this be done in the NIF? It can be _much_ faster to build a
     # list there, but at the expense that it could block other dirty nifs from
     # getting work done.
     #
     # For now this just works
+    chunk_size = Application.get_env(:exqlite, :default_chunk_size, 50)
     fetch_all(conn, statement, chunk_size, [])
   end
 

mix.exs

@@ -1,7 +1,7 @@
 defmodule Exqlite.MixProject do
   use Mix.Project
 
-  @version "0.5.2"
+  @version "0.5.3"
 
   def project do
     [