riak_client

%% -------------------------------------------------------------------
%%
%% riakc_ts_get_operator.erl: helper functions for get requests to Riak TS
%%
%% Copyright (c) 2016 Basho Technologies, Inc.  All Rights Reserved.
%%
%% This file is provided to you 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.
%%
%% -------------------------------------------------------------------

%% @doc Helper functions for put requests to Riak TS

-module(riakc_ts_get_operator).

-include_lib("include/riak_pb.hrl").
-include_lib("include/riak_ts_ttb.hrl").
-include_lib("include/riak_ts_pb.hrl").

-export([serialize/3,
         deserialize/1]).


serialize(Table, Key, true) ->
    T = riakc_utils:characters_to_unicode_binary(Table),
    #tsgetreq{table = T, key = Key};
serialize(Table, Key, false) ->
    T = riakc_utils:characters_to_unicode_binary(Table),
    SerializedKey = riak_pb_ts_codec:encode_cells_non_strict(Key),
    #tsgetreq{table = T, key = SerializedKey}.

deserialize({error, {Code, Message}}) when is_integer(Code), is_list(Message) ->
    {error, {Code, iolist_to_binary(Message)}};
deserialize({error, {Code, Message}}) when is_integer(Code), is_atom(Message) ->
    {error, {Code, iolist_to_binary(atom_to_list(Message))}};
deserialize({error, Message}) ->
    {error, Message};
deserialize(tsgetresp) ->
    {ok, {[], []}};
deserialize({tsgetresp, {ColumnNames, _ColumnTypes, Rows}}) ->
    {ok, {ColumnNames, Rows}};
deserialize(#tsgetresp{columns = C, rows = R}) ->
    ColumnNames = [CName || #tscolumndescription{name = CName} <- C],
    Rows = riak_pb_ts_codec:decode_rows(R),
    {ok, {ColumnNames, Rows}}.