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
|
module Groonga
module Sharding
class RangeExpressionBuilder
attr_reader :match_columns_expression
attr_writer :match_columns
attr_writer :query
attr_writer :filter
def initialize(key, target_range)
@key = key
@target_range = target_range
@match_columns_expression = nil
@match_columns = nil
@query = nil
@filter = nil
end
def build(expression, shard_range)
cover_type = @target_range.cover_type(shard_range)
case cover_type
when :all
build_all(expression)
when :partial_min
build_partial_min(expression)
when :partial_max
build_partial_max(expression)
when :partial_min_and_max
build_partial_min_and_max(expression)
when :none
end
end
def build_all(expression)
build_condition(expression)
end
def build_partial_min(expression)
expression.append_object(@key, Operator::PUSH, 1)
expression.append_operator(Operator::GET_VALUE, 1)
expression.append_constant(@target_range.min, Operator::PUSH, 1)
if @target_range.min_border == :include
expression.append_operator(Operator::GREATER_EQUAL, 2)
else
expression.append_operator(Operator::GREATER, 2)
end
build_condition(expression)
end
def build_partial_max(expression)
expression.append_object(@key, Operator::PUSH, 1)
expression.append_operator(Operator::GET_VALUE, 1)
expression.append_constant(@target_range.max, Operator::PUSH, 1)
if @target_range.max_border == :include
expression.append_operator(Operator::LESS_EQUAL, 2)
else
expression.append_operator(Operator::LESS, 2)
end
build_condition(expression)
end
def build_partial_min_and_max(expression)
between = Groonga::Context.instance["between"]
expression.append_object(between, Operator::PUSH, 1)
expression.append_object(@key, Operator::PUSH, 1)
expression.append_operator(Operator::GET_VALUE, 1)
expression.append_constant(@target_range.min, Operator::PUSH, 1)
expression.append_constant(@target_range.min_border,
Operator::PUSH, 1)
expression.append_constant(@target_range.max, Operator::PUSH, 1)
expression.append_constant(@target_range.max_border,
Operator::PUSH, 1)
expression.append_operator(Operator::CALL, 5)
build_condition(expression)
end
private
def build_condition(expression)
if @query
is_empty = expression.empty?
if @match_columns
table = Context.instance[expression[0].domain_id]
@match_columns_expression = Expression.create(table)
@match_columns_expression.parse(@match_columns)
end
flags = Expression::SYNTAX_QUERY |
Expression::ALLOW_PRAGMA |
Expression::ALLOW_COLUMN
expression.parse(@query,
default_column: @match_columns_expression,
flags: flags)
expression.append_operator(Operator::AND, 2) unless is_empty
end
if @filter
is_empty = expression.empty?
expression.parse(@filter)
expression.append_operator(Operator::AND, 2) unless is_empty
end
end
end
end
end
|
