vendor/github.com/apache/thrift/lib/java/src/org/apache/thrift/transport/AutoExpandingBufferWriteTransport.java

   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 package org.apache.thrift.transport;
  20
  21 /**
  22  * TTransport for writing to an AutoExpandingBuffer.
  23  */
  24 public final class AutoExpandingBufferWriteTransport extends TTransport {
  25
  26   private final AutoExpandingBuffer buf;
  27   private int pos;
  28
  29   public AutoExpandingBufferWriteTransport(int initialCapacity, double growthCoefficient) {
  30     this.buf = new AutoExpandingBuffer(initialCapacity, growthCoefficient);
  31     this.pos = 0;
  32   }
  33
  34   @Override
  35   public void close() {}
  36
  37   @Override
  38   public boolean isOpen() {return true;}
  39
  40   @Override
  41   public void open() throws TTransportException {}
  42
  43   @Override
  44   public int read(byte[] buf, int off, int len) throws TTransportException {
  45     throw new UnsupportedOperationException();
  46   }
  47
  48   @Override
  49   public void write(byte[] toWrite, int off, int len) throws TTransportException {
  50     buf.resizeIfNecessary(pos + len);
  51     System.arraycopy(toWrite, off, buf.array(), pos, len);
  52     pos += len;
  53   }
  54
  55   public AutoExpandingBuffer getBuf() {
  56     return buf;
  57   }
  58
  59   public int getPos() {
  60     return pos;
  61   }
  62
  63   public void reset() {
  64     pos = 0;
  65   }
  66 }