active_tsv 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9aed9addfc8d5c3e3edfc4ba6bc47b4bc3da51f1
-  data.tar.gz: 25dd4eb7b93d1967c361fff185576ba589acb9f6
+  metadata.gz: fa1189fb2be4c0a87c3e95522ea91a23cf35f956
+  data.tar.gz: 18aa344f90494fe0a3d42ae74688a2771598eed9
 SHA512:
-  metadata.gz: a3ebbfd04906f7a62897a3affe6640c12d610e2bf6ebb012f48280e6334b9a4a870ef930117b29111e32ae11dac63cb44a9fa070dfc9d567c17c62d8819ebca4
-  data.tar.gz: a14b51ff040d8a55cf1cbc3540852775119083424926368a906d502db0fc23dc1d3786f45d1a33543a3453daa1be1cf5c6dba254e17860a76a5fb71129749b7f
+  metadata.gz: c905b4740b81faf9f7068c5da07f4b1f5805cee13fa1667ce2f05d6bbf190481e71bdfd0819af2402202d79b77da6f37b371c5b4d2541be1d4c6b994b6bd00f4
+  data.tar.gz: 14d82e1931e8aaee4643d5a2caa6ec443d269d7bee6c5b130a0b34f68ef254ec131fa6482c3db7d2ed1a5d521ad5cfea5792a4f6fe8967a4f17d8b3f2ac96419

data/README.md

@@ -6,6 +6,8 @@ A Class of Active record pattern for TSV/CSV
 
 ## Usage
 
+data/users.tsv
+
 ```tsv
 id	name	age
 1	ksss	30
@@ -14,6 +16,7 @@ id	name	age
 ```
 
 ```ruby
+require 'active_tsv'
 class User < ActiveTsv::Base
   self.table_path = "data/users.tsv"
 end

data/Rakefile

@@ -1,6 +1,6 @@
 require "bundler/gem_tasks"
 
 task :test do
-  sh "bundle ex rgot -v"
+  sh "bundle ex rgot -v --bench ."
 end
 task default: :test

data/lib/active_tsv/base.rb

@@ -7,6 +7,8 @@ module ActiveTsv
   #   end
   class Base
     SEPARATER = "\t"
+    BUF_SIZE = 1024
+
     class << self
       attr_reader :table_path
 
@@ -15,7 +17,6 @@ module ActiveTsv
       end
 
       def reload(path)
-        old_table_path = table_path
         if @keys
           keys.each do |k|
             remove_method(k)
@@ -29,9 +30,6 @@ module ActiveTsv
           define_method(k) { @attrs[k] }
           define_method("#{k}=") { |v| @attrs[k] = v }
         end
-      rescue
-        reload(old_table_path)
-        raise
       end
 
       def all
@@ -44,8 +42,22 @@ module ActiveTsv
       end
 
       def last
-        last_value = open { |csv| csv.to_a.last }
-        new(keys.zip(last_value).to_h)
+        last_value = File.open(table_path) do |f|
+          f.seek(0, IO::SEEK_END)
+          size = f.size
+          buf_size = [size, self::BUF_SIZE].min
+          while true
+            f.seek(-buf_size, IO::SEEK_CUR)
+            buf = f.read(buf_size)
+            if index = buf.rindex($INPUT_RECORD_SEPARATOR, -2)
+              f.seek(-buf_size + index + 1, IO::SEEK_CUR)
+              break f.read.chomp
+            else
+              f.seek(-buf_size, IO::SEEK_CUR)
+            end
+          end
+        end
+        new(keys.zip(CSV.new(last_value, col_sep: self::SEPARATER).shift).to_h)
       end
 
       def open(&block)
@@ -57,11 +69,15 @@ module ActiveTsv
       end
 
       def where(condition = nil)
-        if condition
-          Relation.new(self, [Condition.new(:==, condition)])
-        else
-          WhereChain.new(self, [])
-        end
+        all.where(condition)
+      end
+
+      def count
+        all.count
+      end
+
+      def order(*columns)
+        all.order(*columns)
       end
     end
 
@@ -86,9 +102,7 @@ module ActiveTsv
     end
 
     def to_h
-      h = {}
-      self.class.keys.map { |k| h[k] = __send__ k }
-      h
+      @attrs.dup
     end
   end
 end

data/lib/active_tsv/relation.rb

@@ -4,16 +4,21 @@ module ActiveTsv
   class Relation
     include Enumerable
 
-    def initialize(table, conditions)
-      @table = table
-      @conditions = conditions
+    attr_reader :model
+    attr_reader :where_values
+    def initialize(model, where_values, order_values = [])
+      @model = model
+      @where_values = where_values
+      @order_values = order_values
     end
 
-    def where(condition = nil)
-      if condition
-        self.class.new(@table, @conditions << Condition.new(:==, condition))
+    def where(where_value = nil)
+      if where_value
+        dup.tap do |r|
+          r.where_values << Condition.new(:==, where_value)
+        end
       else
-        WhereChain.new(@table, @conditions)
+        WhereChain.new(dup)
       end
     end
 
@@ -25,21 +30,36 @@ module ActiveTsv
       to_a.last
     end
 
-    def each
-      return to_enum(:each) unless block_given?
+    def order(*columns)
+      @order_values += columns
+      self
+    end
+
+    def each(*args, &block)
+      to_a.each(*args, &block)
+    end
 
-      keys = @table.keys
+    def to_a
+      ret = []
+      keys = @model.keys
       key_to_value_index = keys.each_with_index.map { |k, index| [k, index] }.to_h
-      @table.open do |csv|
+      @model.open do |csv|
         csv.gets
         csv.each do |value|
-          yield @table.new(@table.keys.zip(value).to_h) if @conditions.all? { |cond|
+          ret << @model.new(keys.zip(value).to_h) if @where_values.all? { |cond|
             cond.values.all? do |k, v|
               value[key_to_value_index[k]].__send__(cond.method_name, v.to_s)
             end
           }
         end
       end
+      if @order_values.empty?
+        ret
+      else
+        ret.sort_by do |i|
+          @order_values.map { |m| i[m] }.join('-')
+        end
+      end
     end
   end
 end

data/lib/active_tsv/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveTsv
-  VERSION = "0.0.3"
+  VERSION = "0.1.0"
 end

data/lib/active_tsv/where_chain.rb

@@ -1,12 +1,13 @@
 module ActiveTsv
   class WhereChain
-    def initialize(table, conditions)
-      @table = table
-      @conditions = conditions
+    def initialize(relation)
+      @relation = relation
     end
 
     def not(condition)
-      Relation.new(@table, @conditions << Condition.new(:!=, condition))
+      @relation.dup.tap do |r|
+        r.where_values << Condition.new(:!=, condition)
+      end
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: active_tsv
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - ksss
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-04-30 00:00:00.000000000 Z
+date: 2016-05-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -98,7 +98,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.5.1
+rubygems_version: 2.6.3
 signing_key: 
 specification_version: 4
 summary: A Class of Active record pattern for TSV/CSV