]>
Commit | Line | Data |
---|---|---|
f67539c2 TL |
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 | require 'thrift/protocol/protocol_decorator' | |
20 | ||
21 | module Thrift | |
22 | class MultiplexedProtocol < BaseProtocol | |
23 | ||
24 | include ProtocolDecorator | |
25 | ||
26 | def initialize(protocol, service_name) | |
27 | super(protocol) | |
28 | @service_name = service_name | |
29 | end | |
30 | ||
31 | def write_message_begin(name, type, seqid) | |
32 | case type | |
33 | when MessageTypes::CALL, MessageTypes::ONEWAY | |
34 | @protocol.write_message_begin("#{@service_name}:#{name}", type, seqid) | |
35 | else | |
36 | @protocol.write_message_begin(name, type, seqid) | |
37 | end | |
38 | end | |
39 | ||
40 | def to_s | |
41 | "multiplexed(#{@service_name=@protocol.to_s})" | |
42 | end | |
43 | end | |
44 | end |