mongodb/mongoid

# frozen_string_literal: true
# rubocop:todo all

module Mongoid
  module Association

    # Mixin module for Mongoid::Association::Proxy which adds
    # custom Marshal.dump functionality.
    module Marshalable

      # Provides the data needed to Marshal.dump an association proxy.
      #
      # @example Dump the proxy.
      #   Marshal.dump(proxy)
      #
      # @return [ Array<Object> ] The dumped data.
      def marshal_dump
        [ _base, _target, _association ]
      end

      # Takes the provided data and sets it back on the proxy.
      #
      # @example Load the proxy.
      #   Marshal.load(proxy)
      #
      # @param [ Array<Object> ] data The data to set on the proxy.
      #
      # @return [ Array<Object> ] The loaded data.
      def marshal_load(data)
        @_base, @_target, @_association = data
        extend_proxy(_association.extension) if _association.extension
      end
    end
  end
end