shoden 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 47609f0a06edadfafbcc844d6b91d23eca91244e
-  data.tar.gz: e7507a40336852be18a4cd30e7c3b78168f4bc3f
+  metadata.gz: c7c861e4cb7f7edd9609864081c18f45913c8a82
+  data.tar.gz: af092fb9fb7046d81bc703d90900e9f2bb83905f
 SHA512:
-  metadata.gz: aa632c8f3b300e52655d5a3c5ce27777e348e5cea84f8e75d6533c594911bc367c1f765f0bf94633081d192bf112e339a8952528058c878369c734604e840cfb
-  data.tar.gz: 2bc95a32031322ca1601cff586950f305fb12623829848c1207d9a9bac383f9c36382fdd137244f45e23a0ab0755b1c2bad604a90a642e5a147d5e0ed67a14f4
+  metadata.gz: faef768a2f83007e65b9bbaa5ae81414166175b55fac9de499faff85445af98390409572a608f7f54a13b63d9feeeb5933599b7c263aafc4ecd345237c8ab21a
+  data.tar.gz: a24ed94a720330c281f39a88df8ba7868a72438818acb3ecafd6b97984f2afcb96a419ecb4881cc68071e990256569a8dc7e6f0f16bf67cc26678c673398de91

data/README.md

@@ -1,4 +1,4 @@
-# Shôden
+# Shôden - [![Gem Version](https://badge.fury.io/rb/shoden.svg)](http://badge.fury.io/rb/shoden)
 
 ![Elephant god](http://www.redprintdna.com/wp-content/uploads/2011/09/L-Elephant-Against-Sky.jpg)
 
@@ -46,3 +46,15 @@ class Post < Shoden::Model
   reference :owner, :User
 end
 ```
+
+## Indexing
+
+```ruby
+class User < Shoden::Model
+  attribute :email
+  attribute :country
+
+  index  :country
+  unique :email
+end
+```

data/lib/shoden.rb

@@ -3,8 +3,10 @@ require 'sequel'
 Sequel.extension :pg_hstore, :pg_hstore_ops
 
 module Shoden
-  MissingID = Class.new(StandardError)
-  NotFound  = Class.new(StandardError)
+  Error                = Class.new(StandardError)
+  MissingID            = Class.new(Error)
+  NotFound             = Class.new(Error)
+  UniqueIndexViolation = Class.new(Error)
 
   Proxy = Struct.new(:klass, :parent) do
     def create(args = {})
@@ -62,9 +64,16 @@ module Shoden
       if defined? @_id
         table.where(id: @_id).update data: sanitized_attrs
       else
-        @_id = table.insert data: sanitized_attrs
+        begin
+          @_id = table.insert data: sanitized_attrs
+        rescue Sequel::UniqueConstraintViolation
+          raise UniqueIndexViolation
+        end
       end
 
+      self.class.indices.each { |i| create_index(i) }
+      self.class.uniques.each { |i| create_index(i, :unique) }
+
       self
     end
 
@@ -74,6 +83,18 @@ module Shoden
       self
     end
 
+    def self.all
+      collect
+    end
+
+    def self.first
+      collect("ORDER BY id ASC LIMIT 1").first
+    end
+
+    def self.last
+      collect("ORDER BY id DESC LIMIT 1").first
+    end
+
     def self.create(attrs = {})
       new(attrs).save
     end
@@ -82,10 +103,26 @@ module Shoden
       @attributes ||= []
     end
 
+    def self.indices
+      @indices ||= []
+    end
+
+    def self.uniques
+      @uniques ||= []
+    end
+
     def self.[](id)
       new(id: id).load!
     end
 
+    def self.index(name)
+      indices << name if !indices.include?(name)
+    end
+
+    def self.unique(name)
+      uniques << name if !uniques.include?(name)
+    end
+
     def self.attribute(name, caster = ->(x) { x })
       attributes << name if !attributes.include?(name)
 
@@ -117,6 +154,19 @@ module Shoden
 
     private
 
+    def self.collect(condition = '')
+      models = []
+      Shoden.connection.fetch("SELECT * FROM \"#{table_name}\" #{condition}") do |r|
+        attrs = r[:data].merge(id: r[:id])
+        models << new(attrs)
+      end
+      models
+    end
+
+    def self.table_name
+      :"Shoden::#{self.name}"
+    end
+
     def self.to_reference
       name.to_s.
         match(/^(?:.*::)*(.*)$/)[1].
@@ -124,11 +174,18 @@ module Shoden
         downcase.to_sym
     end
 
+    def create_index(name, type = '')
+      conn.execute <<EOS
+        CREATE #{type.upcase} INDEX index_#{self.class.name}_#{name}
+        ON "#{table_name}" (( data -> '#{name}'))
+        WHERE ( data ? '#{name}' );
+EOS
+    end
+
     def sanitized_attrs
       sanitized = @attributes.map do |k, _|
         val = send(k)
-        return if !val
-
+        return if val.nil?
         [k, val.to_s]
       end.compact
 
@@ -153,7 +210,7 @@ module Shoden
     end
 
     def table_name
-      :"Shoden::#{self.class.name}"
+      self.class.table_name
     end
 
     def table

data/shoden.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name              = "shoden"
-  s.version           = "0.2.0"
+  s.version           = "0.3.0"
   s.summary           = "Object hash mapper for postgres"
   s.description       = "Slim postgres models"
   s.authors           = ["elcuervo"]

data/test/shoden_test.rb

@@ -78,3 +78,29 @@ test 'casting' do
 
   assert_equal a_prime.n, 1
 end
+
+test 'indices' do
+  class Person < Shoden::Model
+    attribute :email
+    attribute :origin
+
+    index  :origin
+    unique :email
+  end
+
+  person = Person.create(email: 'elcuervo@elcuervo.net', origin: 'The internerd')
+
+  assert person.id
+
+  assert_raise Shoden::UniqueIndexViolation do
+    Person.create(email: 'elcuervo@elcuervo.net', origin: 'Montevideo City')
+  end
+end
+
+test 'basic querying' do
+  5.times { User.create }
+
+  assert_equal User.all.size, 5
+  assert_equal User.first.id, 1
+  assert_equal User.last.id,  5
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: shoden
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - elcuervo
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-10-28 00:00:00.000000000 Z
+date: 2014-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel