fix uncaught exception on bad raw text value being encoded

tsloughter · Nov 21, 2024 · 16782cc · 16782cc
1 parent 1033dbd
commit 16782cc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/src/pg_raw.erl b/src/pg_raw.erl
@@ -15,7 +15,7 @@ init(_Opts) ->
       <<"byteasend">>, <<"unknownsend">>,
       <<"citextsend">>], []}.
 
-encode(Text, _) ->
+encode(Text, _) when is_list(Text) ->
     [<<(iolist_size(Text)):?int32>>, Text].
 
 decode(Text, _) ->

diff --git a/src/pg_types.erl b/src/pg_types.erl
@@ -43,11 +43,17 @@ encode(Value, TypeInfo=#type_info{module=Module}) ->
     try
         Module:encode(Value, TypeInfo)
     catch
-        error:function_clause ->
-            erlang:error(#{error => badarg_encoding,
-                           module => ?MODULE,
-                           value => Value,
-                           type_info => TypeInfo})
+        error:function_clause:Stacktrace ->
+            erlang:raise(error, #{error => badarg_encoding,
+                                  module => ?MODULE,
+                                  value => Value,
+                                  type_info => TypeInfo}, Stacktrace);
+        error:badarg:Stacktrace ->
+            erlang:raise(error, #{error => badarg_encoding,
+                                  module => ?MODULE,
+                                  value => Value,
+                                  type_info => TypeInfo}, Stacktrace)
+
     end.
 
 -spec decode(binary(), type_info()) -> term().