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
19require 'thrift/protocol/protocol_decorator'
20
21module 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
44end
45