Merge pull request #145 from ssepml/jiffy_binary

Ensure 'jiffy:encode/1' returns a binary
potatosalad · Dec 7, 2023 · c53977d · c53977d
2 parents b4df4f2 + 173dcc4
commit c53977d
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/json/jose_json_jiffy.erl b/src/json/jose_json_jiffy.erl
@@ -23,11 +23,11 @@ decode(Binary) ->
 	jiffy:decode(Binary, [return_maps]).
 
 encode(Map) when is_map(Map) ->
-	jiffy:encode(sort(Map));
+	ensure_binary(jiffy:encode(sort(Map)));
 encode(List) when is_list(List) ->
-	jiffy:encode(sort(List));
+	ensure_binary(jiffy:encode(sort(List)));
 encode(Term) ->
-	jiffy:encode(Term).
+	ensure_binary(jiffy:encode(Term)).
 
 %%%-------------------------------------------------------------------
 %%% Internal functions
@@ -40,3 +40,11 @@ sort(List) when is_list(List) ->
 	[sort(Term) || Term <- List];
 sort(Term) ->
 	Term.
+
+%% @private
+%% NOTE: jiffy may return an iolist instead of a binary when encoding
+%%       big objects.
+ensure_binary(List) when is_list(List) ->
+	iolist_to_binary(List);
+ensure_binary(Binary) ->
+	Binary.