grn_mini 0.3.0 → 0.4.0

data/test/test_grn_mini_hash.rb

@@ -1,14 +1,17 @@
+# -*- coding: utf-8 -*-
 require 'minitest_helper'
 
 class TestGrnMiniHash < MiniTest::Unit::TestCase
   def test_initialize
-    Dir.mktmpdir do |dir|
-      hash = GrnMini::Hash.new(File.join(dir, "test.db"))
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
     end
   end
 
   def test_add
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash.add("aaa", text:"aaa", number:1)
       assert_equal 1, hash.size
 
@@ -19,12 +22,14 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_select
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
       hash["c"] = {text:"ccc", number:3}
 
-      results = hash.select("bb")
+      results = hash.select("text:@bb")
 
       assert_equal 1, results.size
       assert_equal "b", results.first.key.key
@@ -33,13 +38,15 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_select2
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
       hash["c"] = {text:"bbb", number:20}
       hash["d"] = {text:"ccc", number:3}
 
-      results = hash.select("bb number:<10")
+      results = hash.select("text:@bb number:<10")
 
       assert_equal 1, results.size
       assert_equal "b", results.first.key.key
@@ -49,19 +56,21 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_size
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       assert_equal 0, hash.size
-      assert_equal 0, hash.length
       
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
       assert_equal 2, hash.size
-      assert_equal 2, hash.length
     end
   end
 
   def test_empty?
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       assert_equal true, hash.empty?
 
       hash["a"] = {text:"aaa", number:1}
@@ -70,7 +79,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_each
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["c"] = {text:"ccc", number:3}
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
@@ -86,7 +97,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_read_by_key
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
       hash["c"] = {text:"ccc", number:3}
@@ -99,7 +112,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_write_by_key
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
       hash["c"] = {text:"ccc", number:3}
@@ -116,7 +131,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_delete_by_id
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
       hash["c"] = {text:"ccc", number:3}
@@ -133,7 +150,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_delete_by_block
-    GrnMini::Hash.tmpdb do |hash|
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaa", number:1}
       hash["b"] = {text:"bbb", number:2}
       hash["c"] = {text:"ccc", number:3}
@@ -149,7 +168,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_float_column
-    GrnMini::Hash.tmpdb do |hash| 
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaaa", float: 1.5}
       hash["b"] = {text:"bbbb", float: 2.5}
       hash["c"] = {text:"cccc", float: 3.5}
@@ -162,7 +183,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_time_column
-    GrnMini::Hash.tmpdb do |hash| 
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaaa", timestamp: Time.new(2013)} # 2013-01-01
       hash["b"] = {text:"bbbb", timestamp: Time.new(2014)} # 2014-01-01
       hash["c"] = {text:"cccc", timestamp: Time.new(2015)} # 2015-01-01
@@ -176,7 +199,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_record_attributes
-    GrnMini::Hash.tmpdb do |hash| 
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaaa", int: 1}
       hash["b"] = {text:"bbbb", int: 2}
       hash["c"] = {text:"cccc", int: 3}
@@ -188,7 +213,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_assign_long_text_to_short_text
-    GrnMini::Hash.tmpdb do |hash| 
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {filename:"a.txt"}
       hash["b"] = {filename:"a"*4095 + ".txt" } # Over 4095 byte (ShortText limit)
 
@@ -198,7 +225,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_grn_object
-    GrnMini::Hash.tmpdb do |hash| 
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text: "aaaa", filename:"a.txt", int: 1, float: 1.5, time: Time.at(2001)}
 
       raw = hash.grn
@@ -221,7 +250,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_sort
-    GrnMini::Hash.tmpdb do |hash| 
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {name:"Tanaka",  age: 11, height: 162.5}
       hash["b"] = {name:"Suzuki",  age: 31, height: 170.0}
       hash["c"] = {name:"Hayashi", age: 21, height: 175.4}
@@ -245,7 +276,9 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
   end
 
   def test_group_from_hash
