github.com/matm/etcd@v0.3.1-0.20140328024009-5b4a473f1453/third_party/code.google.com/p/gogoprotobuf/proto/extensions_gogo.go (about) 1 // Copyright (c) 2013, Vastech SA (PTY) LTD. All rights reserved. 2 // http://code.google.com/p/gogoprotobuf/gogoproto 3 // 4 // Redistribution and use in source and binary forms, with or without 5 // modification, are permitted provided that the following conditions are 6 // met: 7 // 8 // * Redistributions of source code must retain the above copyright 9 // notice, this list of conditions and the following disclaimer. 10 // * Redistributions in binary form must reproduce the above 11 // copyright notice, this list of conditions and the following disclaimer 12 // in the documentation and/or other materials provided with the 13 // distribution. 14 // 15 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 16 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 17 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 18 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 19 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 21 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 25 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 27 package proto 28 29 import ( 30 "bytes" 31 "fmt" 32 "reflect" 33 "sort" 34 ) 35 36 func GetBoolExtension(pb extendableProto, extension *ExtensionDesc, ifnotset bool) bool { 37 if reflect.ValueOf(pb).IsNil() { 38 return ifnotset 39 } 40 value, err := GetExtension(pb, extension) 41 if err != nil { 42 return ifnotset 43 } 44 if value == nil { 45 return ifnotset 46 } 47 if value.(*bool) == nil { 48 return ifnotset 49 } 50 return *(value.(*bool)) 51 } 52 53 func (this *Extension) Equal(that *Extension) bool { 54 return bytes.Equal(this.enc, that.enc) 55 } 56 57 func SizeOfExtensionMap(m map[int32]Extension) (n int) { 58 return sizeExtensionMap(m) 59 } 60 61 func EncodeExtensionMap(m map[int32]Extension, data []byte) (n int, err error) { 62 if err := encodeExtensionMap(m); err != nil { 63 return 0, err 64 } 65 keys := make([]int, 0, len(m)) 66 for k := range m { 67 keys = append(keys, int(k)) 68 } 69 sort.Ints(keys) 70 for _, k := range keys { 71 n += copy(data[n:], m[int32(k)].enc) 72 } 73 return n, nil 74 } 75 76 func GetRawExtension(m map[int32]Extension, id int32) ([]byte, error) { 77 if m[id].value == nil || m[id].desc == nil { 78 return m[id].enc, nil 79 } 80 if err := encodeExtensionMap(m); err != nil { 81 return nil, err 82 } 83 return m[id].enc, nil 84 } 85 86 func NewExtension(e []byte) Extension { 87 ee := Extension{enc: make([]byte, len(e))} 88 copy(ee.enc, e) 89 return ee 90 } 91 92 func (this Extension) GoString() string { 93 return fmt.Sprintf("proto.NewExtension(%#v)", this.enc) 94 }