Added optional schema to collections

gabynaiman · Sep 20, 2016 · c6eebe0 · c6eebe0
1 parent e3e5384
commit c6eebe0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/rasti/db/collection.rb b/lib/rasti/db/collection.rb
@@ -58,8 +58,9 @@ def many_to_many(name, options={})
 
       end
 
-      def initialize(db)
+      def initialize(db, schema=nil)
         @db = db
+        @schema = schema
       end
 
       def insert(attributes)
@@ -98,10 +99,10 @@ def query(&block)
 
       private
 
-      attr_reader :db
+      attr_reader :db, :schema
 
       def dataset
-        db[self.class.collection_name]
+        db[schema.nil? ? self.class.collection_name : "#{schema}__#{self.class.collection_name}".to_sym]
       end
 
     end