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
10 # http://www.apache.org/licenses/LICENSE-2.0
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
20 from thrift.Thrift import TMessageType
21 from thrift.protocol import TProtocolDecorator
25 class TMultiplexedProtocol(TProtocolDecorator.TProtocolDecorator):
26 def __init__(self, protocol, serviceName):
27 TProtocolDecorator.TProtocolDecorator.__init__(self, protocol)
28 self.serviceName = serviceName
30 def writeMessageBegin(self, name, type, seqid):
31 if (type == TMessageType.CALL or
32 type == TMessageType.ONEWAY):
33 self.protocol.writeMessageBegin(
34 self.serviceName + SEPARATOR + name,
39 self.protocol.writeMessageBegin(name, type, seqid)