emqx4.3/src/emqx_json.erl

%%--------------------------------------------------------------------
%% Copyright (c) 2020 EMQ Technologies Co., Ltd. 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.
%%--------------------------------------------------------------------

-module(emqx_json).

-compile(inline).

-export([ encode/1
        , encode/2
        , safe_encode/1
        , safe_encode/2
        ]).

-compile({inline,
          [ encode/1
          , encode/2
          ]}).

-export([ decode/1
        , decode/2
        , safe_decode/1
        , safe_decode/2
        ]).

-compile({inline,
          [ decode/1
          , decode/2
          ]}).

-type(encode_options() :: jiffy:encode_options()).
-type(decode_options() :: jiffy:decode_options()).

-type(json_text() :: iolist() | binary()).
-type(json_term() :: jiffy:jiffy_decode_result()).

-export_type([json_text/0, json_term/0]).
-export_type([decode_options/0, encode_options/0]).

-spec(encode(json_term()) -> json_text()).
encode(Term) ->
    encode(Term, [force_utf8]).

-spec(encode(json_term(), encode_options()) -> json_text()).
encode(Term, Opts) ->
    to_binary(jiffy:encode(to_ejson(Term), Opts)).

-spec(safe_encode(json_term())
      -> {ok, json_text()} | {error, Reason :: term()}).
safe_encode(Term) ->
    safe_encode(Term, [force_utf8]).

-spec(safe_encode(json_term(), encode_options())
      -> {ok, json_text()} | {error, Reason :: term()}).
safe_encode(Term, Opts) ->
    try encode(Term, Opts) of
        Json -> {ok, Json}
    catch
        error:Reason ->
            {error, Reason}
    end.

-spec(decode(json_text()) -> json_term()).
decode(Json) -> decode(Json, []).

-spec(decode(json_text(), decode_options()) -> json_term()).
decode(Json, Opts) ->
    from_ejson(jiffy:decode(Json, Opts)).

-spec(safe_decode(json_text())
      -> {ok, json_term()} | {error, Reason :: term()}).
safe_decode(Json) ->
    safe_decode(Json, []).

-spec(safe_decode(json_text(), decode_options())
      -> {ok, json_term()} | {error, Reason :: term()}).
safe_decode(Json, Opts) ->
    try decode(Json, Opts) of
        Term -> {ok, Term}
    catch
        error:Reason ->
            {error, Reason}
    end.

%%--------------------------------------------------------------------
%% Helpers
%%--------------------------------------------------------------------

-compile({inline,
          [ to_ejson/1
          , from_ejson/1
          ]}).

to_ejson([{}]) ->
    {[]};
to_ejson([{_, _}|_] = L) ->
    {[{K, to_ejson(V)} || {K, V} <- L ]};
to_ejson(L) when is_list(L) ->
    [to_ejson(E) || E <- L];
to_ejson(T) -> T.

from_ejson(L) when is_list(L) ->
    [from_ejson(E) || E <- L];
from_ejson({[]}) ->
    [{}];
from_ejson({L}) ->
    [{Name, from_ejson(Value)} || {Name, Value} <- L];
from_ejson(T) -> T.

to_binary(B) when is_binary(B) -> B;
to_binary(L) when is_list(L) ->
    iolist_to_binary(L).