metrix_db 0.0.2 → 0.0.3

data/Gemfile.lock

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

data/README.md

@@ -14,7 +14,7 @@ Usage
 
     class MyDB
       include MetrixDB
-      field :col1
+      field :col1, :uniq => false
       field :col2
       dataset [[1,2],[3,4]]
     end
@@ -25,6 +25,14 @@ Usage
 * Set dataset, dataset is the real data of one DB
 
 
+Field
+-----
+
+Options for field definition
+
+  `uniq`: Check if column is unique. Default value is true.
+
+
 Query
 -----
 

data/lib/metrix_db/version.rb

@@ -1,3 +1,3 @@
 module MetrixDb
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/metrix_db.rb

@@ -1,4 +1,6 @@
 module MetrixDB
+  class UniqueFieldError < Exception; end
+
   def self.included(base)
     base.extend ClassMethods
     base.send(:include, InstanceMethods)
@@ -11,9 +13,14 @@ module MetrixDB
   end
 
   module ClassMethods 
-    def field(name)
+    def field(name, options = {})
+      default_options = {:uniq => true}
+      options = default_options.merge(options)
       @fields ||= {}
-      @fields[name.to_sym] = @fields.size 
+      index = @fields.size 
+      @fields[name.to_sym] = index
+      @uniq_fields ||= {}
+      @uniq_fields[index] = name.to_s if options[:uniq] 
       define_method "#{name}" do  
         @data[self.class.fields[name.to_sym]]
       end
@@ -25,9 +32,21 @@ module MetrixDB
 
     def dataset(_dataset)
       @dataset = _dataset
+      check_uniq_fields(_dataset)
+    end
+
+    def check_uniq_fields(_dataset) 
+      return unless @uniq_fields
+      @uniq_fields.each_pair do |index, field_name| 
+        col_data = _dataset.map {|data| data[index]}
+        if col_data.uniq! 
+          raise UniqueFieldError.new("Field '#{field_name}' is not unique")
+        end
+      end
     end
 
     def where(conditions={}) 
+      return nil if @dataset.nil? || @fields.nil? 
       @dataset.each do |data|
         conditions.each_pair do |k, v|
           return self.new(data) if data[@fields[k.to_sym]] == v

data/spec/metrix_db/field_spec.rb

@@ -28,4 +28,23 @@ describe MetrixDB do
       ins.field2.should == 2
     end
   end
+
+  context "unique field" do 
+    it "should raise exception if field is unique" do 
+      lambda { @class = Class.new {
+        include MetrixDB 
+        field :field
+        dataset [[1],[1]]
+      }}.should raise_error MetrixDB::UniqueFieldError, "Field 'field' is not unique"
+    end
+
+    it "should not raise exception if field is not unique" do 
+      lambda { @class = Class.new {
+        include MetrixDB 
+        field :field, :uniq => false
+        dataset [[1],[1]]
+      }}.should_not raise_error
+    end
+
+  end
 end

data/spec/metrix_db/where_spec.rb

@@ -3,7 +3,7 @@ require 'spec_helper'
 describe MetrixDB do 
   context ".where" do 
     before do 
-      dataset = [[1,2],[2,3]]
+      dataset = [[1,2],[2,3]] 
       @dataset = dataset
       @class = Class.new do 
         include MetrixDB
@@ -38,6 +38,7 @@ describe MetrixDB do
        inst.col1.should == 1
        inst.col2.should == 2
     end
+
   end 
 end
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: metrix_db
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: false
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Allen Wei