-    GrnMini::Hash.tmpdb do |hash| 
+    GrnMini::tmpdb do
+      hash = GrnMini::Hash.new
+
       hash["a"] = {text:"aaaa.txt", suffix:"txt", type:1}
       hash["b"] = {text:"aaaa.doc", suffix:"doc", type:2}
       hash["c"] = {text:"aabb.txt", suffix:"txt", type:2}
@@ -257,4 +290,90 @@ class TestGrnMiniHash < MiniTest::Unit::TestCase
       assert_equal ["doc", 1], [groups[1].key, groups[1].n_sub_records]
     end
   end
+
+  def test_select_table_element
+    GrnMini::tmpdb do
+      links = GrnMini::Hash.new("Links")
+      links.setup_columns(next: links)
+
+      links["aaa"] = {}
+      links["bbb"] = {}
+      links["ccc"] = {}
+
+      links["aaa"].next = links["bbb"]
+      links["bbb"].next = links["ccc"]
+      links["ccc"].next = links["aaa"]
+
+      assert_equal links["bbb"], links["aaa"].next
+
+      assert_equal links["ccc"], links.select("next: aaa").first.key
+    end
+  end
+
+  def test_nodes_table
+    GrnMini::tmpdb do
+      nodes = GrnMini::Hash.new("Nodes")
+      nodes.setup_columns(childs: [nodes], numbers: [0])
+
+      nodes["aaa"] = {}
+      nodes["bbb"] = {}
+      nodes["ccc"] = {}
+      nodes["ddd"] = {}
+      nodes["eee"] = {}
+
+      nodes["aaa"].childs = ["bbb", "ccc"]
+      # nodes["bbb"].childs = nodes["bbb"].childs + [nodes["ddd"]]
+      nodes["bbb"].childs += [nodes["ddd"]]
+      nodes["bbb"].childs += [nodes["eee"]] 
+
+      assert_equal 2, nodes["aaa"].childs.size
+      assert_equal 2, nodes["bbb"].childs.size
+
+      assert_equal nodes["aaa"], nodes.select("childs: bbb").first.key
+      assert_equal 0, nodes.select("childs: aaa").size
+    end
+  end
+  
+  def test_users_and_articles
+    GrnMini::tmpdb do
+      users = GrnMini::Hash.new("Users")
+      articles = GrnMini::Hash.new("Articles")
+
+      users.setup_columns(name: "", favorites: [articles])
+      articles.setup_columns(author: users, text: "")
+
+      users["aaa"] = {name: "Mr.A"}
+      users["bbb"] = {name: "Mr.B"}
+      users["ccc"] = {name: "Mr.C"}
+
+      articles["aaa:1"] = {author: "aaa", text: "111"}
+      articles["aaa:2"] = {author: "aaa", text: "222"}
+      articles["aaa:3"] = {author: "aaa", text: "333"}
+      articles["bbb:1"] = {author: "bbb", text: "111"}
+      articles["bbb:2"] = {author: "bbb", text: "222"}
+      articles["ccc:1"] = {author: "ccc", text: "111"}
+
+      users["aaa"].favorites = ["aaa:1", "bbb:2"]
+      users["bbb"].favorites = ["aaa:2"]
+      users["ccc"].favorites = ["aaa:1", "bbb:1", "ccc:1"]
+
+      assert_equal ["aaa", "ccc"], select_keys(users) { |record| record.favorites == "aaa:1" }
+      assert_equal ["bbb"], select_keys(users) { |record| record.favorites == "aaa:2" }
+      assert_equal [], select_keys(users) { |record| record.favorites == "aaa:3" }
+
+      assert_equal ["aaa", "ccc"], select_keys(users) { |record| record.favorites.text =~ "111" }
+      assert_equal ["aaa", "bbb"], select_keys(users) { |record| record.favorites.text =~ "222" }
+      assert_equal [], select_keys(users) { |record| record.favorites.text =~ "333" }
+
+      # assert_equal ["aaa", "ccc"], select_keys(users) { |record| record.favorites.text == "111" }      # TODO ==だと動かない
+      # assert_equal ["aaa", "ccc"], select_keys(users) { |record| record.favorites.author == "bbb" } # TODO 正しく動かない
+    end
+  end
+
+  def select_keys(table)
+    table.select { |record|
+      yield record
+    }.map { |record| record._key }.sort
+  end
+
 end

