1 // Protocol Buffers for Go with Gadgets 2 // 3 // Copyright (c) 2013, The GoGo Authors. All rights reserved. 4 // http://github.com/gogo/protobuf 5 // 6 // Redistribution and use in source and binary forms, with or without 7 // modification, are permitted provided that the following conditions are 8 // met: 9 // 10 // * Redistributions of source code must retain the above copyright 11 // notice, this list of conditions and the following disclaimer. 12 // * Redistributions in binary form must reproduce the above 13 // copyright notice, this list of conditions and the following disclaimer 14 // in the documentation and/or other materials provided with the 15 // distribution. 16 // 17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29 package io 30 31 import ( 32 "github.com/gogo/protobuf/proto" 33 "io" 34 ) 35 36 type Writer interface { 37 WriteMsg(proto.Message) error 38 } 39 40 type WriteCloser interface { 41 Writer 42 io.Closer 43 } 44 45 type Reader interface { 46 ReadMsg(msg proto.Message) error 47 } 48 49 type ReadCloser interface { 50 Reader 51 io.Closer 52 } 53 54 type marshaler interface { 55 MarshalTo(data []byte) (n int, err error) 56 } 57 58 func getSize(v interface{}) (int, bool) { 59 if sz, ok := v.(interface { 60 Size() (n int) 61 }); ok { 62 return sz.Size(), true 63 } else if sz, ok := v.(interface { 64 ProtoSize() (n int) 65 }); ok { 66 return sz.ProtoSize(), true 67 } else { 68 return 0, false 69 } 70 } 71