tmemory_buffer.go

/*
 * 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.
 */

package thrift

import (
  "bytes"
  "io"
)

/**
 * Memory buffer-based implementation of the TTransport interface.
 *
 */
type TMemoryBuffer struct {
  buf  *bytes.Buffer
  size int
}

type TMemoryBufferTransportFactory struct {
  size int
}

func (p *TMemoryBufferTransportFactory) GetTransport(trans TTransport) TTransport {
  if trans != nil {
    t, ok := trans.(*TMemoryBuffer)
    if ok && t.size > 0 {
      return NewTMemoryBufferLen(t.size)
    }
  }
  return NewTMemoryBufferLen(p.size)
}

func NewTMemoryBufferTransportFactory(size int) *TMemoryBufferTransportFactory {
  return &TMemoryBufferTransportFactory{size: size}
}

func NewTMemoryBuffer() *TMemoryBuffer {
  return &TMemoryBuffer{buf: &bytes.Buffer{}, size: 0}
}

func NewTMemoryBufferLen(size int) *TMemoryBuffer {
  buf := make([]byte, 0, size)
  return &TMemoryBuffer{buf: bytes.NewBuffer(buf), size: size}
}

func (p *TMemoryBuffer) IsOpen() bool {
  return true
}

func (p *TMemoryBuffer) Open() error {
  return nil
}

func (p *TMemoryBuffer) Peek() bool {
  return p.IsOpen()
}

func (p *TMemoryBuffer) Close() error {
  p.buf.Reset()
  return nil
}

func (p *TMemoryBuffer) Read(buf []byte) (int, error) {
  return p.buf.Read(buf)
}

func (p *TMemoryBuffer) ReadAll(buf []byte) (int, error) {
  return ReadAllTransport(p, buf)
}

func (p *TMemoryBuffer) ReadByte() (byte, error) {
  return p.buf.ReadByte()
}

func (p *TMemoryBuffer) ReadFrom(r io.Reader) (int64, error) {
  return p.buf.ReadFrom(r)
}

func (p *TMemoryBuffer) Write(buf []byte) (int, error) {
  return p.buf.Write(buf)
}

func (p *TMemoryBuffer) WriteString(buf string) (int, error) {
  return p.buf.WriteString(buf)
}

func (p *TMemoryBuffer) WriteTo(w io.Writer) (int64, error) {
  return p.buf.WriteTo(w)
}

func (p *TMemoryBuffer) Flush() error {
  return nil
}

func (p *TMemoryBuffer) Reset() {
  p.buf.Reset()
}

func (p *TMemoryBuffer) Bytes() []byte {
  return p.buf.Bytes()
}

func (p *TMemoryBuffer) Len() int {
  return p.buf.Len()
}

func (p *TMemoryBuffer) String() string {
  return p.buf.String()
}