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 29package io 30 31import ( 32 "github.com/gogo/protobuf/proto" 33 "io" 34) 35 36type Writer interface { 37 WriteMsg(proto.Message) error 38} 39 40type WriteCloser interface { 41 Writer 42 io.Closer 43} 44 45type Reader interface { 46 ReadMsg(msg proto.Message) error 47} 48 49type ReadCloser interface { 50 Reader 51 io.Closer 52} 53 54type marshaler interface { 55 MarshalTo(data []byte) (n int, err error) 56} 57 58func 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