package com.cloudera.flume.handlers.thrift;
import java.util.concurrent.atomic.AtomicLong;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.TTransportException;
TTransport trans;
AtomicLong bytesRead = new AtomicLong();
AtomicLong bytesWritten = new AtomicLong();
this.trans = trans;
}
@Override
trans.close();
}
@Override
return trans.isOpen();
}
@Override
public void open()
throws TTransportException {
trans.open();
}
@Override
public int read(
byte[] buf,
int off,
int len)
throws TTransportException {
int ret = trans.read(buf, off, len);
bytesRead.addAndGet(ret);
return ret;
}
@Override
public void flush()
throws TTransportException {
trans.flush();
}
public void write(
byte[] buf,
int off,
int len)
throws TTransportException {
trans.write(buf, off, len);
bytesWritten.addAndGet(len);
}
return bytesRead.get();
}
return bytesWritten.get();
}
}