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  }