80c663882a
Summary: First draft. Unit tests pass. Test Plan: unit tests attached Reviewers: heyongqiang Reviewed By: heyongqiang Differential Revision: https://reviews.facebook.net/D3969
83 lines
3.3 KiB
C++
83 lines
3.3 KiB
C++
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
|
|
#ifndef PEEKPROCESSOR_H
|
|
#define PEEKPROCESSOR_H
|
|
|
|
#include <string>
|
|
#include "thrift/lib/cpp/TProcessor.h"
|
|
#include "thrift/lib/cpp/transport/TTransport.h"
|
|
#include "thrift/lib/cpp/transport/TTransportUtils.h"
|
|
#include "thrift/lib/cpp/transport/TBufferTransports.h"
|
|
#include <boost/shared_ptr.hpp>
|
|
|
|
namespace apache { namespace thrift { namespace processor {
|
|
|
|
namespace server {
|
|
class TConnectionContext;
|
|
}
|
|
|
|
/*
|
|
* Class for peeking at the raw data that is being processed by another processor
|
|
* and gives the derived class a chance to change behavior accordingly
|
|
*
|
|
*/
|
|
class PeekProcessor : public apache::thrift::TProcessor {
|
|
|
|
public:
|
|
PeekProcessor();
|
|
virtual ~PeekProcessor();
|
|
|
|
// Input here: actualProcessor - the underlying processor
|
|
// protocolFactory - the protocol factory used to wrap the memory buffer
|
|
// transportFactory - this TPipedTransportFactory is used to wrap the source transport
|
|
// via a call to getPipedTransport
|
|
void initialize(boost::shared_ptr<apache::thrift::TProcessor> actualProcessor,
|
|
boost::shared_ptr<apache::thrift::protocol::TProtocolFactory> protocolFactory,
|
|
boost::shared_ptr<apache::thrift::transport::TPipedTransportFactory> transportFactory);
|
|
|
|
boost::shared_ptr<apache::thrift::transport::TTransport> getPipedTransport(boost::shared_ptr<apache::thrift::transport::TTransport> in);
|
|
|
|
void setTargetTransport(boost::shared_ptr<apache::thrift::transport::TTransport> targetTransport);
|
|
|
|
virtual bool process(boost::shared_ptr<apache::thrift::protocol::TProtocol> in,
|
|
boost::shared_ptr<apache::thrift::protocol::TProtocol> out,
|
|
TConnectionContext* connectionContext);
|
|
|
|
// The following three functions can be overloaded by child classes to
|
|
// achieve desired peeking behavior
|
|
virtual void peekName(const std::string& fname);
|
|
virtual void peekBuffer(uint8_t* buffer, uint32_t size);
|
|
virtual void peek(boost::shared_ptr<apache::thrift::protocol::TProtocol> in,
|
|
apache::thrift::protocol::TType ftype,
|
|
int16_t fid);
|
|
virtual void peekEnd();
|
|
|
|
private:
|
|
boost::shared_ptr<apache::thrift::TProcessor> actualProcessor_;
|
|
boost::shared_ptr<apache::thrift::protocol::TProtocol> pipedProtocol_;
|
|
boost::shared_ptr<apache::thrift::transport::TPipedTransportFactory> transportFactory_;
|
|
boost::shared_ptr<apache::thrift::transport::TMemoryBuffer> memoryBuffer_;
|
|
boost::shared_ptr<apache::thrift::transport::TTransport> targetTransport_;
|
|
};
|
|
|
|
}}} // apache::thrift::processor
|
|
|
|
#endif
|