ceph/src/jaegertracing/thrift/lib/rb/lib/thrift/protocol/multiplexed_protocol.rb

   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