github.com/gogo/protobuf@v1.3.2/vanity/field.go (about)

     1  // Protocol Buffers for Go with Gadgets
     2  //
     3  // Copyright (c) 2015, The GoGo Authors.  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 vanity
    30  
    31  import (
    32  	"github.com/gogo/protobuf/gogoproto"
    33  	"github.com/gogo/protobuf/proto"
    34  	descriptor "github.com/gogo/protobuf/protoc-gen-gogo/descriptor"
    35  )
    36  
    37  func FieldHasBoolExtension(field *descriptor.FieldDescriptorProto, extension *proto.ExtensionDesc) bool {
    38  	if field.Options == nil {
    39  		return false
    40  	}
    41  	value, err := proto.GetExtension(field.Options, extension)
    42  	if err != nil {
    43  		return false
    44  	}
    45  	if value == nil {
    46  		return false
    47  	}
    48  	if value.(*bool) == nil {
    49  		return false
    50  	}
    51  	return true
    52  }
    53  
    54  func SetBoolFieldOption(extension *proto.ExtensionDesc, value bool) func(field *descriptor.FieldDescriptorProto) {
    55  	return func(field *descriptor.FieldDescriptorProto) {
    56  		if FieldHasBoolExtension(field, extension) {
    57  			return
    58  		}
    59  		if field.Options == nil {
    60  			field.Options = &descriptor.FieldOptions{}
    61  		}
    62  		if err := proto.SetExtension(field.Options, extension, &value); err != nil {
    63  			panic(err)
    64  		}
    65  	}
    66  }
    67  
    68  func TurnOffNullable(field *descriptor.FieldDescriptorProto) {
    69  	if field.IsRepeated() && !field.IsMessage() {
    70  		return
    71  	}
    72  	SetBoolFieldOption(gogoproto.E_Nullable, false)(field)
    73  }
    74  
    75  func TurnOffNullableForNativeTypes(field *descriptor.FieldDescriptorProto) {
    76  	if field.IsRepeated() || field.IsMessage() {
    77  		return
    78  	}
    79  	SetBoolFieldOption(gogoproto.E_Nullable, false)(field)
    80  }
    81  
    82  func TurnOffNullableForNativeTypesWithoutDefaultsOnly(field *descriptor.FieldDescriptorProto) {
    83  	if field.IsRepeated() || field.IsMessage() {
    84  		return
    85  	}
    86  	if field.DefaultValue != nil {
    87  		return
    88  	}
    89  	SetBoolFieldOption(gogoproto.E_Nullable, false)(field)
    90  }