File: thread_safety_index_test.rb

package info (click to toggle)
ruby-ferret 0.11.8.4%2Bdebian-2
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 4,368 kB
  • sloc: ansic: 69,786; ruby: 8,131; makefile: 5
file content (88 lines) | stat: -rw-r--r-- 2,119 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#$:.unshift('.')
require 'monitor'
require File.dirname(__FILE__) + "/../test_helper"
require File.dirname(__FILE__) + "/number_to_spoken.rb"
require 'thread'

class IndexThreadSafetyTest < Test::Unit::TestCase
  include Ferret::Index

  INDEX_DIR = File.expand_path(File.join(File.dirname(__FILE__), "index"))
  ITERATIONS = 100
  NUM_THREADS = 3
  ANALYZER = Ferret::Analysis::StandardAnalyzer.new()

  def setup
    index = Index.new(:path => INDEX_DIR,
                      :create => true,
                      :analyzer => ANALYZER,
                      :default_field => :content)
    index.close
  end

  def indexing_thread()
    index = Index.new(:path => INDEX_DIR,
                      :analyzer => ANALYZER,
                      :auto_flush => true,
                      :default_field => :content)

    ITERATIONS.times do
      choice = rand()

      if choice > 0.98
        do_optimize(index)
      elsif choice > 0.7
        do_delete_doc(index)
      elsif choice > 0.5
        do_search(index)
      else
        do_add_doc(index)
      end
      index.commit
    end
  rescue Exception => e
    puts e
    puts e.backtrace
    raise 'hell'
  end 

  def do_optimize(index)
    puts "Optimizing the index"
    index.optimize
  end

  def do_delete_doc(index)
    return if index.size == 0
    doc_num = rand(index.size)
    puts "Deleting #{doc_num} from index which has#{index.has_deletions? ? "" : " no"} deletions"
    puts "document was already deleted" if (index.deleted?(doc_num))
    index.delete(doc_num)
  end

  def do_add_doc(index)
    n = rand(0xFFFFFFFF)
    d = {:id => n, :content => n.to_spoken}
    puts("Adding #{n}")
    index << d
  end
  
  def do_search(index)
    n = rand(0xFFFFFFFF)
    puts("Searching for #{n}")
    hits = index.search_each(n.to_spoken, :num_docs => 3) do |d, s|
      puts "Hit for #{n}: #{index[d][:id]} - #{s}"
    end
    puts("Searched for #{n}: total = #{hits}")
  end

  def test_threading
    threads = []
    NUM_THREADS.times do
      threads << Thread.new { indexing_thread }
    end

    threads.each {|t| 
      t.join
    }
  end
end