machi/src/machi_pb_translate.erl

874 lines
36 KiB
Erlang
Raw Normal View History

%% -------------------------------------------------------------------
%%
%% Copyright (c) 2007-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.
%%
%% -------------------------------------------------------------------
-module(machi_pb_translate).
2015-06-26 07:58:24 +00:00
%% @doc Adapt impedence mismatches between Erlang and Protocol Buffers.
-include("machi.hrl").
-include("machi_pb.hrl").
-include("machi_projection.hrl").
2015-06-26 07:58:24 +00:00
-export([from_pb_request/1,
from_pb_response/1,
to_pb_request/2,
to_pb_response/3
]).
2015-06-26 07:58:24 +00:00
from_pb_request(#mpb_ll_request{
req_id=ReqID,
echo=#mpb_echoreq{message=Msg}}) ->
2015-06-26 11:47:55 +00:00
{ReqID, {low_echo, undefined, Msg}};
2015-06-26 07:58:24 +00:00
from_pb_request(#mpb_ll_request{
req_id=ReqID,
auth=#mpb_authreq{user=User, password=Pass}}) ->
2015-06-26 11:47:55 +00:00
{ReqID, {low_auth, undefined, User, Pass}};
2015-06-26 07:58:24 +00:00
from_pb_request(#mpb_ll_request{
req_id=ReqID,
2015-06-26 09:22:15 +00:00
append_chunk=#mpb_ll_appendchunkreq{
epoch_id=PB_EpochID,
placement_key=PKey,
prefix=Prefix,
chunk=Chunk,
csum=#mpb_chunkcsum{type=CSum_type, csum=CSum},
chunk_extra=ChunkExtra}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
CSum_tag = conv_to_csum_tag(CSum_type),
{ReqID, {low_append_chunk, EpochID, PKey, Prefix, Chunk, CSum_tag, CSum,
ChunkExtra}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
write_chunk=#mpb_ll_writechunkreq{
epoch_id=PB_EpochID,
file=File,
offset=Offset,
chunk=Chunk,
csum=#mpb_chunkcsum{type=CSum_type, csum=CSum}}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
CSum_tag = conv_to_csum_tag(CSum_type),
{ReqID, {low_write_chunk, EpochID, File, Offset, Chunk, CSum_tag, CSum}};
2015-06-26 09:59:07 +00:00
from_pb_request(#mpb_ll_request{
req_id=ReqID,
read_chunk=#mpb_ll_readchunkreq{
epoch_id=PB_EpochID,
file=File,
offset=Offset,
size=Size,
flag_get_checksum=PB_GetChecksum,
flag_no_chunk=PB_GetNoChunk}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
Opts = [{get_checksum, conv_to_boolean(PB_GetChecksum)},
{no_chunk, conv_to_boolean(PB_GetNoChunk)}],
{ReqID, {low_read_chunk, EpochID, File, Offset, Size, Opts}};
2015-06-26 09:22:15 +00:00
from_pb_request(#mpb_ll_request{
req_id=ReqID,
checksum_list=#mpb_ll_checksumlistreq{
epoch_id=PB_EpochID,
file=File}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
{ReqID, {low_checksum_list, EpochID, File}};
2015-06-26 09:59:07 +00:00
from_pb_request(#mpb_ll_request{
req_id=ReqID,
list_files=#mpb_ll_listfilesreq{
epoch_id=PB_EpochID}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
{ReqID, {low_list_files, EpochID}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
wedge_status=#mpb_ll_wedgestatusreq{}}) ->
2015-06-26 11:47:55 +00:00
{ReqID, {low_wedge_status, undefined}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
delete_migration=#mpb_ll_deletemigrationreq{
epoch_id=PB_EpochID,
file=File}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
{ReqID, {low_delete_migration, EpochID, File}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
trunc_hack=#mpb_ll_trunchackreq{
epoch_id=PB_EpochID,
file=File}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
{ReqID, {low_trunc_hack, EpochID, File}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
proj_gl=#mpb_ll_getlatestepochidreq{type=ProjType}}) ->
{ReqID, {low_proj, {get_latest_epochid, conv_to_type(ProjType)}}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
proj_rl=#mpb_ll_readlatestprojectionreq{type=ProjType}}) ->
{ReqID, {low_proj, {read_latest_projection, conv_to_type(ProjType)}}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
proj_rp=#mpb_ll_readprojectionreq{type=ProjType,
epoch_number=Epoch}}) ->
{ReqID, {low_proj, {read_projection, conv_to_type(ProjType), Epoch}}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
proj_wp=#mpb_ll_writeprojectionreq{type=ProjType,
proj=ProjM}}) ->
Proj = conv_to_projection_v1(ProjM),
{ReqID, {low_proj, {write_projection, conv_to_type(ProjType), Proj}}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
proj_ga=#mpb_ll_getallprojectionsreq{type=ProjType}}) ->
{ReqID, {low_proj, {get_all_projections, conv_to_type(ProjType)}}};
from_pb_request(#mpb_ll_request{
req_id=ReqID,
proj_la=#mpb_ll_listallprojectionsreq{type=ProjType}}) ->
{ReqID, {low_proj, {list_all_projections, conv_to_type(ProjType)}}};
2015-06-26 09:59:07 +00:00
%%qqq
2015-06-26 07:58:24 +00:00
from_pb_request(#mpb_request{req_id=ReqID,
echo=#mpb_echoreq{message=Msg}}) ->
{ReqID, {high_echo, Msg}};
from_pb_request(#mpb_request{req_id=ReqID,
auth=#mpb_authreq{user=User, password=Pass}}) ->
{ReqID, {high_auth, User, Pass}};
from_pb_request(#mpb_request{req_id=ReqID,
append_chunk=IR=#mpb_appendchunkreq{}}) ->
#mpb_appendchunkreq{placement_key=__todoPK,
prefix=Prefix,
2015-06-26 09:22:15 +00:00
chunk=Chunk,
2015-06-26 07:58:24 +00:00
csum=CSum,
chunk_extra=ChunkExtra} = IR,
2015-06-26 09:22:15 +00:00
TaggedCSum = make_tagged_csum(CSum, Chunk),
{ReqID, {high_append_chunk, __todoPK, Prefix, Chunk, TaggedCSum,
2015-06-26 07:58:24 +00:00
ChunkExtra}};
from_pb_request(#mpb_request{req_id=ReqID,
write_chunk=IR=#mpb_writechunkreq{}}) ->
#mpb_writechunkreq{file=File,
offset=Offset,
2015-06-26 09:22:15 +00:00
chunk=Chunk,
2015-06-26 07:58:24 +00:00
csum=CSum} = IR,
2015-06-26 09:22:15 +00:00
TaggedCSum = make_tagged_csum(CSum, Chunk),
{ReqID, {high_write_chunk, File, Offset, Chunk, TaggedCSum}};
2015-06-26 07:58:24 +00:00
from_pb_request(#mpb_request{req_id=ReqID,
read_chunk=IR=#mpb_readchunkreq{}}) ->
#mpb_readchunkreq{file=File,
offset=Offset,
size=Size} = IR,
{ReqID, {high_read_chunk, File, Offset, Size}};
from_pb_request(#mpb_request{req_id=ReqID,
checksum_list=IR=#mpb_checksumlistreq{}}) ->
#mpb_checksumlistreq{file=File} = IR,
{ReqID, {high_checksum_list, File}};
from_pb_request(#mpb_request{req_id=ReqID,
list_files=_IR=#mpb_listfilesreq{}}) ->
{ReqID, {high_list_files}};
from_pb_request(#mpb_request{req_id=ReqID}) ->
{ReqID, {high_error, 999966, "Unknown request"}};
2015-06-26 12:36:07 +00:00
from_pb_request(_Else) ->
io:format(user, "\nRRR from_pb_request(~p)\n", [_Else]), timer:sleep(2000),
2015-06-26 07:58:24 +00:00
{<<>>, {high_error, 999667, "Unknown PB request"}}.
from_pb_response(#mpb_ll_response{
req_id=ReqID,
echo=#mpb_echoresp{message=Msg}}) ->
{ReqID, Msg};
2015-06-26 09:22:15 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
auth=#mpb_authresp{code=Code}}) ->
{ReqID, Code};
from_pb_response(#mpb_ll_response{
req_id=ReqID,
append_chunk=#mpb_ll_appendchunkresp{status=Status,
chunk_pos=ChunkPos}}) ->
case Status of
'OK' ->
#mpb_chunkpos{offset=Offset,
chunk_size=Size,
file_name=File} = ChunkPos,
{ReqID, {ok, {Offset, Size, list_to_binary(File)}}};
2015-06-26 09:22:15 +00:00
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
from_pb_response(#mpb_ll_response{
req_id=ReqID,
write_chunk=#mpb_ll_writechunkresp{status=Status}}) ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)};
2015-06-26 09:59:07 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
read_chunk=#mpb_ll_readchunkresp{status=Status,
chunk=Chunk}}) ->
case Status of
'OK' ->
{ReqID, {ok, Chunk}};
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
2015-06-26 07:58:24 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
checksum_list=#mpb_ll_checksumlistresp{
status=Status, chunk=Chunk}}) ->
case Status of
'OK' ->
{ReqID, {ok, Chunk}};
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
2015-06-26 09:59:07 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
list_files=#mpb_ll_listfilesresp{
status=Status, files=PB_Files}}) ->
case Status of
'OK' ->
Files = [{Size, list_to_binary(Name)} ||
2015-06-26 09:59:07 +00:00
#mpb_fileinfo{file_size=Size,
file_name=Name} <- PB_Files],
{ReqID, {ok, Files}};
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
from_pb_response(#mpb_ll_response{
req_id=ReqID,
wedge_status=#mpb_ll_wedgestatusresp{
epoch_id=PB_EpochID, wedged_flag=PB_Wedged}}) ->
EpochID = conv_to_epoch_id(PB_EpochID),
Wedged_p = if PB_Wedged == 1 -> true;
PB_Wedged == 0 -> false
end,
2015-06-26 11:47:55 +00:00
{ReqID, {ok, {Wedged_p, EpochID}}};
from_pb_response(#mpb_ll_response{
req_id=ReqID,
delete_migration=#mpb_ll_deletemigrationresp{
status=Status}}) ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)};
from_pb_response(#mpb_ll_response{
req_id=ReqID,
trunc_hack=#mpb_ll_trunchackresp{
status=Status}}) ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)};
2015-06-26 09:59:07 +00:00
%%qqq
2015-06-26 07:58:24 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
proj_gl=#mpb_ll_getlatestepochidresp{
status=Status, epoch_id=EID}}) ->
case Status of
'OK' ->
#mpb_epochid{epoch_number=Epoch, epoch_csum=CSum} = EID,
{ReqID, {ok, {Epoch, CSum}}};
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
2015-06-26 07:58:24 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
proj_rl=#mpb_ll_readlatestprojectionresp{
status=Status, proj=P}}) ->
case Status of
'OK' ->
{ReqID, {ok, conv_to_projection_v1(P)}};
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
2015-06-26 07:58:24 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
proj_rp=#mpb_ll_readprojectionresp{
status=Status, proj=P}}) ->
case Status of
'OK' ->
{ReqID, {ok, conv_to_projection_v1(P)}};
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
2015-06-26 07:58:24 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
proj_wp=#mpb_ll_writeprojectionresp{
status=Status}}) ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)};
2015-06-26 07:58:24 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
proj_ga=#mpb_ll_getallprojectionsresp{
status=Status, projs=ProjsM}}) ->
case Status of
'OK' ->
{ReqID, {ok, [conv_to_projection_v1(ProjM) || ProjM <- ProjsM]}};
_ ->
{ReqID, machi_pb_high_client:convert_general_status_code(Status)}
end;
2015-06-26 07:58:24 +00:00
from_pb_response(#mpb_ll_response{
req_id=ReqID,
proj_la=#mpb_ll_listallprojectionsresp{
status=Status, epochs=Epochs}}) ->
case Status of
'OK' ->
{ReqID, {ok, Epochs}};
_ ->
{ReqID< machi_pb_high_client:convert_general_status_code(Status)}
2015-06-26 07:58:24 +00:00
end.
2015-06-26 09:22:15 +00:00
%% TODO: move the #mbp_* record making code from
%% machi_pb_high_client:do_send_sync() clauses into to_pb_request().
2015-06-26 11:47:55 +00:00
to_pb_request(ReqID, {low_echo, _BogusEpochID, Msg}) ->
#mpb_ll_request{
2015-06-29 07:10:43 +00:00
req_id=ReqID, do_not_alter=2,
2015-06-26 07:58:24 +00:00
echo=#mpb_echoreq{message=Msg}};
2015-06-26 11:47:55 +00:00
to_pb_request(ReqID, {low_auth, _BogusEpochID, User, Pass}) ->
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
2015-06-26 09:59:07 +00:00
auth=#mpb_authreq{user=User, password=Pass}};
2015-06-26 09:22:15 +00:00
to_pb_request(ReqID, {low_append_chunk, EpochID, PKey, Prefix, Chunk,
CSum_tag, CSum, ChunkExtra}) ->
PB_EpochID = conv_from_epoch_id(EpochID),
CSum_type = conv_from_csum_tag(CSum_tag),
PB_CSum = #mpb_chunkcsum{type=CSum_type, csum=CSum},
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
2015-06-26 09:59:07 +00:00
append_chunk=#mpb_ll_appendchunkreq{
epoch_id=PB_EpochID,
placement_key=PKey,
prefix=Prefix,
chunk=Chunk,
csum=PB_CSum,
chunk_extra=ChunkExtra}};
2015-06-26 09:22:15 +00:00
to_pb_request(ReqID, {low_write_chunk, EpochID, File, Offset, Chunk, CSum_tag, CSum}) ->
PB_EpochID = conv_from_epoch_id(EpochID),
CSum_type = conv_from_csum_tag(CSum_tag),
PB_CSum = #mpb_chunkcsum{type=CSum_type, csum=CSum},
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
2015-06-26 09:59:07 +00:00
write_chunk=#mpb_ll_writechunkreq{
epoch_id=PB_EpochID,
file=File,
offset=Offset,
chunk=Chunk,
csum=PB_CSum}};
to_pb_request(ReqID, {low_read_chunk, EpochID, File, Offset, Size, _Opts}) ->
%% TODO: stop ignoring Opts ^_^
PB_EpochID = conv_from_epoch_id(EpochID),
2015-06-26 09:22:15 +00:00
#mpb_ll_request{
2015-06-29 07:10:43 +00:00
req_id=ReqID, do_not_alter=2,
2015-06-26 09:59:07 +00:00
read_chunk=#mpb_ll_readchunkreq{
2015-06-26 09:22:15 +00:00
epoch_id=PB_EpochID,
2015-06-26 09:59:07 +00:00
file=File,
2015-06-26 09:22:15 +00:00
offset=Offset,
2015-06-26 09:59:07 +00:00
size=Size}};
2015-06-26 07:58:24 +00:00
to_pb_request(ReqID, {low_checksum_list, EpochID, File}) ->
PB_EpochID = conv_from_epoch_id(EpochID),
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
checksum_list=#mpb_ll_checksumlistreq{
epoch_id=PB_EpochID,
file=File}};
2015-06-26 09:59:07 +00:00
to_pb_request(ReqID, {low_list_files, EpochID}) ->
PB_EpochID = conv_from_epoch_id(EpochID),
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
2015-06-26 09:59:07 +00:00
list_files=#mpb_ll_listfilesreq{epoch_id=PB_EpochID}};
2015-06-26 11:47:55 +00:00
to_pb_request(ReqID, {low_wedge_status, _BogusEpochID}) ->
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
wedge_status=#mpb_ll_wedgestatusreq{}};
to_pb_request(ReqID, {low_delete_migration, EpochID, File}) ->
PB_EpochID = conv_from_epoch_id(EpochID),
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
delete_migration=#mpb_ll_deletemigrationreq{
epoch_id=PB_EpochID,
file=File}};
to_pb_request(ReqID, {low_trunc_hack, EpochID, File}) ->
PB_EpochID = conv_from_epoch_id(EpochID),
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
trunc_hack=#mpb_ll_trunchackreq{
epoch_id=PB_EpochID,
file=File}};
to_pb_request(ReqID, {low_proj, {get_latest_epochid, ProjType}}) ->
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
proj_gl=#mpb_ll_getlatestepochidreq{type=conv_from_type(ProjType)}};
to_pb_request(ReqID, {low_proj, {read_latest_projection, ProjType}}) ->
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
proj_rl=#mpb_ll_readlatestprojectionreq{type=conv_from_type(ProjType)}};
to_pb_request(ReqID, {low_proj, {read_projection, ProjType, Epoch}}) ->
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
proj_rp=#mpb_ll_readprojectionreq{type=conv_from_type(ProjType),
epoch_number=Epoch}};
to_pb_request(ReqID, {low_proj, {write_projection, ProjType, Proj}}) ->
ProjM = conv_from_projection_v1(Proj),
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
proj_wp=#mpb_ll_writeprojectionreq{type=conv_from_type(ProjType),
proj=ProjM}};
to_pb_request(ReqID, {low_proj, {get_all_projections, ProjType}}) ->
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
proj_ga=#mpb_ll_getallprojectionsreq{type=conv_from_type(ProjType)}};
to_pb_request(ReqID, {low_proj, {list_all_projections, ProjType}}) ->
2015-06-29 07:10:43 +00:00
#mpb_ll_request{req_id=ReqID, do_not_alter=2,
proj_la=#mpb_ll_listallprojectionsreq{type=conv_from_type(ProjType)}}.
2015-06-26 09:59:07 +00:00
%%qqq
2015-06-26 09:22:15 +00:00
2015-06-29 07:10:43 +00:00
to_pb_response(ReqID, _, {low_error, ErrCode, ErrMsg}) ->
make_ll_error_resp(ReqID, ErrCode, ErrMsg);
2015-06-26 11:47:55 +00:00
to_pb_response(ReqID, {low_echo, _BogusEpochID, _Msg}, Resp) ->
#mpb_ll_response{
2015-06-26 07:58:24 +00:00
req_id=ReqID,
echo=#mpb_echoresp{message=Resp}};
2015-06-29 07:10:43 +00:00
to_pb_response(ReqID, {low_auth, _, _, _}, __TODO_Resp) ->
#mpb_ll_response{req_id=ReqID,
generic=#mpb_errorresp{code=1,
msg="AUTH not implemented"}};
2015-06-26 09:22:15 +00:00
to_pb_response(ReqID, {low_append_chunk, _EID, _PKey, _Pfx, _Ch, _CST, _CS, _CE}, Resp)->
case Resp of
{ok, {Offset, Size, File}} ->
Where = #mpb_chunkpos{offset=Offset,
chunk_size=Size,
file_name=File},
#mpb_ll_response{req_id=ReqID,
append_chunk=#mpb_ll_appendchunkresp{status='OK',
chunk_pos=Where}};
2015-06-26 12:36:07 +00:00
{error, _}=Error ->
Status = conv_from_status(Error),
2015-06-26 09:22:15 +00:00
#mpb_ll_response{req_id=ReqID,
append_chunk=#mpb_ll_appendchunkresp{status=Status}};
_Else ->
2015-06-29 07:10:43 +00:00
make_ll_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
2015-06-26 09:22:15 +00:00
end;
to_pb_response(ReqID, {low_write_chunk, _EID, _Fl, _Off, _Ch, _CST, _CS},Resp)->
2015-06-26 09:59:07 +00:00
Status = conv_from_status(Resp),
2015-06-26 09:22:15 +00:00
#mpb_ll_response{req_id=ReqID,
write_chunk=#mpb_ll_writechunkresp{status=Status}};
2015-06-26 09:59:07 +00:00
to_pb_response(ReqID, {low_read_chunk, _EID, _Fl, _Off, _Sz, _Opts}, Resp)->
case Resp of
{ok, Chunk} ->
CSum = undefined, % TODO not implemented
#mpb_ll_response{req_id=ReqID,
read_chunk=#mpb_ll_readchunkresp{status='OK',
chunk=Chunk,
csum=CSum}};
2015-06-26 12:36:07 +00:00
{error, _}=Error ->
Status = conv_from_status(Error),
2015-06-26 09:59:07 +00:00
#mpb_ll_response{req_id=ReqID,
read_chunk=#mpb_ll_readchunkresp{status=Status}};
_Else ->
2015-06-29 07:10:43 +00:00
make_ll_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
2015-06-26 09:59:07 +00:00
end;
2015-06-26 07:58:24 +00:00
to_pb_response(ReqID, {low_checksum_list, _EpochID, _File}, Resp) ->
case Resp of
2015-06-26 09:22:15 +00:00
{ok, Chunk} ->
#mpb_ll_response{req_id=ReqID,
checksum_list=#mpb_ll_checksumlistresp{status='OK',
chunk=Chunk}};
2015-06-26 07:58:24 +00:00
{error, _}=Error ->
2015-06-26 09:22:15 +00:00
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
checksum_list=#mpb_ll_checksumlistresp{status=Status}};
_Else ->
make_ll_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
end;
2015-06-26 09:59:07 +00:00
to_pb_response(ReqID, {low_list_files, _EpochID}, Resp) ->
case Resp of
{ok, FileInfo} ->
PB_Files = [#mpb_fileinfo{file_size=Size, file_name=Name} ||
{Size, Name} <- FileInfo],
#mpb_ll_response{req_id=ReqID,
list_files=#mpb_ll_listfilesresp{status='OK',
files=PB_Files}};
{error, _}=Error ->
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
list_files=#mpb_ll_listfilesresp{status=Status}};
_Else ->
make_ll_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
end;
2015-06-26 11:47:55 +00:00
to_pb_response(ReqID, {low_wedge_status, _BogusEpochID}, Resp) ->
2015-06-29 07:42:05 +00:00
case Resp of
{error, _}=Error ->
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
2015-06-29 08:06:28 +00:00
wedge_status=#mpb_ll_wedgestatusresp{status=Status}};
2015-06-29 07:42:05 +00:00
{Wedged_p, EpochID} ->
PB_Wedged = conv_from_boolean(Wedged_p),
PB_EpochID = conv_from_epoch_id(EpochID),
#mpb_ll_response{req_id=ReqID,
wedge_status=#mpb_ll_wedgestatusresp{
status='OK',
epoch_id=PB_EpochID,
wedged_flag=PB_Wedged}}
end;
to_pb_response(ReqID, {low_delete_migration, _EID, _Fl}, Resp)->
Status = conv_from_status(Resp),
#mpb_ll_response{req_id=ReqID,
delete_migration=#mpb_ll_deletemigrationresp{status=Status}};
to_pb_response(ReqID, {low_trunc_hack, _EID, _Fl}, Resp)->
Status = conv_from_status(Resp),
#mpb_ll_response{req_id=ReqID,
trunc_hack=#mpb_ll_trunchackresp{status=Status}};
to_pb_response(ReqID, {low_proj, {get_latest_epochid, _ProjType}}, Resp)->
case Resp of
{ok, {Epoch, CSum}} ->
EID = #mpb_epochid{epoch_number=Epoch, epoch_csum=CSum},
#mpb_ll_response{req_id=ReqID,
proj_gl=#mpb_ll_getlatestepochidresp{
status='OK', epoch_id=EID}};
{error, _}=Error ->
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
proj_gl=#mpb_ll_getlatestepochidresp{status=Status}}
end;
to_pb_response(ReqID, {low_proj, {read_latest_projection, _ProjType}}, Resp) ->
case Resp of
{ok, Proj} ->
ProjM = conv_from_projection_v1(Proj),
#mpb_ll_response{req_id=ReqID,
proj_rl=#mpb_ll_readlatestprojectionresp{
status='OK', proj=ProjM}};
{error, _}=Error ->
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
proj_rl=#mpb_ll_readlatestprojectionresp{status=Status}}
end;
to_pb_response(ReqID, {low_proj, {read_projection, _ProjType, _Epoch}}, Resp)->
case Resp of
{ok, Proj} ->
ProjM = conv_from_projection_v1(Proj),
#mpb_ll_response{req_id=ReqID,
proj_rp=#mpb_ll_readprojectionresp{
status='OK', proj=ProjM}};
{error, _}=Error ->
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
proj_rp=#mpb_ll_readprojectionresp{status=Status}}
end;
to_pb_response(ReqID, {low_proj, {write_projection, _ProjType, _Proj}}, Resp) ->
Status = conv_from_status(Resp),
#mpb_ll_response{req_id=ReqID,
proj_wp=#mpb_ll_writeprojectionresp{status=Status}};
to_pb_response(ReqID, {low_proj, {get_all_projections, _ProjType}}, Resp)->
case Resp of
{ok, Projs} ->
ProjsM = [conv_from_projection_v1(Proj) || Proj <- Projs],
#mpb_ll_response{req_id=ReqID,
proj_ga=#mpb_ll_getallprojectionsresp{
status='OK', projs=ProjsM}};
{error, _}=Error ->
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
proj_ga=#mpb_ll_getallprojectionsresp{
status=Status}}
end;
to_pb_response(ReqID, {low_proj, {list_all_projections, _ProjType}}, Resp)->
case Resp of
{ok, Epochs} ->
#mpb_ll_response{req_id=ReqID,
proj_la=#mpb_ll_listallprojectionsresp{
status='OK', epochs=Epochs}};
{error, _}=Error ->
Status = conv_from_status(Error),
#mpb_ll_response{req_id=ReqID,
proj_la=#mpb_ll_listallprojectionsresp{
status=Status}}
end;
2015-06-26 09:59:07 +00:00
%%qqq
2015-06-29 07:10:43 +00:00
to_pb_response(ReqID, _, {high_error, ErrCode, ErrMsg}) ->
make_error_resp(ReqID, ErrCode, ErrMsg);
2015-06-26 07:58:24 +00:00
to_pb_response(ReqID, {high_echo, _Msg}, Resp) ->
Msg = Resp,
#mpb_response{req_id=ReqID,
echo=#mpb_echoresp{message=Msg}};
2015-06-26 07:58:24 +00:00
to_pb_response(ReqID, {high_auth, _User, _Pass}, _Resp) ->
#mpb_response{req_id=ReqID,
generic=#mpb_errorresp{code=1,
msg="AUTH not implemented"}};
2015-06-26 09:22:15 +00:00
to_pb_response(ReqID, {high_append_chunk, _TODO, _Prefix, _Chunk, _TSum, _CE}, Resp)->
case Resp of
{ok, {Offset, Size, File}} ->
2015-06-26 09:22:15 +00:00
Where = #mpb_chunkpos{offset=Offset,
chunk_size=Size,
file_name=File},
#mpb_response{req_id=ReqID,
append_chunk=#mpb_appendchunkresp{status='OK',
chunk_pos=Where}};
2015-06-26 12:36:07 +00:00
{error, _}=Error ->
Status = conv_from_status(Error),
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
append_chunk=#mpb_appendchunkresp{status=Status}};
_Else ->
make_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
end;
2015-06-26 09:22:15 +00:00
to_pb_response(ReqID, {high_write_chunk, _File, _Offset, _Chunk, _TaggedCSum}, Resp) ->
case Resp of
{ok, {_,_,_}} ->
%% machi_cr_client returns ok 2-tuple, convert to simple ok.
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
write_chunk=#mpb_writechunkresp{status='OK'}};
2015-06-26 12:36:07 +00:00
{error, _}=Error ->
Status = conv_from_status(Error),
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
write_chunk=#mpb_writechunkresp{status=Status}};
_Else ->
make_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
end;
2015-06-26 07:58:24 +00:00
to_pb_response(ReqID, {high_read_chunk, _File, _Offset, _Size}, Resp) ->
case Resp of
{ok, Chunk} ->
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
read_chunk=#mpb_readchunkresp{status='OK',
chunk=Chunk}};
2015-06-26 12:36:07 +00:00
{error, _}=Error ->
Status = conv_from_status(Error),
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
read_chunk=#mpb_readchunkresp{status=Status}};
_Else ->
make_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
end;
2015-06-26 07:58:24 +00:00
to_pb_response(ReqID, {high_checksum_list, _File}, Resp) ->
case Resp of
{ok, Chunk} ->
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
checksum_list=#mpb_checksumlistresp{status='OK',
chunk=Chunk}};
2015-06-26 12:36:07 +00:00
{error, _}=Error ->
Status = conv_from_status(Error),
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
checksum_list=#mpb_checksumlistresp{status=Status}};
_Else ->
make_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
end;
2015-06-26 07:58:24 +00:00
to_pb_response(ReqID, {high_list_files}, Resp) ->
case Resp of
{ok, FileInfo} ->
2015-06-26 09:22:15 +00:00
Files = [#mpb_fileinfo{file_size=Size, file_name=Name} ||
{Size, Name} <- FileInfo],
#mpb_response{req_id=ReqID,
list_files=#mpb_listfilesresp{status='OK',
files=Files}};
2015-06-26 12:36:07 +00:00
{error, _}=Error ->
Status = conv_from_status(Error),
2015-06-26 09:22:15 +00:00
#mpb_response{req_id=ReqID,
list_files=#mpb_listfilesresp{status=Status}};
_Else ->
make_error_resp(ReqID, 66, io_lib:format("err ~p", [_Else]))
2015-06-29 07:10:43 +00:00
end.
2015-06-26 09:22:15 +00:00
make_tagged_csum(#mpb_chunkcsum{type='CSUM_TAG_NONE'}, Chunk) ->
C = machi_util:checksum_chunk(Chunk),
machi_util:make_tagged_csum(server_sha, C);
make_tagged_csum(#mpb_chunkcsum{type='CSUM_TAG_CLIENT_SHA', csum=CSum}, _CB) ->
machi_util:make_tagged_csum(client_sha, CSum).
make_ll_error_resp(ReqID, Code, Msg) ->
#mpb_ll_response{req_id=ReqID,
generic=#mpb_errorresp{code=Code,
msg=Msg}}.
make_error_resp(ReqID, Code, Msg) ->
#mpb_response{req_id=ReqID,
generic=#mpb_errorresp{code=Code,
msg=Msg}}.
conv_from_epoch_id({Epoch, EpochCSum}) ->
#mpb_epochid{epoch_number=Epoch,
epoch_csum=EpochCSum}.
conv_to_epoch_id(#mpb_epochid{epoch_number=Epoch,
epoch_csum=EpochCSum}) ->
{Epoch, EpochCSum}.
conv_to_projection_v1(#mpb_projectionv1{epoch_number=Epoch,
epoch_csum=CSum,
author_server=Author,
all_members=AllMembers,
creation_time=CTime,
mode=Mode,
upi=UPI,
repairing=Repairing,
down=Down,
opaque_flap=Flap,
opaque_inner=Inner,
opaque_dbg=Dbg,
opaque_dbg2=Dbg2,
members_dict=MembersDict}) ->
#projection_v1{epoch_number=Epoch,
epoch_csum=CSum,
author_server=to_atom(Author),
all_members=[to_atom(X) || X <- AllMembers],
creation_time=conv_to_now(CTime),
mode=conv_to_mode(Mode),
upi=[to_atom(X) || X <- UPI],
repairing=[to_atom(X) || X <- Repairing],
down=[to_atom(X) || X <- Down],
flap=dec_optional_sexp(Flap),
inner=dec_optional_sexp(Inner),
dbg=dec_sexp(Dbg),
dbg2=dec_sexp(Dbg2),
members_dict=conv_to_members_dict(MembersDict)}.
enc_sexp(T) ->
term_to_binary(T).
dec_sexp(Bin) when is_binary(Bin) ->
binary_to_term(Bin).
enc_optional_sexp(undefined) ->
undefined;
enc_optional_sexp(T) ->
enc_sexp(T).
dec_optional_sexp(undefined) ->
undefined;
dec_optional_sexp(T) ->
dec_sexp(T).
conv_from_members_dict(D) ->
%% Use list_to_binary() here to "flatten" the serialized #p_srvr{}
[#mpb_membersdictentry{key=to_list(K), val=conv_from_p_srvr(V)} ||
{K, V} <- orddict:to_list(D)].
conv_to_members_dict(List) ->
orddict:from_list([{to_atom(K), conv_to_p_srvr(V)} ||
#mpb_membersdictentry{key=K, val=V} <- List]).
conv_from_p_srvr(#p_srvr{name=Name,
2015-06-26 07:58:24 +00:00
proto_mod=ProtoMod,
address=Address,
port=Port,
props=Props}) ->
#mpb_p_srvr{name=to_list(Name),
proto_mod=to_list(ProtoMod),
address=to_list(Address),
port=to_list(Port),
opaque_props=enc_sexp(Props)}.
conv_to_p_srvr(#mpb_p_srvr{name=Name,
proto_mod=ProtoMod,
address=Address,
port=Port,
opaque_props=Props}) ->
#p_srvr{name=to_atom(Name),
proto_mod=to_atom(ProtoMod),
address=to_list(Address),
port=to_integer(Port),
props=dec_sexp(Props)}.
to_list(X) when is_atom(X) ->
atom_to_list(X);
to_list(X) when is_binary(X) ->
binary_to_list(X);
to_list(X) when is_integer(X) ->
integer_to_list(X);
to_list(X) when is_list(X) ->
X.
to_atom(X) when is_list(X) ->
list_to_atom(X);
to_atom(X) when is_binary(X) ->
erlang:binary_to_atom(X, latin1);
to_atom(X) when is_atom(X) ->
X.
to_integer(X) when is_list(X) ->
list_to_integer(X);
to_integer(X) when is_binary(X) ->
list_to_binary(binary_to_list(X));
to_integer(X) when is_integer(X) ->
X.
2015-06-26 09:22:15 +00:00
conv_from_csum_tag(CSum_tag) ->
case CSum_tag of
?CSUM_TAG_NONE -> 'CSUM_TAG_NONE';
?CSUM_TAG_CLIENT_SHA -> 'CSUM_TAG_CLIENT_SHA';
?CSUM_TAG_SERVER_SHA -> 'CSUM_TAG_SERVER_SHA';
?CSUM_TAG_SERVER_REGEN_SHA -> 'CSUM_TAG_SERVER_REGEN_SHA'
end.
conv_to_csum_tag(Type) ->
case Type of
'CSUM_TAG_NONE' -> ?CSUM_TAG_NONE;
'CSUM_TAG_CLIENT_SHA' -> ?CSUM_TAG_CLIENT_SHA;
'CSUM_TAG_SERVER_SHA' -> ?CSUM_TAG_SERVER_SHA;
'CSUM_TAG_SERVER_REGEN_SHA' -> ?CSUM_TAG_SERVER_REGEN_SHA
end.
conv_from_now({A,B,C}) ->
#mpb_now{sec=(1000000 * A) + B,
usec=C}.
conv_to_now(#mpb_now{sec=Sec, usec=USec}) ->
{Sec div 1000000, Sec rem 1000000, USec}.
conv_from_mode(ap_mode) -> 'AP_MODE';
conv_from_mode(cp_mode) -> 'CP_MODE'.
conv_to_mode('AP_MODE') -> ap_mode;
conv_to_mode('CP_MODE') -> cp_mode.
conv_from_type(private) -> 'PRIVATE';
conv_from_type(public) -> 'PUBLIC'.
conv_to_type('PRIVATE') -> private;
conv_to_type('PUBLIC') -> public.
conv_from_status(ok) ->
'OK';
conv_from_status({error, bad_arg}) ->
'BAD_ARG';
conv_from_status({error, wedged}) ->
'WEDGED';
conv_from_status({error, bad_checksum}) ->
'BAD_CHECKSUM';
conv_from_status({error, partition}) ->
'PARTITION';
conv_from_status({error, not_written}) ->
'NOT_WRITTEN';
conv_from_status({error, written}) ->
'WRITTEN';
conv_from_status({error, no_such_file}) ->
'NO_SUCH_FILE';
2015-06-26 12:36:07 +00:00
conv_from_status({error, partial_read}) ->
'PARTIAL_READ';
2015-07-01 06:51:25 +00:00
conv_from_status({error, bad_epoch}) ->
'BAD_EPOCH';
conv_from_status(_OOPS) ->
io:format(user, "HEY, ~s:~w got ~w\n", [?MODULE, ?LINE, _OOPS]),
'BAD_JOSS'.
2015-06-26 09:59:07 +00:00
conv_to_boolean(undefined) ->
false;
conv_to_boolean(0) ->
false;
conv_to_boolean(N) when is_integer(N) ->
true.
2015-06-26 11:47:55 +00:00
conv_from_boolean(false) ->
0;
conv_from_boolean(true) ->
1.
conv_from_projection_v1(#projection_v1{epoch_number=Epoch,
epoch_csum=CSum,
author_server=Author,
all_members=AllMembers,
creation_time=CTime,
mode=Mode,
upi=UPI,
repairing=Repairing,
down=Down,
flap=Flap,
inner=Inner,
dbg=Dbg,
dbg2=Dbg2,
members_dict=MembersDict}) ->
#mpb_projectionv1{epoch_number=Epoch,
epoch_csum=CSum,
author_server=to_list(Author),
all_members=[to_list(X) || X <- AllMembers],
creation_time=conv_from_now(CTime),
mode=conv_from_mode(Mode),
upi=[to_list(X) || X <- UPI],
repairing=[to_list(X) || X <- Repairing],
down=[to_list(X) || X <- Down],
opaque_flap=enc_optional_sexp(Flap),
opaque_inner=enc_optional_sexp(Inner),
opaque_dbg=enc_sexp(Dbg),
opaque_dbg2=enc_sexp(Dbg2),
members_dict=conv_from_members_dict(MembersDict)}.