data/test/test_micro_blog.rb

@@ -0,0 +1,67 @@
+require 'minitest_helper'
+
+class TestMicroBlog < MiniTest::Unit::TestCase
+  def test_micro_blog
+    GrnMini::tmpdb do
+      engine = MicroBlog.new
+
+      # Add user
+      engine.add_user("aaa", "Mr. A", "Hello.")
+      engine.add_user("bbb", "Mr. B", "Good Afternoon.")
+      engine.add_user("ccc", "Mr. C", "Hi.")
+
+      # Follow
+      engine.follow("bbb", "aaa")
+      engine.follow("ccc", "aaa")
+      engine.follow("ccc", "bbb")
+
+      users = engine.users
+      assert_equal 0, users["aaa"].follower.size
+      assert_equal 1, users["bbb"].follower.size
+      assert_equal 2, users["ccc"].follower.size
+
+      # Followee
+      assert_equal 2, engine.followee("aaa").size
+      assert_equal 1, engine.followee("bbb").size
+      assert_equal 0, engine.followee("ccc").size
+    end
+  end
+
+  class MicroBlog
+    attr_reader :users
+    attr_reader :comments
+    attr_reader :hash_tags
+    
+    def initialize
+      @users = GrnMini::Hash.new("Users")
+      @comments = GrnMini::Hash.new("Comments")
+      @hash_tags = GrnMini::Hash.new("HashTags")
+
+      @users.setup_columns(name: "User Name",
+                           follower: [users],
+                           favorites: [comments],
+                           description: ""
+                           )
+
+      @comments.setup_columns(comment: "",
+                              last_modified: Time.new,
+                              replied_to: [comments],
+                              replied_users: [users],
+                              hash_tags: [hash_tags],
+                              posted_by: users
+                              )
+    end
+
+    def add_user(id, name, description)
+      @users[id] = {name: name, description: description }
+    end
+
+    def follow(user_id, follow_id)
+      @users[user_id].follower += [@users[follow_id]]
+    end
+
+    def followee(user_id)
+      @users.select("follower:#{user_id}")
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: grn_mini
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - ongaeshi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-09 00:00:00.000000000 Z
+date: 2014-01-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -79,6 +79,7 @@ files:
 - .gitignore
 - .travis.yml
 - Gemfile
+- HISTORY.md
 - LICENSE.txt
 - README.md
 - Rakefile
@@ -86,13 +87,14 @@ files:
 - lib/grn_mini.rb
 - lib/grn_mini/array.rb
 - lib/grn_mini/hash.rb
+- lib/grn_mini/table.rb
 - lib/grn_mini/util.rb
 - lib/grn_mini/version.rb
-- sample/mini-directory-search.rb
 - test/minitest_helper.rb
 - test/test_grn_mini.rb
 - test/test_grn_mini_array.rb
 - test/test_grn_mini_hash.rb
+- test/test_micro_blog.rb
 homepage: ''
 licenses:
 - MIT
@@ -122,3 +124,4 @@ test_files:
 - test/test_grn_mini.rb
 - test/test_grn_mini_array.rb
 - test/test_grn_mini_hash.rb
+- test/test_micro_blog.rb

data/sample/mini-directory-search.rb

