riak_client

%% -------------------------------------------------------------------
%%
%% riakc_ts_put_operator.erl: helper functions for put requests to Riak TS
%%
%% Copyright (c) 2015 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_put_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]).


%% As of 2015-11-05, columns parameter is ignored, Riak TS
%% expects the full set of fields in each element of Data.
serialize(TableName, Measurements, true) ->
    T = riakc_utils:characters_to_unicode_binary(TableName),
    #tsputreq{table   = T,
              columns = [],
              rows    = Measurements};
serialize(TableName, Measurements, false) ->
    T = riakc_utils:characters_to_unicode_binary(TableName),
    SerializedRows = riak_pb_ts_codec:encode_rows_non_strict(Measurements),
    #tsputreq{table   = T,
              columns = [],
              rows    = SerializedRows}.

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(Response) ->
    Response.