mongodb-mongo_record 0.3 → 0.3.1

data/README.rdoc

@@ -18,6 +18,8 @@ A quick code sample:
   class Track < MongoRecord::Base
     collection_name :tracks
     fields :artist, :album, :song, :track
+    index :artist
+
     def to_s
       "artist: #{artist}, album: #{album}, song: #@song, track: #{@track ? @track.to_i : nil}"
     end
@@ -81,6 +83,5 @@ Jim Mulholland, jim at squeejee dot com
 * Assigning created_at and updated_at fields even if they are not declared
 * Alias methods for "first", "last" and "all"
 
-== To Do
-
-* DBRefs
+Clinton R. Nixon, crnixon at gmail dot com
+* Ability to define and query indexes from models

data/lib/mongo_record/base.rb

@@ -147,6 +147,41 @@ module MongoRecord
       # Return the field names.
       def field_names; @field_names; end
 
+      # Creates an index for this collection.
+      # +fields+ should be either a single field name (:title)
+      # or an array of fields ([:title, :author, :date])
+      # or an array of a field name and direction ([:title, :asc] or [:title, :desc])
+      # or an array of field names and directions ([[:title, :asc], [:author, :desc]])
+      # +unique+ should be true or false and indicates whether this index
+      # should enforce a uniqueness constraint.
+      def index(fields, unique = false)
+        fields = Array(fields)
+        
+        if fields.length == 2 && 
+          ( fields[1].to_s == 'asc' || fields[1].to_s == 'desc' ||
+            fields[1] == XGen::Mongo::ASCENDING || fields[1] == XGen::Mongo::DESCENDING )
+          fields = [fields]
+        end
+        
+        fields = fields.map do |field| 
+          field = field.respond_to?(:[]) ? field : [field, :asc]
+          field[1] = (field[1] == :desc) ? XGen::Mongo::DESCENDING : XGen::Mongo::ASCENDING
+          field          
+        end
+        
+        collection.create_index(fields, unique)
+      end
+
+      # Returns list of indexes for model, unless fields are passed.
+      # In that case, creates an index.
+      def indexes(*fields)
+        if fields.empty?
+          collection.index_information
+        else
+          index(*fields)
+        end
+      end
+
       # Return the names of all instance variables that hold objects
       # declared using has_one. The names do not start with '@'.
       #

data/mongo-activerecord-ruby.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = 'mongo_record'
-  s.version = '0.3'
+  s.version = '0.3.1'
   s.platform = Gem::Platform::RUBY
   s.summary = 'ActiveRecord-like models for the 10gen Mongo DB'
   s.description = 'MongoRecord is an ActiveRecord-like framework for the 10gen Mongo database. For more information about Mongo, see http://www.mongodb.org.'

data/tests/test_mongo.rb

@@ -24,6 +24,7 @@ require File.join(File.dirname(__FILE__), 'class_in_module')
 class Track < MongoRecord::Base
   collection_name :tracks
   fields :artist, :album, :song, :track, :created_at
+
   def to_s
     # Uses both accessor methods and ivars themselves
     "artist: #{artist}, album: #{album}, song: #@song, track: #{@track ? @track.to_i : nil}"
@@ -743,4 +744,20 @@ class MongoTest < Test::Unit::TestCase
     assert_equal p.artist, "Nickleback"
   end
 
+  def test_indexing
+    Track.index :artist
+    Track.index [:artist, :created_at]
+    Track.index [:song, :desc], true
+    Track.index [:artist, [:album, :desc]]
+    Track.index [:created_at, XGen::Mongo::ASCENDING]
+
+    assert_equal([ {"artist" => 1}, 
+                   {"artist" => 1, "created_at" => 1}, 
+                   {"song" => -1}, 
+                   {"artist" => 1, "album" => -1},
+                   {"created_at" => 1} ],
+                 Track.indexes.map {|index| index[:keys]}.delete_if { |keys| 
+                   keys.has_key? "_id" })
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: mongodb-mongo_record
 version: !ruby/object:Gem::Version 
-  version: "0.3"
+  version: 0.3.1
 platform: ruby
 authors: 
 - Jim Menard
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-04-08 00:00:00 -07:00
+date: 2009-05-16 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency