metrix_db 0.0.4 → 0.0.5

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    metrix_db (0.0.3)
+    metrix_db (0.0.4)
 
 GEM
   remote: http://rubygems.org/

data/README.md

@@ -15,7 +15,7 @@ Usage
     class MyDB
       include MetrixDB
       field :col1, :uniq => false
-      field :col2
+      field :col2, :ref => true
       dataset [[1,2],[3,4]]
     end
 
@@ -31,6 +31,7 @@ Field
 Options for field definition
 
   `uniq`: Check if column is unique. Default value is true.
+  `ref`: Support `[]` query syntax. If not set for all field will use first field as ref field
 
 
 Query
@@ -45,6 +46,11 @@ or alias :one
     mydb = MyDB.on(:col1 => 1) # nil or instance of MyDB.
     mydb.col1 # => 1
     mydb.col2 # => 2
+
+use `[]` as query syntax, it will use ref field to query data.
+    
+    mydb = MyDB.[1] # nil or instance of MyDB.
+    
     
 
 

data/lib/metrix_db.rb

@@ -1,5 +1,6 @@
 module MetrixDB
   class UniqueFieldError < Exception; end
+  class UsageError < Exception; end
 
   def self.included(base)
     base.extend ClassMethods
@@ -19,19 +20,25 @@ module MetrixDB
       @fields ||= {}
       index = @fields.size 
       @fields[name.to_sym] = index
+
       @uniq_fields ||= {}
       @uniq_fields[index] = name.to_s if options[:uniq] 
+
+      raise UsageError.new("Already defined '#{@ref_field}' as ref column") if options[:ref] && @ref_field 
+      @ref_field = name.to_sym if options[:ref]
+
       define_method "#{name}" do  
         @data[self.class.fields[name.to_sym]]
       end
     end
 
     def fields
-      @fields ||= []
+      @fields ||= {}
     end
 
     def dataset(_dataset)
       @dataset = _dataset
+      @ref_field ||= @fields.keys.first
       check_uniq_fields(_dataset)
     end
 
@@ -52,7 +59,12 @@ module MetrixDB
           return self.new(data) if data[@fields[k.to_sym]] == v
         end
       end
+      return nil
     end
     alias :on :first
+
+    def [](value) 
+      self.first(@ref_field => value)
+    end
   end
 end

data/lib/metrix_db/version.rb

@@ -1,3 +1,3 @@
 module MetrixDb
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

data/spec/metrix_db/data_spec.rb

@@ -7,6 +7,7 @@ describe MetrixDB do
       @dataset = dataset
       @class = Class.new do 
         include MetrixDB
+        field :field1
         dataset dataset
       end      
     end

data/spec/metrix_db/field_spec.rb

@@ -47,7 +47,36 @@ describe MetrixDB do
     end
   end
 
-  context "" do 
+  context "ref index" do 
+    it "should use first field as ref field" do 
+      klass = Class.new do 
+        include MetrixDB 
+        field :field1  
+        field :field2
+        dataset [[1,2],[2,3]]
+      end
+
+      klass.instance_variable_get("@ref_field").should == :field1
+    end
+
+    it "can set ref field" do 
+      klass = Class.new do 
+        include MetrixDB 
+        field :field1
+        field :field2, :ref => true
+        dataset [[1,2],[2,3]]
+      end
+      klass.instance_variable_get("@ref_field").should == :field2
+    end
+
+     it "will raise error if define two ref field" do 
+      lambda {Class.new {
+        include MetrixDB 
+        field :field1, :ref => true
+        field :field2, :ref => true
+        dataset [[1,2],[2,3]]
+      }}.should raise_error MetrixDB::UsageError, "Already defined 'field1' as ref column"
+    end
 
   end
 end

data/spec/metrix_db/{first_spec.rb → query_spec.rb}

@@ -38,7 +38,19 @@ describe MetrixDB do
        inst.col1.should == 1
        inst.col2.should == 2
     end
-
   end 
+
+  context "ref query syntax []" do 
+    it "use first field as search key" do 
+      klass = Class.new do 
+        include MetrixDB 
+        field :field1
+        field :field2
+        dataset [[1,2],[2,3]]
+      end
+      klass[2].data.should == [2,3]
+      klass[123].should be_nil
+    end
+  end
 end
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: metrix_db
 version: !ruby/object:Gem::Version 
-  hash: 23
+  hash: 21
   prerelease: false
   segments: 
   - 0
   - 0
-  - 4
-  version: 0.0.4
+  - 5
+  version: 0.0.5
 platform: ruby
 authors: 
 - Allen Wei
@@ -68,7 +68,7 @@ files:
 - metrix_db.gemspec
 - spec/metrix_db/data_spec.rb
 - spec/metrix_db/field_spec.rb
-- spec/metrix_db/first_spec.rb
+- spec/metrix_db/query_spec.rb
 - spec/metrix_db_spec.rb
 - spec/spec_helper.rb
 has_rdoc: true
@@ -108,6 +108,6 @@ summary: Simple DB base on Ruby Array
 test_files: 
 - spec/metrix_db/data_spec.rb
 - spec/metrix_db/field_spec.rb
-- spec/metrix_db/first_spec.rb
+- spec/metrix_db/query_spec.rb
 - spec/metrix_db_spec.rb
 - spec/spec_helper.rb