rocksdb/thrift/lib/cpp/transport/TShortReadTransport.h

98 lines
2.5 KiB
C
Raw Normal View History

/*
* 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 _THRIFT_TRANSPORT_TSHORTREADTRANSPORT_H_
#define _THRIFT_TRANSPORT_TSHORTREADTRANSPORT_H_ 1
#include <cstdlib>
#include "thrift/lib/cpp/transport/TTransport.h"
#include "thrift/lib/cpp/transport/TVirtualTransport.h"
namespace apache { namespace thrift { namespace transport { namespace test {
/**
* This class is only meant for testing. It wraps another transport.
* Calls to read are passed through with some probability. Otherwise,
* the read amount is randomly reduced before being passed through.
*
*/
class TShortReadTransport : public TVirtualTransport<TShortReadTransport> {
public:
TShortReadTransport(boost::shared_ptr<TTransport> transport, double full_prob)
: transport_(transport)
, fullProb_(full_prob)
{}
bool isOpen() {
return transport_->isOpen();
}
bool peek() {
return transport_->peek();
}
void open() {
transport_->open();
}
void close() {
transport_->close();
}
uint32_t read(uint8_t* buf, uint32_t len) {
if (len == 0) {
return 0;
}
if (rand()/(double)RAND_MAX >= fullProb_) {
len = 1 + rand()%len;
}
return transport_->read(buf, len);
}
void write(const uint8_t* buf, uint32_t len) {
transport_->write(buf, len);
}
void flush() {
transport_->flush();
}
const uint8_t* borrow(uint8_t* buf, uint32_t* len) {
return transport_->borrow(buf, len);
}
void consume(uint32_t len) {
return transport_->consume(len);
}
boost::shared_ptr<TTransport> getUnderlyingTransport() {
return transport_;
}
protected:
boost::shared_ptr<TTransport> transport_;
double fullProb_;
};
}}}} // apache::thrift::transport::test
#endif // #ifndef _THRIFT_TRANSPORT_TSHORTREADTRANSPORT_H_