add a new string representation: '{string, list()}' tuple

This commit is contained in:
Petr Kozorezov 2017-02-28 11:23:24 +03:00
parent bbab95e60e
commit 340e002a25
5 changed files with 28 additions and 9 deletions

View File

@ -48,13 +48,13 @@
-export_type([object/0, msgpack_map/0, options/0, ext_packer/0, ext_unpacker/0]).
-type object() :: msgpack_term().
-type options() ::
-type options() ::
[{spec, new|old} |
{allow_atom, none|pack} |
{known_atoms, [atom()]} |
{unpack_str, as_binary|as_list} |
{unpack_str, as_binary|as_list|with_tag} |
{validate_string, boolean()} |
{pack_str, from_binary|from_list|none} |
{pack_str, from_binary|from_list|with_tag|none} |
{map_format, map|jiffy|jsx} |
{ext, {msgpack:ext_packer(), msgpack:ext_unpacker()} | module()}].
@ -158,14 +158,14 @@ parse_options([{allow_atom,Type}|T], Opt0) ->
parse_options([{known_atoms, Atoms}|T], Opt0) when is_list(Atoms) ->
parse_options(T, Opt0?OPTION{known_atoms=Atoms});
parse_options([{unpack_str, As}|T], Opt0) when As =:= as_binary orelse As =:= as_list ->
parse_options([{unpack_str, As}|T], Opt0) when As =:= as_binary orelse As =:= as_list orelse As =:= with_tag ->
parse_options(T, Opt0?OPTION{unpack_str=As});
parse_options([{validate_string, Bool}|T], Opt) when is_boolean(Bool) ->
parse_options(T, Opt?OPTION{validate_string=Bool});
parse_options([{pack_str, From}|T], Opt)
when From =:= from_binary orelse From =:= from_list orelse From =:= none ->
when From =:= from_binary orelse From =:= from_list orelse From =:= with_tag orelse From =:= none ->
parse_options(T, Opt?OPTION{pack_str=From});
parse_options([{map_format,Type}|T], Opt0)

View File

@ -37,9 +37,9 @@
spec = new :: new | old,
allow_atom = pack :: none | pack, %% allows atom when packing
known_atoms = [] :: [atom()],
unpack_str = as_list :: as_binary | as_list,
unpack_str = as_list :: as_binary | as_list | with_tag,
validate_string = false :: boolean(),
pack_str = from_list :: from_binary | from_list | none,
pack_str = from_list :: from_binary | from_list | with_tag | none,
map_format = ?DEFAULT_MAP_FORMAT :: format_type(),
map_unpack_fun = ?DEFAULT_MAP_UNPACKER_FUN :: msgpack_map_unpacker(),
ext_packer = undefined :: msgpack:ext_packer() | undefined,

View File

@ -61,6 +61,11 @@ pack(Map, Opt = ?OPTION{map_format=jsx}) when Map =:= [{}]->
pack([{_,_}|_] = Map, Opt = ?OPTION{map_format=jsx}) ->
pack_map(Map, Opt);
pack({string, String}, ?OPTION{spec=new, pack_str=with_tag}=Opt) ->
case pack_string(String, Opt) of
{error, _} -> throw({badarg, String});
Bin when is_binary(Bin) -> Bin
end;
pack(List, ?OPTION{spec=new, pack_str=from_list}=Opt) when is_list(List) ->
try
case lists:all(fun is_integer/1, List) of

View File

@ -233,7 +233,8 @@ unpack_str_or_raw(V, ?OPTION{spec=new,
{case UnpackStr of
as_binary when ValidateString -> unpack_str(V), maybe_bin(V, Opt);
as_binary -> maybe_bin(V, Opt);
as_list -> unpack_str(V)
as_list -> unpack_str(V);
with_tag -> {string, unpack_str(V)}
end, Rest}.
maybe_bin(Bin, _) ->

View File

@ -568,6 +568,17 @@ new_spec_pack_test_() ->
?_assertMatch(<<16#DD, 0, 1, 0, 0, _:65536/binary>>, %% 0xDD, four bytes, N objects
msgpack:pack(list_minus_one(65536), [{spec,new},{pack_str,PackStr}]))]
|| PackStr <- [from_list, from_binary, none]]
},
{"pack_str with_tag",
[?_assertEqual(<<2#101:3, 3:5, 97,97,97>>,
msgpack:pack({string, "aaa"}, [{spec,new},{pack_str,with_tag}])),
?_assertMatch(<<16#D9, 32, _:32/binary>>,
msgpack:pack({string, list_a(32)}, [{spec,new},{pack_str,with_tag}])),
?_assertMatch(<<16#DA, 1, 0, _:256/binary>>,
msgpack:pack({string, list_a(256)}, [{spec,new},{pack_str,with_tag}])),
?_assertMatch(<<16#DB, 0, 1, 0, 0, _:65536/binary>>,
msgpack:pack({string, list_a(65536)}, [{spec,new},{pack_str,with_tag}]))
]
}].
new_spec_unpack_test_() ->
@ -603,7 +614,9 @@ new_spec_unpack_test_() ->
?_assertEqual({ok, list_a(256)},
msgpack:unpack(<<16#DA, 1,0, (binary_a(256))/binary>>, [{spec,new},{unpack_str,as_list}])),
?_assertEqual({ok, list_a(65536)},
msgpack:unpack(<<16#DB, 0,1,0,0, (binary_a(65536))/binary>>, [{spec,new},{unpack_str,as_list}]))
msgpack:unpack(<<16#DB, 0,1,0,0, (binary_a(65536))/binary>>, [{spec,new},{unpack_str,as_list}])),
?_assertEqual({ok, {string, "aaa"}},
msgpack:unpack(<<2#101:3, 3:5, 97,97,97>>, [{spec,new},{unpack_str,with_tag}]))
]}].
unpack_str_validation_test_() ->