Contents

module CompositePrimaryKeys
  ID_SEP     = ','
  ID_SET_SEP = ';'

  module ArrayExtension
    def to_composite_keys
      CompositeKeys.new(self)
    end
  end

  def self.normalize(ids)
    ids.map do |id|
      if id.is_a?(Array)
        normalize(id)
      elsif id.is_a?(String) && id.index(ID_SEP)
        id.split(ID_SEP)
      else
        id
      end
    end
  end

  class CompositeKeys < Array

    def self.parse(value)
      case value
      when Array
        value.to_composite_keys
      when String
        self.new(value.split(ID_SEP))
      else
        raise(ArgumentError, "Unsupported type: #{value}")
      end
    end

    def to_s
      # Doing this makes it easier to parse Base#[](attr_name)
      join(ID_SEP)
    end
  end
end

Array.send(:include, CompositePrimaryKeys::ArrayExtension)

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
composite_primary_keys-9.0.1	lib/composite_primary_keys/composite_arrays.rb
composite_primary_keys-9.0.0	lib/composite_primary_keys/composite_arrays.rb
composite_primary_keys-9.0.0.beta1	lib/composite_primary_keys/composite_arrays.rb