diff --git a/eqc/json_eqc.erl b/eqc/json_eqc.erl new file mode 100644 index 0000000000..999024c31b --- /dev/null +++ b/eqc/json_eqc.erl @@ -0,0 +1,56 @@ +-module(json_eqc). + +-include_lib("eqc/include/eqc.hrl"). +-compile([export_all, nowarn_export_all]). + +prop_roundtrip() -> + in_parallel(?FORALL(Obj, object(), equals(Obj, decode(encode(Obj))))). + +prop_escape_all() -> + in_parallel( + ?FORALL(Str, object(), + begin + Encoded = iolist_to_binary(encode_escape_all(Str)), + all_ascii(Encoded) andalso equals(Str, decode(Encoded)) + end)). + +printable_string() -> + Chars = oneof([ + oneof("\n\r\t\v\b\f\e\d"), + choose(16#20, 16#7E), + choose(16#A0, 16#D7FF), + choose(16#E000, 16#FFFD), + choose(16#10000, 16#10FFFF) + ]), + ?LET(L, list(Chars), unicode:characters_to_binary(L)). + +object() -> ?SIZED(Size, object(Size)). + +object(0) -> + oneof([integer(), real(), printable_string(), bool(), null]); +object(Size) -> + ?LAZY(oneof([object(0), + ?LETSHRINK(Objects, list(object(Size div 4)), Objects), + map(printable_string(), object(Size div 4)) + ])). + +integer() -> choose(-16#8fffffff, 16#ffffffff). + +%% Helpers + +all_ascii(Bin) -> + lists:all(fun(C) -> C < 128 end, binary_to_list(Bin)). + +decode(IOList) -> + riak_kv_wm_json:decode(iolist_to_binary(IOList)). + +%% May produce an iolist +encode(Str) -> + riak_kv_wm_json:encode(Str). + +encode_escape_all(Term) -> + Encode = fun + (Binary, _) when is_binary(Binary) -> riak_kv_wm_json:encode_binary_escape_all(Binary); + (Other, Encode) -> riak_kv_wm_json:encode_value(Other, Encode) + end, + riak_kv_wm_json:encode(Term, Encode). diff --git a/include/riak_kv_wm_json.hrl b/include/riak_kv_wm_json.hrl new file mode 100644 index 0000000000..c88bd2ba34 --- /dev/null +++ b/include/riak_kv_wm_json.hrl @@ -0,0 +1,189 @@ +%% +%% %CopyrightBegin% +%% +%% Copyright Ericsson AB 2024. All Rights Reserved. +%% +%% Licensed under the Apache License, Version 2.0 (the "License"); +%% you may not use this file except in compliance with the License. +%% You may obtain a copy of the License at +%% +%% http://www.apache.org/licenses/LICENSE-2.0 +%% +%% Unless required by applicable law or agreed to in writing, software +%% distributed under the License is distributed on an "AS IS" BASIS, +%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +%% See the License for the specific language governing permissions and +%% limitations under the License. +%% +%% %CopyrightEnd% +%% + +%% A lot of the macros below use multi-value comparisons where +%% range checks would have worked just fine. This is because +%% the compiler & JIT can emit better code in some cases when +%% multiple clauses are to be dispatched based on such sets +%% of values. They'll generate an efficient "jump table", +%% which gets to the correct clause in one go, rather +%% than going through a set of comparisons. +%% However, this might not always be the bext way (see is_0_to_9), +%% so as always with any performance work - measure, don't guess! + +-define(is_1_to_9(X), + X =:= $1 orelse + X =:= $2 orelse + X =:= $3 orelse + X =:= $4 orelse + X =:= $5 orelse + X =:= $6 orelse + X =:= $7 orelse + X =:= $8 orelse + X =:= $9 +). + +-define(is_0_to_9(X), X >= $0 andalso X =< $9). + +-define(is_ws(X), X =:= $\s; X =:= $\t; X =:= $\r; X =:= $\n). + +-define(is_ascii_escape(Byte), + Byte =:= 0 orelse + Byte =:= 1 orelse + Byte =:= 2 orelse + Byte =:= 3 orelse + Byte =:= 4 orelse + Byte =:= 5 orelse + Byte =:= 6 orelse + Byte =:= 7 orelse + Byte =:= 8 orelse + Byte =:= 9 orelse + Byte =:= 10 orelse + Byte =:= 11 orelse + Byte =:= 12 orelse + Byte =:= 13 orelse + Byte =:= 14 orelse + Byte =:= 15 orelse + Byte =:= 16 orelse + Byte =:= 17 orelse + Byte =:= 18 orelse + Byte =:= 19 orelse + Byte =:= 20 orelse + Byte =:= 21 orelse + Byte =:= 22 orelse + Byte =:= 23 orelse + Byte =:= 24 orelse + Byte =:= 25 orelse + Byte =:= 26 orelse + Byte =:= 27 orelse + Byte =:= 28 orelse + Byte =:= 29 orelse + Byte =:= 30 orelse + Byte =:= 31 orelse + Byte =:= 34 orelse + Byte =:= 92 +). +-define(is_ascii_plain(Byte), + Byte =:= 32 orelse + Byte =:= 33 orelse + Byte =:= 35 orelse + Byte =:= 36 orelse + Byte =:= 37 orelse + Byte =:= 38 orelse + Byte =:= 39 orelse + Byte =:= 40 orelse + Byte =:= 41 orelse + Byte =:= 42 orelse + Byte =:= 43 orelse + Byte =:= 44 orelse + Byte =:= 45 orelse + Byte =:= 46 orelse + Byte =:= 47 orelse + Byte =:= 48 orelse + Byte =:= 49 orelse + Byte =:= 50 orelse + Byte =:= 51 orelse + Byte =:= 52 orelse + Byte =:= 53 orelse + Byte =:= 54 orelse + Byte =:= 55 orelse + Byte =:= 56 orelse + Byte =:= 57 orelse + Byte =:= 58 orelse + Byte =:= 59 orelse + Byte =:= 60 orelse + Byte =:= 61 orelse + Byte =:= 62 orelse + Byte =:= 63 orelse + Byte =:= 64 orelse + Byte =:= 65 orelse + Byte =:= 66 orelse + Byte =:= 67 orelse + Byte =:= 68 orelse + Byte =:= 69 orelse + Byte =:= 70 orelse + Byte =:= 71 orelse + Byte =:= 72 orelse + Byte =:= 73 orelse + Byte =:= 74 orelse + Byte =:= 75 orelse + Byte =:= 76 orelse + Byte =:= 77 orelse + Byte =:= 78 orelse + Byte =:= 79 orelse + Byte =:= 80 orelse + Byte =:= 81 orelse + Byte =:= 82 orelse + Byte =:= 83 orelse + Byte =:= 84 orelse + Byte =:= 85 orelse + Byte =:= 86 orelse + Byte =:= 87 orelse + Byte =:= 88 orelse + Byte =:= 89 orelse + Byte =:= 90 orelse + Byte =:= 91 orelse + Byte =:= 93 orelse + Byte =:= 94 orelse + Byte =:= 95 orelse + Byte =:= 96 orelse + Byte =:= 97 orelse + Byte =:= 98 orelse + Byte =:= 99 orelse + Byte =:= 100 orelse + Byte =:= 101 orelse + Byte =:= 102 orelse + Byte =:= 103 orelse + Byte =:= 104 orelse + Byte =:= 105 orelse + Byte =:= 106 orelse + Byte =:= 107 orelse + Byte =:= 108 orelse + Byte =:= 109 orelse + Byte =:= 110 orelse + Byte =:= 111 orelse + Byte =:= 112 orelse + Byte =:= 113 orelse + Byte =:= 114 orelse + Byte =:= 115 orelse + Byte =:= 116 orelse + Byte =:= 117 orelse + Byte =:= 118 orelse + Byte =:= 119 orelse + Byte =:= 120 orelse + Byte =:= 121 orelse + Byte =:= 122 orelse + Byte =:= 123 orelse + Byte =:= 124 orelse + Byte =:= 125 orelse + Byte =:= 126 orelse + Byte =:= 127 +). + +-define(are_all_ascii_plain(B1, B2, B3, B4, B5, B6, B7, B8), + (?is_ascii_plain(B1)) andalso + (?is_ascii_plain(B2)) andalso + (?is_ascii_plain(B3)) andalso + (?is_ascii_plain(B4)) andalso + (?is_ascii_plain(B5)) andalso + (?is_ascii_plain(B6)) andalso + (?is_ascii_plain(B7)) andalso + (?is_ascii_plain(B8)) +). \ No newline at end of file diff --git a/priv/riak_kv.schema b/priv/riak_kv.schema index ebb1d57e9b..091a3da491 100644 --- a/priv/riak_kv.schema +++ b/priv/riak_kv.schema @@ -1517,6 +1517,29 @@ {default, disabled} ]}. +%% @doc Default Secondary Index Timeout (milliseconds) +%% Set the default timeout for running secondary index queries in millisceconds +%% This is the time to run the query, and collate the results - the encoding of +%% those results for dispatch back to the client is outside of the scope of +%% this timeout. Setting to 0 (default) implies no timeout. +%% On hitting the timeout a 503 error will be returned on the HTTP API +{mapping, "secondary_index_timeout", "riak_kv.secondary_index_timeout", [ + {datatype, integer}, + {default, 0} +]}. + +%% @doc Json library for 2i results +%% Prior to Riak KV 3.2.1 the mochijson2 library was used to json encode 2i +%% results. For larger results, the performance of this library is slower +%% when compared to alternatives (e.g. otp). A back-ported version of the +%% OTP 27 proposed Json library is now the default - but may return keys in a +%% different order (i.e. the continuation key may now be before the results +%% key). To reverse back to mochijson, then reset to mochijson here. +{mapping, "secondary_index_json", "riak_kv.secondary_index_json", [ + {datatype, {enum, [otp, mochijson]}}, + {default, otp} +]}. + %% @doc For $key index queries, should keys which are tombstones be returned. %% This config will only make a difference with the leveled backend, it is %% ignored on other backends. Disable to change default behaviour and stop @@ -1524,4 +1547,4 @@ {mapping, "dollarkey_readtombs", "riak_kv.dollarkey_readtombs", [ {datatype, {flag, enabled, disabled}}, {default, enabled} -]}. \ No newline at end of file +]}. diff --git a/src/riak_client.erl b/src/riak_client.erl index e0f414a200..860949284f 100644 --- a/src/riak_client.erl +++ b/src/riak_client.erl @@ -1187,8 +1187,10 @@ wait_for_query_results(ReqId, Timeout) -> wait_for_query_results(ReqId, Timeout, Acc) -> receive {ReqId, done} -> {ok, lists:flatten(lists:reverse(Acc))}; - {ReqId,{results, Res}} -> wait_for_query_results(ReqId, Timeout, [Res | Acc]); - {ReqId, Error} -> {error, Error} + {ReqId, {results, Res}} -> + wait_for_query_results(ReqId, Timeout, [Res | Acc]); + {ReqId, {error, Error}} -> {error, Error}; + {ReqId, UnexpectedMsg} -> {error, UnexpectedMsg} after Timeout -> {error, timeout} end. @@ -1200,7 +1202,8 @@ wait_for_query_results(ReqId, Timeout, Acc) -> wait_for_fold_results(ReqId, Timeout) -> receive {ReqId, {results, Results}} -> {ok, Results}; - {ReqId, Error} -> {error, Error} + {ReqId, {error, Error}} -> {error, Error}; + {ReqId, UnexpectedMsg} -> {error, UnexpectedMsg} after Timeout -> {error, timeout} end. diff --git a/src/riak_index.erl b/src/riak_index.erl index 0d68f6953d..f0c2986c19 100644 --- a/src/riak_index.erl +++ b/src/riak_index.erl @@ -58,7 +58,7 @@ %% See GH610, this default is for backwards compat, so 2i behaves as %% it did before the FSM timeout bug was "fixed" --define(DEFAULT_TIMEOUT, infinity). +-define(DEFAULT_TIMEOUT, 0). %% @type data_type_defs() = [data_type_def()]. %% @type data_type_def() = {MatchFunction :: function(), ParseFunction :: function()}. @@ -503,17 +503,16 @@ decode_continuation(undefined) -> decode_continuation(Bin) -> binary_to_term(base64:decode(Bin)). -%% @doc add the `timeout' option tuple to the -%% `Opts' proplist. if `Timeout' is undefined -%% then the `app.config' property -%% `{riak_kv, seconady_index timeout}' is used -%% If that config property is defined, then -%% a default of `infinity' is used. -%% We use `infinity' as the default to -%% match the behavior pre 1.4 +%% @doc add the `timeout' option tuple to the `Opts' proplist. if `Timeout' is +%% undefined then the `app.config' property `{riak_kv, seconady_index timeout}' +%% is used. If that config property is undefined, then a default of `infinity' +%% (a timeout of 0 will be translated to infinity) is used. +%% We use `infinity' as the default to match the behavior pre 1.4 add_timeout_opt(undefined, Opts) -> - Timeout = app_helper:get_env(riak_kv, secondary_index_timeout, ?DEFAULT_TIMEOUT), - [{timeout, Timeout} | Opts]; + Timeout = + app_helper:get_env( + riak_kv, secondary_index_timeout, ?DEFAULT_TIMEOUT), + add_timeout_opt(Timeout, Opts); add_timeout_opt(0, Opts) -> [{timeout, infinity} | Opts]; add_timeout_opt(Timeout, Opts) -> diff --git a/src/riak_kv_clusteraae_fsm.erl b/src/riak_kv_clusteraae_fsm.erl index 2cbc230c40..b0d4b58146 100644 --- a/src/riak_kv_clusteraae_fsm.erl +++ b/src/riak_kv_clusteraae_fsm.erl @@ -1059,4 +1059,3 @@ convert_validate_test() -> [IQ1, IQ2, IQ3, IQ4, IQ5, IQ6, IQ7]). -endif. - diff --git a/src/riak_kv_wm_index.erl b/src/riak_kv_wm_index.erl index fcd2bbca68..1aa54bc60d 100644 --- a/src/riak_kv_wm_index.erl +++ b/src/riak_kv_wm_index.erl @@ -56,16 +56,24 @@ %% webmachine resource exports -export([ - init/1, - service_available/2, - is_authorized/2, - forbidden/2, - malformed_request/2, - content_types_provided/2, - encodings_provided/2, - resource_exists/2, - produce_index_results/2 - ]). + init/1, + service_available/2, + is_authorized/2, + forbidden/2, + malformed_request/2, + content_types_provided/2, + encodings_provided/2, + resource_exists/2, + produce_index_results/2 +]). +-export([ + mochijson_encode_results/2, + mochijson_encode_results/3, + otp_encode_results/2, + otp_encode_results/3, + results_encode/2, + keys_encode/2 +]). -record(ctx, { client, %% riak_client() - the store client @@ -82,6 +90,8 @@ }). -type context() :: #ctx{}. +-define(DEFAULT_JSON_ENCODING, otp). + -include_lib("kernel/include/logger.hrl"). -include_lib("webmachine/include/webmachine.hrl"). @@ -482,12 +492,12 @@ handle_all_in_memory_index_query(RD, Ctx) -> %% Do the index lookup... case riak_client:get_index(Bucket, Query, Opts, Client) of {ok, Results} -> - Continuation = make_continuation(MaxResults, - Results, - length(Results)), - JsonResults = encode_results(ReturnTerms, - Results, - Continuation), + Continuation = + make_continuation( + MaxResults, Results, length(Results)), + JsonResults = + encode_results( + ReturnTerms, Results, Continuation), {JsonResults, RD, Ctx}; {error, timeout} -> {{halt, 503}, @@ -502,18 +512,38 @@ handle_all_in_memory_index_query(RD, Ctx) -> end. -encode_results(ReturnTerms, Results) -> - encode_results(ReturnTerms, Results, undefined). +%% @doc Like `lists:last/1' but doesn't choke on an empty list +-spec last_result([] | list()) -> term() | undefined. +last_result([]) -> + undefined; +last_result(L) -> + lists:last(L). + +%% @doc if this is a paginated query make a continuation +-spec make_continuation(Max::non_neg_integer() | undefined, + list(), + ResultCount::non_neg_integer()) -> binary() | undefined. +make_continuation(MaxResults, Results, MaxResults) -> + riak_index:make_continuation(Results); +make_continuation(_, _, _) -> + undefined. + +%% =================================================================== +%% JSON Encoding implementations +%% =================================================================== + +mochijson_encode_results(ReturnTerms, Results) -> + mochijson_encode_results(ReturnTerms, Results, undefined). -encode_results(true, Results, Continuation) -> +mochijson_encode_results(true, Results, Continuation) -> JsonKeys2 = {struct, [{?Q_RESULTS, [{struct, [{Val, Key}]} || {Val, Key} <- Results]}] ++ - mochify_continuation(Continuation)}, + mochify_continuation(Continuation)}, mochijson2:encode(JsonKeys2); -encode_results(false, Results, Continuation) -> +mochijson_encode_results(false, Results, Continuation) -> JustTheKeys = filter_values(Results), JsonKeys1 = {struct, [{?Q_KEYS, JustTheKeys}] ++ mochify_continuation(Continuation)}, mochijson2:encode(JsonKeys1). - + mochify_continuation(undefined) -> []; mochify_continuation(Continuation) -> @@ -526,18 +556,207 @@ filter_values([{_, _} | _T]=Results) -> filter_values(Results) -> Results. -%% @doc Like `lists:last/1' but doesn't choke on an empty list --spec last_result([] | list()) -> term() | undefined. -last_result([]) -> - undefined; -last_result(L) -> - lists:last(L). -%% @doc if this is a paginated query make a continuation --spec make_continuation(Max::non_neg_integer() | undefined, - list(), - ResultCount::non_neg_integer()) -> binary() | undefined. -make_continuation(MaxResults, Results, MaxResults) -> - riak_index:make_continuation(Results); -make_continuation(_, _, _) -> - undefined. +otp_encode_results(ReturnTerms, Results) -> + otp_encode_results(ReturnTerms, Results, undefined). + +otp_encode_results(true, Results, undefined) -> + riak_kv_wm_json:encode( + #{?Q_RESULTS_BIN => Results}, + fun results_encode/2 + ); +otp_encode_results(true, Results, Continuation) -> + riak_kv_wm_json:encode( + #{?Q_RESULTS_BIN => Results, + ?Q_2I_CONTINUATION_BIN => Continuation}, + fun results_encode/2 + ); +otp_encode_results(false, Results, undefined) -> + riak_kv_wm_json:encode( + #{?Q_KEYS_BIN => Results}, + fun keys_encode/2 + ); +otp_encode_results(false, Results, Continuation) -> + riak_kv_wm_json:encode( + #{?Q_KEYS_BIN => Results, + ?Q_2I_CONTINUATION_BIN => Continuation}, + fun keys_encode/2 + ). + +results_encode({Term, Key}, Encode) when is_binary(Term), is_binary(Key) -> + ["{", [Encode(Term, Encode), $: | Encode(Key, Encode)], "}"]; +results_encode({Term, Key}, Encode) when is_integer(Term), is_binary(Key) -> + ["{", + [Encode(integer_to_binary(Term), Encode), $: | Encode(Key, Encode)], + "}"]; +results_encode(Result, Encode) -> + riak_kv_wm_json:encode_value(Result, Encode). + +keys_encode({_Term, Key}, Encode) when is_binary(Key) -> + riak_kv_wm_json:encode_value(Key, Encode); +keys_encode(Object, Encode) -> + riak_kv_wm_json:encode_value(Object, Encode). + +encode_results(ReturnTerms, Results) -> + encode_results(ReturnTerms, Results, undefined). + +encode_results(ReturnTerms, Results, Continuation) -> + Library = + app_helper:get_env( + riak_kv, secondary_index_json, ?DEFAULT_JSON_ENCODING), + case Library of + mochijson -> + mochijson_encode_results(ReturnTerms, Results, Continuation); + otp -> + otp_encode_results(ReturnTerms, Results, Continuation) + end. + + + +%% =================================================================== +%% EUnit tests +%% =================================================================== + +-ifdef(TEST). + +-include_lib("eunit/include/eunit.hrl"). + + +otp_decode(JsonIOL) -> + riak_kv_wm_json:decode(iolist_to_binary(JsonIOL)). + +compare_encode_test() -> + Results = large_results(10), + OTPjsonA = otp_encode_results(true, Results), + MjsonA = mochijson_encode_results(true, Results), + ?assert(mochijson2:decode(MjsonA) == mochijson2:decode(OTPjsonA)), + ?assert(otp_decode(MjsonA) == otp_decode(OTPjsonA)), + ?assert(iolist_to_binary(MjsonA) == iolist_to_binary(OTPjsonA)), + Continuation = make_continuation(10, Results, 10), + OTPjsonB = otp_encode_results(true, Results, Continuation), + MjsonB = mochijson_encode_results(true, Results, Continuation), + {struct, MDecodeOTPjsonB} = mochijson2:decode(OTPjsonB), + {struct, MDecodeMjsonB} = mochijson2:decode(MjsonB), + ?assert(lists:sort(MDecodeOTPjsonB) == lists:sort(MDecodeMjsonB)), + OTPjsonC = otp_encode_results(false, Results), + MjsonC = mochijson_encode_results(false, Results), + ?assert(mochijson2:decode(MjsonC) == mochijson2:decode(OTPjsonC)), + OTPjsonD = otp_encode_results(false, Results, Continuation), + MjsonD = mochijson_encode_results(false, Results, Continuation), + {struct, MDecodeOTPjsonD} = mochijson2:decode(OTPjsonD), + {struct, MDecodeMjsonD} = mochijson2:decode(MjsonD), + ?assert(lists:sort(MDecodeOTPjsonD) == lists:sort(MDecodeMjsonD)), + IntIdxResults = [{1, <<"K1">>}, {2, <<"K2">>}], + OTPjsonE = otp_encode_results(true, IntIdxResults), + MjsonE = mochijson_encode_results(true, IntIdxResults), + ?assert(mochijson2:decode(MjsonE) == mochijson2:decode(OTPjsonE)), + ?assert(otp_decode(MjsonE) == otp_decode(OTPjsonE)), + ?assert(iolist_to_binary(MjsonE) == iolist_to_binary(OTPjsonE)) + + . + +encoder_test_() -> + {timeout, 600, fun encode_tester/0}. + +encode_tester() -> + timer:sleep(100), % awkward silence to tidy screen output + encode_implementation_tester(mochijson2), + encode_implementation_tester(otp). + +encode_implementation_tester(Imp) -> + garbage_collect(), + + io:format(user, "~n~nTesting Implementation ~w~n", [Imp]), + ResultSetsTiny = + [{<<"1K">>, large_results(1000)}, + {<<"2K">>, large_results(2000)}, + {<<"3K">>, large_results(3000)}, + {<<"5K">>, large_results(5000)}, + {<<"8K">>, large_results(8000)}], + encode_tester(Imp, ResultSetsTiny, microseconds), + + garbage_collect(), + + ResultSetsSmall = + [{<<"13K">>, large_results(13000)}, + {<<"21K">>, large_results(21000)}, + {<<"34K">>, large_results(34000)}, + {<<"55K">>, large_results(55000)}], + encode_tester(Imp, ResultSetsSmall, milliseconds), + + garbage_collect(), + + ResultSetsMid = + [{<<"100K">>, large_results(100000)}, + {<<"200K">>, large_results(200000)}, + {<<"300K">>, large_results(300000)}, + {<<"500K">>, large_results(500000)}], + encode_tester(Imp, ResultSetsMid, milliseconds), + + % garbage_collect(), + + % ResultSetsLarge = + % [{<<"1M">>, large_results(1000000)}, + % {<<"2M">>, large_results(2000000)}, + % {<<"3M">>, large_results(3000000)}, + % {<<"5M">>, large_results(5000000)}], + % encode_tester(Imp, ResultSetsLarge, milliseconds), + + % garbage_collect(), + % ResultSetsHuge = + % [{<<"8M">>, large_results(8000000)}], + % encode_tester(Imp, ResultSetsHuge, milliseconds), + + % garbage_collect(), + % ResultSetsSuperSize = + % [{<<"13M">>, large_results(13000000)}], + % encode_tester(Imp, ResultSetsSuperSize, milliseconds), + + ok. + +encode_tester(Lib, ResultSets, Unit) -> + Divisor = + case Unit of + microseconds -> + 1; + milliseconds -> + 1000 + end, + Fun = + case Lib of + mochijson2 -> + fun mochijson_encode_results/2; + otp -> + fun otp_encode_results/2 + end, + + TotalTime = + lists:sum( + lists:map( + fun({Tag, RS}) -> + garbage_collect(), + {TC, _Json} = + timer:tc(fun() -> Fun(true, RS) end), + io:format( + user, + "Result set of ~s in ~w ~p ", + [Tag, TC div Divisor, Unit]), + TC + end, + ResultSets + ) + ), + io:format(user, "Total time ~w ~p~n", [TotalTime div Divisor, Unit]). + +large_results(N) -> + lists:map( + fun(I) -> {generate_term(I), generate_key(I)} end, + lists:seq(1, N)). + +generate_term(I) -> + iolist_to_binary(io_lib:format("q~9..0B", [I])). + +generate_key(K) -> + iolist_to_binary(io_lib:format("k~9..0B", [rand:uniform(K)])). + +-endif. \ No newline at end of file diff --git a/src/riak_kv_wm_json.erl b/src/riak_kv_wm_json.erl new file mode 100644 index 0000000000..8330f72272 --- /dev/null +++ b/src/riak_kv_wm_json.erl @@ -0,0 +1,1142 @@ +%% +%% %CopyrightBegin% +%% +%% Copyright Ericsson AB 2024-2024. All Rights Reserved. +%% +%% Licensed under the Apache License, Version 2.0 (the "License"); +%% you may not use this file except in compliance with the License. +%% You may obtain a copy of the License at +%% +%% http://www.apache.org/licenses/LICENSE-2.0 +%% +%% Unless required by applicable law or agreed to in writing, software +%% distributed under the License is distributed on an "AS IS" BASIS, +%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +%% See the License for the specific language governing permissions and +%% limitations under the License. +%% +%% %CopyrightEnd% +%% % @format +%% +-module(riak_kv_wm_json). + +%% @doc +%% A library for encoding and decoding JSON. +%% +%% This module implements EEP68. +%% +%% Both encoder and decoder fully conform to +%% [RFC 8259](https://tools.ietf.org/html/rfc8259) and +%% [ECMA 404](http://www.ecma-international.org/publications/standards/Ecma-404.htm) +%% standards. The decoder is tested using [JSONTestSuite](https://github.com/nst/JSONTestSuite). + +-dialyzer(no_improper_lists). + +-export([ + encode/1, encode/2, + encode_value/2, + encode_atom/2, + encode_integer/1, + encode_float/1, + encode_list/2, + encode_map/2, + encode_map_checked/2, + encode_key_value_list/2, + encode_key_value_list_checked/2, + encode_binary/1, + encode_binary_escape_all/1 +]). +-export_type([encoder/0, encode_value/0]). + +-export([ + decode/1, decode/3 +]). +-export_type([ + from_binary_fun/0, + array_start_fun/0, + array_push_fun/0, + array_finish_fun/0, + object_start_fun/0, + object_push_fun/0, + object_finish_fun/0, + decoders/0, + decode_value/0 +]). + +-compile(warn_missing_spec). + +-compile({inline, [ + encode_atom/2, + encode_integer/1, + encode_float/1, + encode_object/1, + escape/1, + escape_binary/1, + escape_all/1, + utf8t/0, + utf8s/0, + utf8s0/0, + hex_to_int/4, + string/6 +]}). + +%% =================================================================== +%% Compatibility changes +%% =================================================================== + +-include("riak_kv_wm_json.hrl"). + +%% @doc +%% For backwards compatibility some additional helper functions required: +%% - use of float_to_binary/2 needs to be replaced with format_float, that may +%% not return reliable results in OTP versions < 25 +%% - use of error/3 needs to be replaced with extended_error/3 which will not +%% provide additional information about position in OTP versions < 24 +%% - the OTP 26 type dynamic() is replaced with term() in earlier versions +%% +%% As -doc is not supported pre OTP 27, these doc references have been +%% commented +%% +%% The do_encode_map/2 function uses a map comprehension, and as this is not +%% supported prior to OTP 26, this is replaced with a map:fold/3 + +-if(?OTP_RELEASE < 26). +-type dynamic() :: term(). +-endif. + +-if(?OTP_RELEASE >= 25). +format_float(Float) -> float_to_binary(Float, [short]). +-else. +% Results may be unpredictable for encioding floats +% Bodged as no float encoding required in Riak +format_float(Float) -> float_to_binary(Float). +-endif. + +-spec extended_error( + {atom(), binary()}, none, non_neg_integer()) -> no_return(). +-if(?OTP_RELEASE >= 24). +extended_error(Error, none, Position) -> + error(Error, none, error_info(Position)). + +error_info(Skip) -> + [{error_info, #{cause => #{position => Skip}}}]. +-else. +extended_error(Error, none, _Position) -> + error(Error, none). +-endif. + +-if(?OTP_RELEASE >= 26). +do_encode_map(Map, Encode) when is_function(Encode, 2) -> + encode_object([[$,, key(Key, Encode), $: | Encode(Value, Encode)] || Key := Value <- Map]). +-else. +do_encode_map(Map, Encode) when is_function(Encode, 2) -> + encode_object( + maps:fold( + fun(Key, Value, Acc) -> + [[$,, key(Key, Encode), $: | Encode(Value, Encode)]|Acc] + end, + [], + Map) + ). +-endif. + +%% =================================================================== + + +-define(UTF8_ACCEPT, 0). +-define(UTF8_REJECT, 12). + +%% +%% Encoding implementation +%% + +-type encoder() :: fun((dynamic(), encoder()) -> iodata()). + +% -doc """ +% Simple JSON value encodeable with `json:encode/1`. +% """. +-type encode_value() :: + integer() + | float() + | boolean() + | null + | binary() + | atom() + | list(encode_value()) + | encode_map(encode_value()). + +-type encode_map(Value) :: #{binary() | atom() | integer() => Value}. + +% -doc """ +% Generates JSON corresponding to `Term`. + +% Supports basic data mapping: + +% | **Erlang** | **JSON** | +% |------------------------|----------| +% | `integer() \| float()` | Number | +% | `true \| false ` | Boolean | +% | `null` | Null | +% | `binary()` | String | +% | `atom()` | String | +% | `list()` | Array | +% | `#{binary() => _}` | Object | +% | `#{atom() => _}` | Object | +% | `#{integer() => _}` | Object | + +% This is equivalent to `encode(Term, fun json:encode_value/2)`. + +% ## Examples + +% ```erlang +% > iolist_to_binary(json:encode(#{foo => <<"bar">>})). +% <<"{\"foo\":\"bar\"}">> +% ``` +% """. +-spec encode(encode_value()) -> iodata(). +encode(Term) -> encode(Term, fun do_encode/2). + +% -doc """ +% Generates JSON corresponding to `Term`. + +% Can be customised with the `Encoder` callback. +% The callback will be recursively called for all the data +% to be encoded and is expected to return the corresponding +% encoded JSON as iodata. + +% Various `encode_*` functions in this module can be used +% to help in constructing such callbacks. + +% ## Examples + +% An encoder that uses a heuristic to differentiate object-like +% lists of key-value pairs from plain lists: + +% ```erlang +% > encoder([{_, _} | _] = Value, Encode) -> json:encode_key_value_list(Value, Encode); +% > encoder(Other, Encode) -> json:encode_value(Other, Encode). +% > custom_encode(Value) -> json:encode(Value, fun(Value, Encode) -> encoder(Value, Encode) end). +% > iolist_to_binary(custom_encode([{a, []}, {b, 1}])). +% <<"{\"a\":[],\"b\":1}">> +% ``` +% """. +-spec encode(dynamic(), encoder()) -> iodata(). +encode(Term, Encoder) when is_function(Encoder, 2) -> + Encoder(Term, Encoder). + +% -doc """ +% Default encoder used by `json:encode/1`. + +% Recursively calls `Encode` on all the values in `Value`. +% """. +-spec encode_value(dynamic(), encoder()) -> iodata(). +encode_value(Value, Encode) -> + do_encode(Value, Encode). + +-spec do_encode(dynamic(), encoder()) -> iodata(). +do_encode(Value, Encode) when is_atom(Value) -> + encode_atom(Value, Encode); +do_encode(Value, _Encode) when is_binary(Value) -> + escape_binary(Value); +do_encode(Value, _Encode) when is_integer(Value) -> + encode_integer(Value); +do_encode(Value, _Encode) when is_float(Value) -> + encode_float(Value); +do_encode(Value, Encode) when is_list(Value) -> + do_encode_list(Value, Encode); +do_encode(Value, Encode) when is_map(Value) -> + do_encode_map(Value, Encode); +do_encode(Other, _Encode) -> + error({unsupported_type, Other}). + +% -doc """ +% Default encoder for atoms used by `json:encode/1`. + +% Encodes the atom `null` as JSON `null`, +% atoms `true` and `false` as JSON booleans, +% and everything else as JSON strings calling the `Encode` +% callback with the corresponding binary. +% """. +-spec encode_atom(atom(), encoder()) -> iodata(). +encode_atom(null, _Encode) -> <<"null">>; +encode_atom(true, _Encode) -> <<"true">>; +encode_atom(false, _Encode) -> <<"false">>; +encode_atom(Other, Encode) -> Encode(atom_to_binary(Other, utf8), Encode). + +% -doc """ +% Default encoder for integers as JSON numbers used by `json:encode/1`. +% """. +-spec encode_integer(integer()) -> iodata(). +encode_integer(Integer) -> integer_to_binary(Integer). + +% -doc """ +% Default encoder for floats as JSON numbers used by `json:encode/1`. +% """. +-spec encode_float(float()) -> iodata(). +encode_float(Float) -> format_float(Float). + +% -doc """ +% Default encoder for lists as JSON arrays used by `json:encode/1`. +% """. +-spec encode_list(list(), encoder()) -> iodata(). +encode_list(List, Encode) when is_list(List) -> + do_encode_list(List, Encode). + +do_encode_list([], _Encode) -> + <<"[]">>; +do_encode_list([First | Rest], Encode) when is_function(Encode, 2) -> + [$[, Encode(First, Encode) | list_loop(Rest, Encode)]. + +list_loop([], _Encode) -> "]"; +list_loop([Elem | Rest], Encode) -> [$,, Encode(Elem, Encode) | list_loop(Rest, Encode)]. + +% -doc """ +% Default encoder for maps as JSON objects used by `json:encode/1`. + +% Accepts maps with atom, binary, integer, or float keys. +% """. +-spec encode_map(encode_map(dynamic()), encoder()) -> iodata(). +encode_map(Map, Encode) when is_map(Map) -> + do_encode_map(Map, Encode). + +% -doc """ +% Encoder for maps as JSON objects. + +% Accepts maps with atom, binary, integer, or float keys. +% Verifies that no duplicate keys will be produced in the +% resulting JSON object. + +% ## Errors + +% Raises `error({duplicate_key, Key})` if there are duplicates. +% """. +-spec encode_map_checked(map(), encoder()) -> iodata(). +encode_map_checked(Map, Encode) -> + do_encode_checked(maps:to_list(Map), Encode). + +% -doc """ +% Encoder for lists of key-value pairs as JSON objects. + +% Accepts lists with atom, binary, integer, or float keys. +% """. +-spec encode_key_value_list([{term(), term()}], encoder()) -> iodata(). +encode_key_value_list(List, Encode) when is_function(Encode, 2) -> + encode_object([[$,, key(Key, Encode), $: | Encode(Value, Encode)] || {Key, Value} <- List]). + +% -doc """ +% Encoder for lists of key-value pairs as JSON objects. + +% Accepts lists with atom, binary, integer, or float keys. +% Verifies that no duplicate keys will be produced in the +% resulting JSON object. + +% ## Errors + +% Raises `error({duplicate_key, Key})` if there are duplicates. +% """. +-spec encode_key_value_list_checked([{term(), term()}], encoder()) -> iodata(). +encode_key_value_list_checked(List, Encode) -> + do_encode_checked(List, Encode). + +do_encode_checked(List, Encode) when is_function(Encode, 2) -> + do_encode_checked(List, Encode, #{}). + +do_encode_checked([{Key, Value} | Rest], Encode, Visited0) -> + EncodedKey = iolist_to_binary(key(Key, Encode)), + case is_map_key(EncodedKey, Visited0) of + true -> + error({duplicate_key, Key}); + _ -> + Visited = Visited0#{EncodedKey => true}, + [$,, EncodedKey, $:, Encode(Value, Encode) | do_encode_checked(Rest, Encode, Visited)] + end; +do_encode_checked([], _, _) -> + []. + +%% Dispatching any value through `Encode` could allow incorrect +%% JSON to be emitted (with keys not being strings). To avoid this, +%% the default encoder only supports binaries, atoms, and numbers. +%% Customisation is possible by overriding how maps are encoded in general. +key(Key, Encode) when is_binary(Key) -> Encode(Key, Encode); +key(Key, Encode) when is_atom(Key) -> Encode(atom_to_binary(Key, utf8), Encode); +key(Key, _Encode) when is_integer(Key) -> [$", encode_integer(Key), $"]; +key(Key, _Encode) when is_float(Key) -> [$", encode_float(Key), $"]. + +encode_object([]) -> <<"{}">>; +encode_object([[_Comma | Entry] | Rest]) -> ["{", Entry, Rest, "}"]. + +% -doc """ +% Default encoder for binaries as JSON strings used by `json:encode/1`. + +% ## Errors + +% * `error(unexpected_end)` if the binary contains incomplete UTF-8 sequences. +% * `error({invalid_byte, Byte})` if the binary contains invalid UTF-8 sequences. +% """. +-spec encode_binary(binary()) -> iodata(). +encode_binary(Bin) when is_binary(Bin) -> + escape_binary(Bin). + +% -doc """ +% Encoder for binaries as JSON strings producing pure-ASCII JSON. + +% For any non-ASCII unicode character, a corresponding `\\uXXXX` sequence is used. + +% ## Errors + +% * `error(unexpected_end)` if the binary contains incomplete UTF-8 sequences. +% * `error({invalid_byte, Byte})` if the binary contains invalid UTF-8 sequences. +% """. +-spec encode_binary_escape_all(binary()) -> iodata(). +encode_binary_escape_all(Bin) when is_binary(Bin) -> + escape_all(Bin). + +escape_binary(Bin) -> escape_binary_ascii(Bin, [$"], Bin, 0, 0). + +escape_binary_ascii(Binary, Acc, Orig, Skip, Len) -> + case Binary of + <> when ?are_all_ascii_plain(B1, B2, B3, B4, B5, B6, B7, B8) -> + escape_binary_ascii(Rest, Acc, Orig, Skip, Len + 8); + Other -> + escape_binary(Other, Acc, Orig, Skip, Len) + end. + +escape_binary(<>, Acc, Orig, Skip, Len) when ?is_ascii_plain(Byte) -> + %% we got here because there were either less than 8 bytes left + %% or we have an escape in the next 8 bytes, + %% escape_binary_ascii would fail and dispatch here anyway + escape_binary(Rest, Acc, Orig, Skip, Len + 1); +escape_binary(<>, Acc, Orig, Skip0, Len) when ?is_ascii_escape(Byte) -> + Escape = escape(Byte), + Skip = Skip0 + Len + 1, + case Len of + 0 -> + escape_binary_ascii(Rest, [Acc | Escape], Orig, Skip, 0); + _ -> + Part = binary_part(Orig, Skip0, Len), + escape_binary_ascii(Rest, [Acc, Part | Escape], Orig, Skip, 0) + end; +escape_binary(<>, Acc, Orig, Skip, Len) -> + case element(Byte - 127, utf8s0()) of + ?UTF8_REJECT -> invalid_byte(Orig, Skip + Len); + %% all accept cases are ASCII, already covred above + State -> escape_binary_utf8(Rest, Acc, Orig, Skip, Len, State) + end; +escape_binary(_, _Acc, Orig, 0, _Len) -> + [$", Orig, $"]; +escape_binary(_, Acc, _Orig, _Skip, 0) -> + [Acc, $"]; +escape_binary(_, Acc, Orig, Skip, Len) -> + Part = binary_part(Orig, Skip, Len), + [Acc, Part, $"]. + +escape_binary_utf8(<>, Acc, Orig, Skip, Len, State0) -> + Type = element(Byte + 1, utf8t()), + case element(State0 + Type, utf8s()) of + ?UTF8_ACCEPT -> escape_binary_ascii(Rest, Acc, Orig, Skip, Len + 2); + ?UTF8_REJECT -> invalid_byte(Orig, Skip + Len + 1); + State -> escape_binary_utf8(Rest, Acc, Orig, Skip, Len + 1, State) + end; +escape_binary_utf8(_, _Acc, Orig, Skip, Len, _State) -> + unexpected(Orig, Skip + Len + 1). + +escape_all(Bin) -> escape_all_ascii(Bin, [$"], Bin, 0, 0). + +escape_all_ascii(Binary, Acc, Orig, Skip, Len) -> + case Binary of + <> when ?are_all_ascii_plain(B1, B2, B3, B4, B5, B6, B7, B8) -> + escape_all_ascii(Rest, Acc, Orig, Skip, Len + 8); + Other -> + escape_all(Other, Acc, Orig, Skip, Len) + end. + +escape_all(<>, Acc, Orig, Skip, Len) when ?is_ascii_plain(Byte) -> + escape_all(Rest, Acc, Orig, Skip, Len + 1); +escape_all(<>, Acc, Orig, Skip, Len) when ?is_ascii_escape(Byte) -> + Escape = escape(Byte), + case Len of + 0 -> + escape_all(Rest, [Acc | Escape], Orig, Skip + 1, 0); + _ -> + Part = binary_part(Orig, Skip, Len), + escape_all(Rest, [Acc, Part | Escape], Orig, Skip + Len + 1, 0) + end; +escape_all(<>, Acc, Orig, Skip, 0) -> + escape_char(Rest, Acc, Orig, Skip, Char); +escape_all(<>, Acc, Orig, Skip, Len) -> + Part = binary_part(Orig, Skip, Len), + escape_char(Rest, [Acc | Part], Orig, Skip + Len, Char); +escape_all(<<>>, _Acc, Orig, 0, _Len) -> + [$", Orig, $"]; +escape_all(<<>>, Acc, _Orig, _Skip, 0) -> + [Acc, $"]; +escape_all(<<>>, Acc, Orig, Skip, Len) -> + Part = binary_part(Orig, Skip, Len), + [Acc, Part, $"]; +escape_all(_Other, _Acc, Orig, Skip, Len) -> + invalid_byte(Orig, Skip + Len). + +escape_char(<>, Acc, Orig, Skip, Char) when Char =< 16#FF -> + Acc1 = [Acc, "\\u00" | integer_to_binary(Char, 16)], + escape_all(Rest, Acc1, Orig, Skip + 2, 0); +escape_char(<>, Acc, Orig, Skip, Char) when Char =< 16#7FF -> + Acc1 = [Acc, "\\u0" | integer_to_binary(Char, 16)], + escape_all(Rest, Acc1, Orig, Skip + 2, 0); +escape_char(<>, Acc, Orig, Skip, Char) when Char =< 16#FFF -> + Acc1 = [Acc, "\\u0" | integer_to_binary(Char, 16)], + escape_all(Rest, Acc1, Orig, Skip + 3, 0); +escape_char(<>, Acc, Orig, Skip, Char) when Char =< 16#FFFF -> + Acc1 = [Acc, "\\u" | integer_to_binary(Char, 16)], + escape_all(Rest, Acc1, Orig, Skip + 3, 0); +escape_char(<>, Acc, Orig, Skip, Char0) -> + Char = Char0 - 16#10000, + First = integer_to_binary(16#800 bor (Char bsr 10), 16), + Second = integer_to_binary(16#C00 bor (Char band 16#3FF), 16), + Acc1 = [Acc, "\\uD", First, "\\uD" | Second], + escape_all(Rest, Acc1, Orig, Skip + 4, 0). + +-spec escape(byte()) -> binary() | no. +escape($\x00) -> <<"\\u0000">>; +escape($\x01) -> <<"\\u0001">>; +escape($\x02) -> <<"\\u0002">>; +escape($\x03) -> <<"\\u0003">>; +escape($\x04) -> <<"\\u0004">>; +escape($\x05) -> <<"\\u0005">>; +escape($\x06) -> <<"\\u0006">>; +escape($\x07) -> <<"\\u0007">>; +escape($\b) -> <<"\\b">>; +escape($\t) -> <<"\\t">>; +escape($\n) -> <<"\\n">>; +escape($\x0b) -> <<"\\u000B">>; +escape($\f) -> <<"\\f">>; +escape($\r) -> <<"\\r">>; +escape($\x0e) -> <<"\\u000E">>; +escape($\x0f) -> <<"\\u000F">>; +escape($\x10) -> <<"\\u0010">>; +escape($\x11) -> <<"\\u0011">>; +escape($\x12) -> <<"\\u0012">>; +escape($\x13) -> <<"\\u0013">>; +escape($\x14) -> <<"\\u0014">>; +escape($\x15) -> <<"\\u0015">>; +escape($\x16) -> <<"\\u0016">>; +escape($\x17) -> <<"\\u0017">>; +escape($\x18) -> <<"\\u0018">>; +escape($\x19) -> <<"\\u0019">>; +escape($\x1A) -> <<"\\u001A">>; +escape($\x1B) -> <<"\\u001B">>; +escape($\x1C) -> <<"\\u001C">>; +escape($\x1D) -> <<"\\u001D">>; +escape($\x1E) -> <<"\\u001E">>; +escape($\x1F) -> <<"\\u001F">>; +escape($") -> <<"\\\"">>; +escape($\\) -> <<"\\\\">>; +escape(_) -> no. + +%% This is an adapted table from "Flexible and Economical UTF-8 Decoding" by Bjoern Hoehrmann. +%% http://bjoern.hoehrmann.de/utf-8/decoder/dfa/ + +%% Map character to character class +utf8t() -> + { + 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, + 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, + 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, + 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, + 1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9, + 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7, 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7, + 8,8,2,2,2,2,2,2,2,2,2,2,2,2,2,2, 2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2, + 10,3,3,3,3,3,3,3,3,3,3,3,3,4,3,3, 11,6,6,6,5,8,8,8,8,8,8,8,8,8,8,8 + }. + +%% Transition table mapping combination of state & class to next state +utf8s() -> + { + 12,24,36,60,96,84,12,12,12,48,72, 12,12,12,12,12,12,12,12,12,12,12,12, + 12, 0,12,12,12,12,12, 0,12, 0,12,12, 12,24,12,12,12,12,12,24,12,24,12,12, + 12,12,12,12,12,12,12,24,12,12,12,12, 12,24,12,12,12,12,12,12,12,24,12,12, + 12,12,12,12,12,12,12,36,12,36,12,12, 12,36,12,12,12,12,12,36,12,36,12,12, + 12,36,12,12,12,12,12,12,12,12,12,12 + }. + +%% Optimisation for 1st byte direct state lookup, +%% we know starting state is 0 and ASCII bytes were already handled +utf8s0() -> + { + 12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12, + 12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12, + 12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12, + 12,12,12,12,12,12,12,12,12,12,12,12,12,12,12,12, + 12,12,24,24,24,24,24,24,24,24,24,24,24,24,24,24, + 24,24,24,24,24,24,24,24,24,24,24,24,24,24,24,24, + 48,36,36,36,36,36,36,36,36,36,36,36,36,60,36,36, + 72,84,84,84,96,12,12,12,12,12,12,12,12,12,12,12 + }. + +-spec invalid_byte(binary(), non_neg_integer()) -> no_return(). +invalid_byte(Bin, Skip) -> + Byte = binary:at(Bin, Skip), + extended_error({invalid_byte, Byte}, none, Skip). + +%% +%% Decoding implementation +%% + +-define(ARRAY, array). +-define(OBJECT, object). + +-type from_binary_fun() :: fun((binary()) -> dynamic()). +-type array_start_fun() :: fun((Acc :: dynamic()) -> ArrayAcc :: dynamic()). +-type array_push_fun() :: fun((Value :: dynamic(), Acc :: dynamic()) -> NewAcc :: dynamic()). +-type array_finish_fun() :: fun((ArrayAcc :: dynamic(), OldAcc :: dynamic()) -> {dynamic(), dynamic()}). +-type object_start_fun() :: fun((Acc :: dynamic()) -> ObjectAcc :: dynamic()). +-type object_push_fun() :: fun((Key :: dynamic(), Value :: dynamic(), Acc :: dynamic()) -> NewAcc :: dynamic()). +-type object_finish_fun() :: fun((ObjectAcc :: dynamic(), OldAcc :: dynamic()) -> {dynamic(), dynamic()}). + +-type decoders() :: #{ + array_start => array_start_fun(), + array_push => array_push_fun(), + array_finish => array_finish_fun(), + object_start => object_start_fun(), + object_push => object_push_fun(), + object_finish => object_finish_fun(), + float => from_binary_fun(), + integer => from_binary_fun(), + string => from_binary_fun(), + null => term() +}. + +-record(decode, { + array_start :: array_start_fun() | undefined, + array_push :: array_push_fun() | undefined, + array_finish :: array_finish_fun() | undefined, + object_start :: object_start_fun() | undefined, + object_push :: object_push_fun() | undefined, + object_finish :: object_finish_fun() | undefined, + float = fun erlang:binary_to_float/1 :: from_binary_fun(), + integer = fun erlang:binary_to_integer/1 :: from_binary_fun(), + string :: from_binary_fun() | undefined, + null = null :: term() +}). + +-type acc() :: dynamic(). +-type stack() :: [?ARRAY | ?OBJECT | binary() | acc()]. +-type decode() :: #decode{}. + +-type decode_value() :: + integer() + | float() + | boolean() + | null + | binary() + | list(decode_value()) + | #{binary() => decode_value()}. + +% -doc """ +% Parses a JSON value from `Binary`. + +% Supports basic data mapping: + +% | **JSON** | **Erlang** | +% |----------|------------------------| +% | Number | `integer() \| float()` | +% | Boolean | `true \| false` | +% | Null | `null` | +% | String | `binary()` | +% | Object | `#{binary() => _}` | + +% ## Errors + +% * `error(unexpected_end)` if `Binary` contains incomplete JSON value +% * `error({invalid_byte, Byte})` if `Binary` contains unexpected byte or invalid UTF-8 byte +% * `error({invalid_sequence, Bytes})` if `Binary` contains invalid UTF-8 escape + +% ## Example + +% ```erlang +% > json:decode(<<"{\"foo\": 1}">>). +% #{<<"foo">> => 1} +% """. +-spec decode(binary()) -> decode_value(). +decode(Binary) when is_binary(Binary) -> + case value(Binary, Binary, 0, ok, [], #decode{}) of + {Result, _Acc, <<>>} -> Result; + {_, _, Rest} -> unexpected(Rest, 0) + end. + +% -doc """ +% Parses a JSON value from `Binary`. + +% Similar to `decode/1` except the decoding process +% can be customized with the callbacks specified in +% `Decoders`. The callbacks will use the `Acc` value +% as the initial accumulator. + +% Any leftover, unparsed data in `Binary` will be returned. + +% ## Default callbacks + +% All callbacks are optional. If not provided, they will fall back to +% implementations used by the `decode/1` function: + +% * for `array_start`: `fun(_) -> [] end` +% * for `array_push`: `fun(Elem, Acc) -> [Elem | Acc] end` +% * for `array_finish`: `fun(Acc, OldAcc) -> {lists:reverse(Acc), OldAcc} end` +% * for `object_start`: `fun(_) -> [] end` +% * for `object_push`: `fun(Key, Value, Acc) -> [{Key, Value} | Acc] end` +% * for `object_finish`: `fun(Acc, OldAcc) -> {maps:from_list(Acc), OldAcc} end` +% * for `float`: `fun erlang:binary_to_float/1` +% * for `integer`: `fun erlang:binary_to_integer/1` +% * for `string`: `fun (Value) -> Value end` +% * for `null`: the atom `null` + +% ## Errors + +% * `error(unexpected_end)` if `Binary` contains incomplete JSON value +% * `error({invalid_byte, Byte})` if `Binary` contains unexpected byte or invalid UTF-8 byte +% * `error({invalid_sequence, Bytes})` if `Binary` contains invalid UTF-8 escape + +% ## Example + +% Decoding object keys as atoms: + +% ```erlang +% > Push = fun(Key, Value, Acc) -> [{binary_to_existing_atom(Key), Value} | Acc] end. +% > json:decode(<<"{\"foo\": 1}">>, ok, #{object_push => Push}). +% {#{foo => 1},ok,<<>>} +% ``` +% """. +-spec decode(binary(), dynamic(), decoders()) -> + {Result :: dynamic(), Acc :: dynamic(), binary()}. +decode(Binary, Acc, Decoders) when is_binary(Binary) -> + Decode = maps:fold(fun parse_decoder/3, #decode{}, Decoders), + value(Binary, Binary, 0, Acc, [], Decode). + +parse_decoder(array_start, Fun, Decode) when is_function(Fun, 1) -> + Decode#decode{array_start = Fun}; +parse_decoder(array_push, Fun, Decode) when is_function(Fun, 2) -> + Decode#decode{array_push = Fun}; +parse_decoder(array_finish, Fun, Decode) when is_function(Fun, 2) -> + Decode#decode{array_finish = Fun}; +parse_decoder(object_start, Fun, Decode) when is_function(Fun, 1) -> + Decode#decode{object_start = Fun}; +parse_decoder(object_push, Fun, Decode) when is_function(Fun, 3) -> + Decode#decode{object_push = Fun}; +parse_decoder(object_finish, Fun, Decode) when is_function(Fun, 2) -> + Decode#decode{object_finish = Fun}; +parse_decoder(float, Fun, Decode) when is_function(Fun, 1) -> + Decode#decode{float = Fun}; +parse_decoder(integer, Fun, Decode) when is_function(Fun, 1) -> + Decode#decode{integer = Fun}; +parse_decoder(string, Fun, Decode) when is_function(Fun, 1) -> + Decode#decode{string = Fun}; +parse_decoder(null, Null, Decode) -> + Decode#decode{null = Null}. + +value(<>, Original, Skip, Acc, Stack, Decode) when ?is_ws(Byte) -> + value(Rest, Original, Skip + 1, Acc, Stack, Decode); +value(<<$0, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + number_zero(Rest, Original, Skip, Acc, Stack, Decode, 1); +value(<>, Original, Skip, Acc, Stack, Decode) when ?is_1_to_9(Byte) -> + number(Rest, Original, Skip, Acc, Stack, Decode, 1); +value(<<$-, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + number_minus(Rest, Original, Skip, Acc, Stack, Decode); +value(<<$t, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + true(Rest, Original, Skip, Acc, Stack, Decode); +value(<<$f, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + false(Rest, Original, Skip, Acc, Stack, Decode); +value(<<$n, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + null(Rest, Original, Skip, Acc, Stack, Decode); +value(<<$", Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + string(Rest, Original, Skip + 1, Acc, Stack, Decode); +value(<<$[, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + array_start(Rest, Original, Skip, Acc, Stack, Decode); +value(<<${, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + object_start(Rest, Original, Skip, Acc, Stack, Decode); +value(<>, Original, Skip, _Acc, _Stack, _Decode) when ?is_ascii_plain(Byte) -> + %% this clause is effecively the same as the last one, but necessary to + %% force compiler to emit a jump table dispatch, rather than binary search + invalid_byte(Original, Skip); +value(_, Original, Skip, _Acc, _Stack, _Decode) -> + unexpected(Original, Skip). + +true(<<"rue", Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + continue(Rest, Original, Skip + 4, Acc, Stack, Decode, true); +true(_Rest, Original, Skip, _Acc, _Stack, _Decode) -> + unexpected(Original, Skip + 1). + +false(<<"alse", Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + continue(Rest, Original, Skip + 5, Acc, Stack, Decode, false); +false(_Rest, Original, Skip, _Acc, _Stack, _Decode) -> + unexpected(Original, Skip + 1). + +null(<<"ull", Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + continue(Rest, Original, Skip + 4, Acc, Stack, Decode, Decode#decode.null); +null(_Rest, Original, Skip, _Acc, _Stack, _Decode) -> + unexpected(Original, Skip + 1). + +number_minus(<<$0, Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + number_zero(Rest, Original, Skip, Acc, Stack, Decode, 2); +number_minus(<>, Original, Skip, Acc, Stack, Decode) when ?is_1_to_9(Num) -> + number(Rest, Original, Skip, Acc, Stack, Decode, 2); +number_minus(_Rest, Original, Skip, _Acc, _Stack, _Decode) -> + unexpected(Original, Skip + 1). + +number_zero(<<$., Rest/bits>>, Original, Skip, Acc, Stack, Decode, Len) -> + number_frac(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_zero(<>, Original, Skip, Acc, Stack, Decode, Len) when E =:= $E; E =:= $e -> + number_exp_copy(Rest, Original, Skip, Acc, Stack, Decode, Len + 1, <<"0">>); +number_zero(Rest, Original, Skip, Acc, Stack, Decode, Len) -> + continue(Rest, Original, Skip + Len, Acc, Stack, Decode, 0). + +number(<>, Original, Skip, Acc, Stack, Decode, Len) when ?is_0_to_9(Num) -> + number(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number(<<$., Rest/bits>>, Original, Skip, Acc, Stack, Decode, Len) -> + number_frac(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number(<>, Original, Skip, Acc, Stack, Decode, Len) when E =:= $E; E =:= $e -> + Prefix = binary_part(Original, Skip, Len), + number_exp_copy(Rest, Original, Skip, Acc, Stack, Decode, Len + 1, Prefix); +number(Rest, Original, Skip, Acc, Stack, Decode, Len) -> + Int = (Decode#decode.integer)(binary_part(Original, Skip, Len)), + continue(Rest, Original, Skip + Len, Acc, Stack, Decode, Int). + +number_frac(<>, Original, Skip, Acc, Stack, Decode, Len) when ?is_0_to_9(Byte) -> + number_frac_cont(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_frac(_, Original, Skip, _Acc, _Stack, _Decode, Len) -> + unexpected(Original, Skip + Len). + +number_frac_cont(<>, Original, Skip, Acc, Stack, Decode, Len) when ?is_0_to_9(Byte) -> + number_frac_cont(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_frac_cont(<>, Original, Skip, Acc, Stack, Decode, Len) when E =:= $e; E =:= $E -> + number_exp(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_frac_cont(Rest, Original, Skip, Acc, Stack, Decode, Len) -> + Token = binary_part(Original, Skip, Len), + float_decode(Rest, Original, Skip, Acc, Stack, Decode, Len, Token). + +float_decode(<>, Original, Skip, Acc, Stack, Decode, Len, Token) -> + try (Decode#decode.float)(Token) of + Float -> + continue(Rest, Original, Skip + Len, Acc, Stack, Decode, Float) + catch + _:_ -> unexpected_sequence(Token, Skip) + end. + +number_exp(<>, Original, Skip, Acc, Stack, Decode, Len) when ?is_0_to_9(Byte) -> + number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_exp(<>, Original, Skip, Acc, Stack, Decode, Len) when Sign =:= $+; Sign =:= $- -> + number_exp_sign(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_exp(_, Original, Skip, _Acc, _Stack, _Decode, Len) -> + unexpected(Original, Skip + Len). + +number_exp_sign(<>, Original, Skip, Acc, Stack, Decode, Len) when ?is_0_to_9(Byte) -> + number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_exp_sign(_, Original, Skip, _Acc, _Stack, _Decode, Len) -> + unexpected(Original, Skip + Len). + +number_exp_cont(<>, Original, Skip, Acc, Stack, Decode, Len) when ?is_0_to_9(Byte) -> + number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len + 1); +number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len) -> + Token = binary_part(Original, Skip, Len), + float_decode(Rest, Original, Skip, Acc, Stack, Decode, Len, Token). + +number_exp_copy(<>, Original, Skip, Acc, Stack, Decode, Len, Prefix) when ?is_0_to_9(Byte) -> + number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len, Prefix, 1); +number_exp_copy(<>, Original, Skip, Acc, Stack, Decode, Len, Prefix) when Sign =:= $+; Sign =:= $- -> + number_exp_sign(Rest, Original, Skip, Acc, Stack, Decode, Len, Prefix, 1); +number_exp_copy(_, Original, Skip, _Acc, _Stack, _Decode, Len, _Prefix) -> + unexpected(Original, Skip + Len). + +number_exp_sign(<>, Original, Skip, Acc, Stack, Decode, Len, Prefix, ExpLen) when ?is_0_to_9(Byte) -> + number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len, Prefix, ExpLen + 1); +number_exp_sign(_, Original, Skip, _Acc, _Stack, _Decode, Len, _Prefix, ExpLen) -> + unexpected(Original, Skip + Len + ExpLen). + +number_exp_cont(<>, Original, Skip, Acc, Stack, Decode, Len, Prefix, ExpLen) when ?is_0_to_9(Byte) -> + number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len, Prefix, ExpLen + 1); +number_exp_cont(Rest, Original, Skip, Acc, Stack, Decode, Len, Prefix, ExpLen) -> + Suffix = binary_part(Original, Skip + Len, ExpLen), + Token = <>, + float_decode(Rest, Original, Skip, Acc, Stack, Decode, Len + ExpLen, Token). + +string(Binary, Original, Skip, Acc, Stack, Decode) -> + string_ascii(Binary, Original, Skip, Acc, Stack, Decode, 0). + +string_ascii(Binary, Original, Skip, Acc, Stack, Decode, Len) -> + case Binary of + <> when ?are_all_ascii_plain(B1, B2, B3, B4, B5, B6, B7, B8) -> + string_ascii(Rest, Original, Skip, Acc, Stack, Decode, Len + 8); + Other -> + string(Other, Original, Skip, Acc, Stack, Decode, Len) + end. + +-spec string(binary(), binary(), integer(), acc(), stack(), decode(), integer()) -> dynamic(). +string(<>, Orig, Skip, Acc, Stack, Decode, Len) when ?is_ascii_plain(Byte) -> + string(Rest, Orig, Skip, Acc, Stack, Decode, Len + 1); +string(<<$\\, Rest/bits>>, Orig, Skip, Acc, Stack, Decode, Len) -> + Part = binary_part(Orig, Skip, Len), + SAcc = <<>>, + unescape(Rest, Orig, Skip, Acc, Stack, Decode, Len, <>); +string(<<$", Rest/bits>>, Orig, Skip0, Acc, Stack, Decode, Len) -> + Value = binary_part(Orig, Skip0, Len), + Skip = Skip0 + Len + 1, + case Decode#decode.string of + undefined -> continue(Rest, Orig, Skip, Acc, Stack, Decode, Value); + Fun -> continue(Rest, Orig, Skip, Acc, Stack, Decode, Fun(Value)) + end; +string(<>, Orig, Skip, _Acc, _Stack, _Decode, Len) when ?is_ascii_escape(Byte) -> + invalid_byte(Orig, Skip + Len); +string(<>, Orig, Skip, Acc, Stack, Decode, Len) -> + case element(Byte - 127, utf8s0()) of + ?UTF8_REJECT -> invalid_byte(Orig, Skip + Len); + %% all accept cases are ASCII, already covered above + State -> string_utf8(Rest, Orig, Skip, Acc, Stack, Decode, Len, State) + end; +string(_, Orig, Skip, _Acc, _Stack, _Decode, Len) -> + unexpected(Orig, Skip + Len). + +string_utf8(<>, Orig, Skip, Acc, Stack, Decode, Len, State0) -> + Type = element(Byte + 1, utf8t()), + case element(State0 + Type, utf8s()) of + ?UTF8_ACCEPT -> string_ascii(Rest, Orig, Skip, Acc, Stack, Decode, Len + 2); + ?UTF8_REJECT -> invalid_byte(Orig, Skip + Len + 1); + State -> string_utf8(Rest, Orig, Skip, Acc, Stack, Decode, Len + 1, State) + end; +string_utf8(_, Orig, Skip, _Acc, _Stack, _Decode, Len, _State0) -> + unexpected(Orig, Skip + Len + 1). + +string_ascii(Binary, Original, Skip, Acc, Stack, Decode, Len, SAcc) -> + case Binary of + <> when ?are_all_ascii_plain(B1, B2, B3, B4, B5, B6, B7, B8) -> + string_ascii(Rest, Original, Skip, Acc, Stack, Decode, Len + 8, SAcc); + Other -> + string(Other, Original, Skip, Acc, Stack, Decode, Len, SAcc) + end. + +-spec string(binary(), binary(), integer(), acc(), stack(), decode(), integer(), binary()) -> dynamic(). +string(<>, Orig, Skip, Acc, Stack, Decode, Len, SAcc) when ?is_ascii_plain(Byte) -> + string(Rest, Orig, Skip, Acc, Stack, Decode, Len + 1, SAcc); +string(<<$\\, Rest/bits>>, Orig, Skip, Acc, Stack, Decode, Len, SAcc) -> + Part = binary_part(Orig, Skip, Len), + unescape(Rest, Orig, Skip, Acc, Stack, Decode, Len, <>); +string(<<$", Rest/bits>>, Orig, Skip0, Acc, Stack, Decode, Len, SAcc) -> + Part = binary_part(Orig, Skip0, Len), + Value = <>, + Skip = Skip0 + Len + 1, + case Decode#decode.string of + undefined -> continue(Rest, Orig, Skip, Acc, Stack, Decode, Value); + Fun -> continue(Rest, Orig, Skip, Acc, Stack, Decode, Fun(Value)) + end; +string(<>, Orig, Skip, _Acc, _Stack, _Decode, Len, _SAcc) when ?is_ascii_escape(Byte) -> + invalid_byte(Orig, Skip + Len); +string(<>, Orig, Skip, Acc, Stack, Decode, Len, SAcc) -> + case element(Byte - 127, utf8s0()) of + ?UTF8_REJECT -> invalid_byte(Orig, Skip + Len); + %% all accept cases are ASCII, already covred above + State -> string_utf8(Rest, Orig, Skip, Acc, Stack, Decode, Len, SAcc, State) + end; +string(_, Orig, Skip, _Acc, _Stack, _Decode, Len, _SAcc) -> + unexpected(Orig, Skip + Len). + +string_utf8(<>, Orig, Skip, Acc, Stack, Decode, Len, SAcc, State0) -> + Type = element(Byte + 1, utf8t()), + case element(State0 + Type, utf8s()) of + ?UTF8_ACCEPT -> string_ascii(Rest, Orig, Skip, Acc, Stack, Decode, Len + 2, SAcc); + ?UTF8_REJECT -> invalid_byte(Orig, Skip + Len + 1); + State -> string_utf8(Rest, Orig, Skip, Acc, Stack, Decode, Len + 1, SAcc, State) + end; +string_utf8(_, Orig, Skip, _Acc, _Stack, _Decode, Len, _SAcc, _State0) -> + unexpected(Orig, Skip + Len + 1). + +unescape(<>, Original, Skip, Acc, Stack, Decode, Len, SAcc) -> + Val = + case Byte of + $b -> $\b; + $f -> $\f; + $n -> $\n; + $r -> $\r; + $t -> $\t; + $" -> $"; + $\\ -> $\\; + $/ -> $/; + $u -> unicode; + _ -> error + end, + case Val of + unicode -> unescapeu(Rest, Original, Skip, Acc, Stack, Decode, Len, SAcc); + error -> unexpected(Original, Skip + Len + 1); + Int -> string_ascii(Rest, Original, Skip + Len + 2, Acc, Stack, Decode, 0, <>) + end; +unescape(_, Original, Skip, _Acc, _Stack, _Decode, Len, _SAcc) -> + unexpected(Original, Skip + Len + 1). + +unescapeu(<>, Original, Skip, Acc, Stack, Decode, Len, SAcc) -> + try hex_to_int(E1, E2, E3, E4) of + CP when CP >= 16#D800, CP =< 16#DBFF -> + unescape_surrogate(Rest, Original, Skip, Acc, Stack, Decode, Len, SAcc, CP); + CP -> + try <> of + SAcc1 -> string_ascii(Rest, Original, Skip + Len + 6, Acc, Stack, Decode, 0, SAcc1) + catch + _:_ -> unexpected_sequence(binary_part(Original, Skip + Len, 6), Skip + Len) + end + catch + _:_ -> + unexpected_sequence(binary_part(Original, Skip + Len, 6), Skip + Len) + end; +unescapeu(_, Original, Skip, _Acc, _Stack, _Decode, Len, _SAcc) -> + unexpected(Original, Skip + Len + 2). + +unescape_surrogate(<<"\\u", E1, E2, E3, E4, Rest/bits>>, Original, Skip, Acc, Stack, Decode, Len, SAcc, Hi) -> + try hex_to_int(E1, E2, E3, E4) of + Lo when Lo >= 16#DC00, Lo =< 16#DFFF -> + CP = 16#10000 + ((Hi band 16#3FF) bsl 10) + (Lo band 16#3FF), + try <> of + SAcc1 -> string_ascii(Rest, Original, Skip + Len + 12, Acc, Stack, Decode, 0, SAcc1) + catch + _:_ -> unexpected_sequence(binary_part(Original, Skip + Len, 12), Skip + Len) + end; + _ -> + unexpected_sequence(binary_part(Original, Skip + Len, 12), Skip + Len) + catch + _:_ -> unexpected_sequence(binary_part(Original, Skip + Len, 12), Skip + Len) + end; +unescape_surrogate(_, Original, Skip, _Acc, _Stack, _Decode, Len, _SAcc, _Hi) -> + unexpected(Original, Skip + Len + 6). + +%% erlfmt-ignore +%% this is a macro instead of an inlined function - compiler refused to inline +-define(hex_digit(C), element(C - $0 + 1, { + 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, n, n, n, n, n, %% 0x30 + n, n, 10,11,12,13,14,15,n, n, n, n, n, n, n, %% 0x40 + n, n, n, n, n, n, n, n, n, n, n, n, n, n, n, %% 0x50 + n, n, n, n, 10,11,12,13,14,15 %% 0x60 +})). + +-spec hex_to_int(byte(), byte(), byte(), byte()) -> integer(). +hex_to_int(H1, H2, H3, H4) -> + ?hex_digit(H4) + 16 * (?hex_digit(H3) + 16 * (?hex_digit(H2) + 16 * ?hex_digit(H1))). + +array_start(<>, Original, Skip, Acc, Stack, Decode) when ?is_ws(Byte) -> + array_start(Rest, Original, Skip + 1, Acc, Stack, Decode); +array_start(<<"]", Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + {Value, NewAcc} = + case {Decode#decode.array_start, Decode#decode.array_finish} of + {undefined, undefined} -> {[], Acc}; + {Start, undefined} -> {lists:reverse(Start(Acc)), Acc}; + {undefined, Finish} -> Finish([], Acc); + {Start, Finish} -> Finish(Start(Acc), Acc) + end, + continue(Rest, Original, Skip + 2, NewAcc, Stack, Decode, Value); +array_start(Rest, Original, Skip0, OldAcc, Stack, Decode) -> + Skip = Skip0 + 1, + case Decode#decode.array_start of + undefined -> value(Rest, Original, Skip, [], [?ARRAY, OldAcc | Stack], Decode); + Fun -> value(Rest, Original, Skip, Fun(OldAcc), [?ARRAY, OldAcc | Stack], Decode) + end. + +array_push(<>, Original, Skip, Acc, Stack, Decode, Value) when ?is_ws(Byte) -> + array_push(Rest, Original, Skip + 1, Acc, Stack, Decode, Value); +array_push(<<"]", Rest/bits>>, Original, Skip, Acc0, Stack0, Decode, Value) -> + Acc = + case Decode#decode.array_push of + undefined -> [Value | Acc0]; + Push -> Push(Value, Acc0) + end, + [_, OldAcc | Stack] = Stack0, + {ArrayValue, NewAcc} = + case Decode#decode.array_finish of + undefined -> {lists:reverse(Acc), OldAcc}; + Finish -> Finish(Acc, OldAcc) + end, + continue(Rest, Original, Skip + 1, NewAcc, Stack, Decode, ArrayValue); +array_push(<<$,, Rest/bits>>, Original, Skip0, Acc, Stack, Decode, Value) -> + Skip = Skip0 + 1, + case Decode#decode.array_push of + undefined -> value(Rest, Original, Skip, [Value | Acc], Stack, Decode); + Fun -> value(Rest, Original, Skip, Fun(Value, Acc), Stack, Decode) + end; +array_push(_, Original, Skip, _Acc, _Stack, _Decode, _Value) -> + unexpected(Original, Skip). + +object_start(<>, Original, Skip, Acc, Stack, Decode) when ?is_ws(Byte) -> + object_start(Rest, Original, Skip + 1, Acc, Stack, Decode); +object_start(<<"}", Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + {Value, NewAcc} = + case {Decode#decode.object_start, Decode#decode.object_finish} of + {undefined, undefined} -> {#{}, Acc}; + {Start, undefined} -> {maps:from_list(Start(Acc)), Acc}; + {undefined, Finish} -> Finish([], Acc); + {Start, Finish} -> Finish(Start(Acc), Acc) + end, + continue(Rest, Original, Skip + 2, NewAcc, Stack, Decode, Value); +object_start(<<$", Rest/bits>>, Original, Skip0, OldAcc, Stack0, Decode) -> + Stack = [?OBJECT, OldAcc | Stack0], + Skip = Skip0 + 2, + case Decode#decode.object_start of + undefined -> + string(Rest, Original, Skip, [], Stack, Decode); + Fun -> + Acc = Fun(OldAcc), + string(Rest, Original, Skip, Acc, Stack, Decode) + end; +object_start(_, Original, Skip, _Acc, _Stack, _Decode) -> + unexpected(Original, Skip + 1). + +object_value(<>, Original, Skip, Acc, Stack, Decode, Key) when ?is_ws(Byte) -> + object_value(Rest, Original, Skip + 1, Acc, Stack, Decode, Key); +object_value(<<$:, Rest/bits>>, Original, Skip, Acc, Stack, Decode, Key) -> + value(Rest, Original, Skip + 1, Acc, [Key | Stack], Decode); +object_value(_, Original, Skip, _Acc, _Stack, _Decode, _Key) -> + unexpected(Original, Skip). + +object_push(<>, Original, Skip, Acc, Stack, Decode, Value, Key) when ?is_ws(Byte) -> + object_push(Rest, Original, Skip + 1, Acc, Stack, Decode, Value, Key); +object_push(<<"}", Rest/bits>>, Original, Skip, Acc0, Stack0, Decode, Value, Key) -> + Acc = + case Decode#decode.object_push of + undefined -> [{Key, Value} | Acc0]; + Fun -> Fun(Key, Value, Acc0) + end, + [_, OldAcc | Stack] = Stack0, + {ObjectValue, NewAcc} = + case Decode#decode.object_finish of + undefined -> {maps:from_list(Acc), OldAcc}; + Finish -> Finish(Acc, OldAcc) + end, + continue(Rest, Original, Skip + 1, NewAcc, Stack, Decode, ObjectValue); +object_push(<<$,, Rest/bits>>, Original, Skip, Acc0, Stack, Decode, Value, Key) -> + case Decode#decode.object_push of + undefined -> object_key(Rest, Original, Skip + 1, [{Key, Value} | Acc0], Stack, Decode); + Fun -> object_key(Rest, Original, Skip + 1, Fun(Key, Value, Acc0), Stack, Decode) + end; +object_push(_, Original, Skip, _Acc, _Stack, _Decode, _Value, _Key) -> + unexpected(Original, Skip). + +object_key(<>, Original, Skip, Acc, Stack, Decode) when ?is_ws(Byte) -> + object_key(Rest, Original, Skip + 1, Acc, Stack, Decode); +object_key(<<$", Rest/bits>>, Original, Skip, Acc, Stack, Decode) -> + string(Rest, Original, Skip + 1, Acc, Stack, Decode); +object_key(_, Original, Skip, _Acc, _Stack, _Decode) -> + unexpected(Original, Skip). + +continue(<>, Original, Skip, Acc, Stack0, Decode, Value) -> + case Stack0 of + [] -> terminate(Rest, Original, Skip, Acc, Value); + [?ARRAY | _] -> array_push(Rest, Original, Skip, Acc, Stack0, Decode, Value); + [?OBJECT | _] -> object_value(Rest, Original, Skip, Acc, Stack0, Decode, Value); + [Key | Stack] -> object_push(Rest, Original, Skip, Acc, Stack, Decode, Value, Key) + end. + +terminate(<>, Original, Skip, Acc, Value) when ?is_ws(Byte) -> + terminate(Rest, Original, Skip + 1, Acc, Value); +terminate(<>, _Original, _Skip, Acc, Value) -> + {Value, Acc, Rest}. + +-spec unexpected(binary(), non_neg_integer()) -> no_return(). +unexpected(Original, Skip) when byte_size(Original) =:= Skip -> + error(unexpected_end); +unexpected(Original, Skip) -> + invalid_byte(Original, Skip). + +-spec unexpected_sequence(binary(), non_neg_integer()) -> no_return(). +unexpected_sequence(Value, Skip) -> + extended_error({unexpected_sequence, Value}, none, Skip). \ No newline at end of file diff --git a/src/riak_kv_wm_raw.hrl b/src/riak_kv_wm_raw.hrl index 08188ccd87..6d4cce0aa4 100644 --- a/src/riak_kv_wm_raw.hrl +++ b/src/riak_kv_wm_raw.hrl @@ -62,6 +62,7 @@ -define(Q_PROPS, "props"). -define(Q_BUCKETS, "buckets"). -define(Q_KEYS, "keys"). +-define(Q_KEYS_BIN, <<"keys">>). -define(Q_FALSE, "false"). -define(Q_TRUE, "true"). -define(Q_STREAM, "stream"). @@ -71,8 +72,10 @@ -define(Q_2I_MAX_RESULTS, "max_results"). -define(Q_2I_TERM_REGEX, "term_regex"). -define(Q_2I_CONTINUATION, "continuation"). +-define(Q_2I_CONTINUATION_BIN, <<"continuation">>). -define(Q_2I_PAGINATION_SORT, "pagination_sort"). -define(Q_RESULTS, "results"). +-define(Q_RESULTS_BIN, <<"results">>). -define(Q_RETURNVALUE, "returnvalue"). -define(Q_2I_MAPFOLD, "mapfold"). -define(Q_MF_MAPFOLDMOD, "mapfoldmod").