Contents

require "active_support/concern"

module Protobuf
  module ActiveRecord
    module Persistence
      extend ::ActiveSupport::Concern

      module ClassMethods
        # :nodoc:
        def create(attributes = {}, &block)
          attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

          super(attributes, &block)
        end

        # :nodoc:
        def create!(attributes = {}, &block)
          attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

          super(attributes, &block)
        end
      end

      # Override Active Record's initialize method so it can accept a protobuf
      # message as it's attributes.
      # :noapi:
      def initialize(*args, &block)
        args[0] = attributes_from_proto(args.first) if args.first.is_a?(::Protobuf::Message)

        super(*args, &block)
      end

      # :nodoc:
      def assign_attributes(attributes)
        attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

        super(attributes)
      end

      # :nodoc:
      def update_attributes(attributes)
        attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

        super(attributes)
      end

      # :nodoc:
      def update_attributes!(attributes)
        attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

        super(attributes)
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
protobuf-activerecord-5.2.0	lib/protobuf/active_record/persistence.rb
protobuf-activerecord-5.1.0	lib/protobuf/active_record/persistence.rb
protobuf-activerecord-5.0.0	lib/protobuf/active_record/persistence.rb
protobuf-activerecord-5.1.0.beta	lib/protobuf/active_record/persistence.rb
protobuf-activerecord-5.0.0.beta	lib/protobuf/active_record/persistence.rb
protobuf-activerecord-3.6.1	lib/protobuf/active_record/persistence.rb