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
|
#!/usr/bin/env ruby
#
# Copyright(C) 2019 Kouhei Sutou <kou@clear-code.com>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
require "json"
require "optparse"
n_records = 50_000_000
max_code_point = 0xFFFF
n_characters_per_record = 100
use_section = false
use_index_large = false
parser = OptionParser.new
parser.on("--n-records=N", Integer,
"[#{n_records}]") do |n|
if n < 0
n_records = Float::INFINITY
else
n_records = n
end
end
parser.on("--[no-]use-section",
"[#{use_section}]") do |boolean|
use_section = boolean
end
parser.on("--[no-]use-index-large",
"[#{use_index_large}]") do |boolean|
use_index_large = boolean
end
parser.parse!
if use_section
columns = 5.times.collect {|i| "value#{i}"}
else
columns = ["value"]
end
puts(<<-COMMANDS)
table_create Data TABLE_NO_KEY
COMMANDS
columns.each do |column|
puts(<<-COMMANDS)
column_create Data #{column} COLUMN_SCALAR|COMPRESS_ZSTD Text
COMMANDS
end
index_column_flags = "COLUMN_INDEX|WITH_POSITION"
if columns.size > 1
index_column_flags += "|WITH_SECTION"
end
if use_index_large
index_column_flags += "|INDEX_LARGE"
end
puts(<<-COMMANDS)
table_create Terms TABLE_HASH_KEY ShortText \
--normalizer NormalizerNFKC100 \
--default_tokenizer TokenRegexp
column_create Terms index #{index_column_flags} Data #{columns.join(",")}
COMMANDS
puts(<<-LOAD)
load --table Data
[
LOAD
(1..n_records).each do |id|
record = {"_id" => id}
columns.each do |column|
value = ""
n_characters_per_record.times do
while true
code_point = rand(max_code_point)
character = [code_point].pack("U")[0]
if character.valid_encoding?
value << character
break
end
end
end
record[column] = value
end
puts(record.to_json)
end
puts(<<-LOAD)
]
LOAD
|