1%%
2%% Licensed to the Apache Software Foundation (ASF) under one
3%% or more contributor license agreements. See the NOTICE file
4%% distributed with this work for additional information
5%% regarding copyright ownership. The ASF licenses this file
6%% to you under the Apache License, Version 2.0 (the
7%% "License"); you may not use this file except in compliance
8%% with the License. You may obtain a copy of the License at
9%%
10%%   http://www.apache.org/licenses/LICENSE-2.0
11%%
12%% Unless required by applicable law or agreed to in writing,
13%% software distributed under the License is distributed on an
14%% "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15%% KIND, either express or implied. See the License for the
16%% specific language governing permissions and limitations
17%% under the License.
18%%
19
20-module(thrift_multiplexed_protocol).
21
22-behaviour(thrift_protocol).
23
24-include("thrift_constants.hrl").
25-include("thrift_protocol.hrl").
26
27-include("thrift_protocol_behaviour.hrl").
28
29-export([new/2,
30         read/2,
31         write/2,
32         flush_transport/1,
33         close_transport/1
34        ]).
35
36-record(protocol, {module, data}).
37-type protocol() :: #protocol{}.
38
39-record (multiplexed_protocol, {protocol_module_to_decorate::atom(),
40								protocol_data_to_decorate::term(),
41								service_name::nonempty_string()}).
42
43-type state() :: #multiplexed_protocol{}.
44
45-spec new(ProtocolToDecorate::protocol(), ServiceName::nonempty_string()) -> {ok, Protocol::protocol()}.
46new(ProtocolToDecorate, ServiceName) when is_record(ProtocolToDecorate, protocol),
47                                          is_list(ServiceName) ->
48    State = #multiplexed_protocol{protocol_module_to_decorate = ProtocolToDecorate#protocol.module,
49                                    protocol_data_to_decorate = ProtocolToDecorate#protocol.data,
50                                                 service_name = ServiceName},
51    thrift_protocol:new(?MODULE, State).
52
53flush_transport(State = #multiplexed_protocol{protocol_module_to_decorate = ProtocolModuleToDecorate,
54                                                protocol_data_to_decorate = State0}) ->
55    {State1, ok} = ProtocolModuleToDecorate:flush_transport(State0),
56    {State#multiplexed_protocol{protocol_data_to_decorate = State1}, ok}.
57
58close_transport(State = #multiplexed_protocol{protocol_module_to_decorate = ProtocolModuleToDecorate,
59                                                protocol_data_to_decorate = State0}) ->
60    {State1, ok} = ProtocolModuleToDecorate:close_transport(State0),
61    {State#multiplexed_protocol{protocol_data_to_decorate = State1}, ok}.
62
63write(State = #multiplexed_protocol{protocol_module_to_decorate = ProtocolModuleToDecorate,
64                                      protocol_data_to_decorate = State0,
65                                                   service_name = ServiceName},
66      Message = #protocol_message_begin{name = Name}) ->
67    {State1, ok} = ProtocolModuleToDecorate:write(State0,
68                                                  Message#protocol_message_begin{name=ServiceName ++
69                                                                                      ?MULTIPLEXED_SERVICE_SEPARATOR ++
70                                                                                      Name}),
71    {State#multiplexed_protocol{protocol_data_to_decorate = State1}, ok};
72
73write(State = #multiplexed_protocol{protocol_module_to_decorate = ProtocolModuleToDecorate,
74                                      protocol_data_to_decorate = State0},
75      Message) ->
76    {State1, ok} = ProtocolModuleToDecorate:write(State0, Message),
77    {State#multiplexed_protocol{protocol_data_to_decorate = State1}, ok}.
78
79read(State = #multiplexed_protocol{protocol_module_to_decorate = ProtocolModuleToDecorate,
80                                     protocol_data_to_decorate = State0},
81     Message) ->
82    {State1, Result} = ProtocolModuleToDecorate:read(State0, Message),
83    {State#multiplexed_protocol{protocol_data_to_decorate = State1}, Result}.
84