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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
# frozen_string_literal: true
require 'erb'
require 'yaml'
require 'tempfile'
if RUBY_PLATFORM == 'java'
require 'java'
require 'jdbc/mysql'
Jdbc::MySQL.load_driver
end
if ENV["TRAVIS"] == "true"
FIXTURE_COMMAND = "LOAD DATA INFILE"
else
FIXTURE_COMMAND = "LOAD DATA LOCAL INFILE"
end
class Sphinx
attr_accessor :host, :username, :password
def initialize
self.host = 'localhost'
self.username = 'root'
self.password = ''
if File.exist?('spec/fixtures/sql/conf.yml')
config = YAML.load(File.open('spec/fixtures/sql/conf.yml'))
self.host = config['host']
self.username = config['username']
self.password = config['password']
end
end
def bin_path
ENV.fetch 'SPHINX_BIN', ''
end
def setup_mysql
databases = mysql_client.query "SHOW DATABASES"
unless databases.include?("riddle")
mysql_client.execute "CREATE DATABASE riddle"
end
mysql_client.execute 'USE riddle'
structure = File.open('spec/fixtures/sql/structure.sql') { |f| f.read }
structure.split(/;/).each { |sql| mysql_client.execute sql }
sql_file 'data.tsv' do |path|
mysql_client.execute <<-SQL
#{FIXTURE_COMMAND} '#{path}' INTO TABLE
`riddle`.`people` FIELDS TERMINATED BY ',' ENCLOSED BY "'" (gender,
first_name, middle_initial, last_name, street_address, city, state,
postcode, email, birthday)
SQL
end
mysql_client.close
end
def mysql_client
@mysql_client ||= if RUBY_PLATFORM == 'java'
JRubyClient.new host, username, password
else
MRIClient.new host, username, password
end
end
def generate_configuration
template = File.open('spec/fixtures/sphinx/configuration.erb') { |f| f.read }
File.open('spec/fixtures/sphinx/spec.conf', 'w') { |f|
f.puts ERB.new(template).result(binding)
}
FileUtils.mkdir_p "spec/fixtures/sphinx/binlog"
end
def index
cmd = "#{bin_path}indexer --config #{fixtures_path}/sphinx/spec.conf --all"
cmd << ' --rotate' if running?
`#{cmd}`
end
def start
return if running?
`#{bin_path}searchd --config #{fixtures_path}/sphinx/spec.conf`
sleep(1)
unless running?
puts 'Failed to start searchd daemon. Check fixtures/sphinx/searchd.log.'
end
end
def stop
return unless running?
stop_flag = '--stopwait'
stop_flag = '--stop' if Riddle.loaded_version.to_i < 1
`#{bin_path}searchd --config #{fixtures_path}/sphinx/spec.conf #{stop_flag}`
end
private
def fixtures_path
File.expand_path File.join(File.dirname(__FILE__), '..', 'fixtures')
end
def pid
if File.exists?("#{fixtures_path}/sphinx/searchd.pid")
`cat #{fixtures_path}/sphinx/searchd.pid`[/\d+/]
else
nil
end
end
def running?
pid && `ps #{pid} | wc -l`.to_i > 1
end
def sql_file(name, &block)
file = Tempfile.new(name)
file.write File.read("#{fixtures_path}/sql/#{name}")
`chmod +r #{file.path}`
file.flush
block.call file.path
file.close
file.unlink
end
end
|