@@ -1,153 +0,0 @@
-require 'grn_mini'
-require 'find'
-require 'kconv'
-require 'sinatra'
-require "sinatra/reloader" if ENV['SINATRA_RELOADER']
-
-module Input
-  module_function
-  
-  def from_dir(array, dir = ".")
-    index = 1
-    puts "Create database .."
-    Find.find(File.expand_path(dir)) do |filename|
-      Find.prune if ignore_dir?(filename)
-
-      if File.file? filename
-        next if ignore_file?(filename)
-        array << {filename: filename, text: read_file(filename), timestamp: File.stat(filename).mtime, suffix: File.extname(filename).sub('.', "")}
-        index += 1
-      end
-    end
-    puts "Input complete : #{array.size} files"
-  end
-
-  def read_file(filename)
-    text = File.read(filename)
-    Kconv.kconv(text, Kconv::UTF8).gsub("\r\n", "\n")
-  end
-
-  def ignore_dir?(filename)
-    basename = File.basename(filename)
-    /(\A\.svn\Z)|(\A\.git\Z)|(\ACVS\Z)/.match(basename)
-  end
-
-  def ignore_file?(filename)
-    (s = File.read(filename, 1024)) && s.index("\x00")
-  end
-end
-
-class Search
-  def initialize(array, params)
-    @array   = array
-    @params  = params
-    @page    = @params[:page] ? @params[:page].to_i : 1
-  end
-
-  def has_query?
-    @params[:query] && !@params[:query].empty?
-  end
-
-  def parse
-    unless has_query?
-      @header     = "<span>#{@array.size} files.</span>"
-      @content    = ""
-      @pagination = ""
-    else
-      results = @array.select(@params[:query])
-
-      page_entries = results.paginate([["_score", :desc]], :page => @page, :size => 20)
-      snippet = GrnMini::Util::html_snippet_from_selection_results(results, "<strong style=\"background-color: #FFEE55\">", "</strong>")
-      elements = []
-
-      page_entries.each do |record|
-        element = "<hr>\n<a href=\"/#{record.value.key.id}\">#{record.filename}</a>\n"
-        
-        snippet.execute(record.text).each do |segment|
-          element += "<pre style=\"border:1px solid #bbb;\">#{segment}</pre>\n"
-        end
-
-        elements << element
-      end
-
-      @header = "<span>#{page_entries.n_records} hit. (#{page_entries.start_offset} - #{page_entries.end_offset})</span>"
-      @content = elements.join("\n")
-
-      if page_entries.n_pages > 1
-        @pagination += page_link(@page - 1, "&lt;-") + "&nbsp;" if @page > 1
-
-        @pagination += page_range(page_entries).map {|v|
-          if (v == @page)
-            "<strong>#{v.to_s}</strong>"
-          else
-            page_link(v, v.to_s)
-          end
-        }.join("&nbsp;")
-
-        @pagination += "&nbsp;" + page_link(@page + 1, "-&gt;") if @page < page_entries.n_pages
-      end
-    end
-  end
-
-  def page_range(page_entries)
-    first_diff = [5 - (@page - 1), 0].max
-    last_diff  = [5 - (page_entries.n_pages - @page), 0].max
-    [@page - 5 - last_diff, 1].max .. [@page + 5 + first_diff, page_entries.n_pages].min
-  end
-
-  def page_link(page, msg)
-    "<a href=\"/?query=#{@params[:query]}&page=#{page}\">#{msg}</a>"
-  end
-
-  def html
-    <<EOF
-#{@header}
-<div class="form">
-  <form method="post" action="/search">
-    <input type="text" style="width: 419px;" name="query" value="#{@params[:query]}">
-    <input type="submit" value="Search">
-  </form>
-</div>
-<div class="content">
- #{@content}
-</div>
-<div class="pagination">
- #{@pagination}
-</div>
-EOF
-  end
-end
-
-### main ###
-configure do
-  $array = GrnMini::Array.new("mini-directory-search.db")
-  Input.from_dir($array) if $array.empty?
-end
-
-get '/' do
-  search = Search.new($array, params)
-  search.parse
-  search.html
-end
-
-get '/:id' do
-  record = $array[params[:id].to_i]
-  
-  <<EOF
-<span>#{record.filename} (#{record.timestamp})</span>
-<div class="form">
-  <form method="post" action="/search">
-    <input type="text" style="width: 419px;" name="query" value="#{@params[:query]}">
-    <input type="submit" value="Search">
-  </form>
-</div>
-<div class="content">
-  <hr>
-  <pre>#{CGI.escapeHTML(record.text)}</pre>
-</div>
-EOF
-end
-
-post '/search' do
-  redirect "/?query=#{escape(params[:query])}"
-end