ceph/src/jaegertracing/thrift/lib/go/thrift/rich_transport.go

   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
  20 package thrift
  21
  22 import "io"
  23
  24 type RichTransport struct {
  25         TTransport
  26 }
  27
  28 // Wraps Transport to provide TRichTransport interface
  29 func NewTRichTransport(trans TTransport) *RichTransport {
  30         return &RichTransport{trans}
  31 }
  32
  33 func (r *RichTransport) ReadByte() (c byte, err error) {
  34         return readByte(r.TTransport)
  35 }
  36
  37 func (r *RichTransport) WriteByte(c byte) error {
  38         return writeByte(r.TTransport, c)
  39 }
  40
  41 func (r *RichTransport) WriteString(s string) (n int, err error) {
  42         return r.Write([]byte(s))
  43 }
  44
  45 func (r *RichTransport) RemainingBytes() (num_bytes uint64) {
  46         return r.TTransport.RemainingBytes()
  47 }
  48
  49 func readByte(r io.Reader) (c byte, err error) {
  50         v := [1]byte{0}
  51         n, err := r.Read(v[0:1])
  52         if n > 0 && (err == nil || err == io.EOF) {
  53                 return v[0], nil
  54         }
  55         if n > 0 && err != nil {
  56                 return v[0], err
  57         }
  58         if err != nil {
  59                 return 0, err
  60         }
  61         return v[0], nil
  62 }
  63
  64 func writeByte(w io.Writer, c byte) error {
  65         v := [1]byte{c}
  66         _, err := w.Write(v[0:1])
  67         return err
  68 }