Sha256: 65614aec63cecdf8eeb3eb4dad61e39d8a4bfd764f1efae0609f485f0c4de554

Contents?: true

Size: 1.34 KB

Versions: 8

Compression:

Stored size: 1.34 KB

Contents

require 'protobuf/field/varint_field'

module Protobuf
  module Field
    class EnumField < VarintField

      ##
      # Class Methods
      #

      def self.default
        raise NoMethodError, "#{self}.#{__method__} must be called on an instance"
      end

      ##
      # Public Instance Methods
      #

      def acceptable?(val)
        ! type_class.fetch(val).nil?
      end

      def encode(value)
        super(value.to_i)
      end

      def decode(value)
        if acceptable?(value)
          value
        end
      end

      def enum?
        true
      end

      private

      ##
      # Private Instance Methods
      #

      def define_setter
        field = self
        message_class.class_eval do
          define_method("#{field.name}=") do |value|
            orig_value = value
            if value.nil?
              @values.delete(field.name)
            else
              value = field.type_class.fetch(value)
              raise TypeError, "Invalid Enum value: #{orig_value.inspect} for #{field.name}" unless value

              @values[field.name] = value
            end
          end
        end
      end

      def typed_default_value
        if default.is_a?(Symbol)
          type_class.const_get(default)
        else
          type_class.fetch(default) || type_class.enums.first
        end
      end

    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
protobuf-3.3.6 lib/protobuf/field/enum_field.rb
protobuf-3.3.5 lib/protobuf/field/enum_field.rb
protobuf-3.3.4 lib/protobuf/field/enum_field.rb
protobuf-3.3.3 lib/protobuf/field/enum_field.rb
protobuf-3.3.2 lib/protobuf/field/enum_field.rb
protobuf-3.3.1 lib/protobuf/field/enum_field.rb
protobuf-3.3.0 lib/protobuf/field/enum_field.rb
protobuf-3.2.1 lib/protobuf/field/enum_field.rb