/
condition.go
79 lines (67 loc) · 2.18 KB
/
condition.go
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
/*
* Copyright 2019 ObjectBox Ltd. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package objectbox
// Condition is used by Query to limit object selection
type Condition interface {
applyTo(qb *QueryBuilder, isRoot bool) (ConditionId, error)
}
type ConditionId = int32
type conditionClosure struct {
applyFun func(qb *QueryBuilder) (ConditionId, error)
}
func (condition *conditionClosure) applyTo(qb *QueryBuilder, isRoot bool) (ConditionId, error) {
return condition.applyFun(qb)
}
// Combines multiple conditions with an operator
type conditionCombination struct {
or bool // AND by default
conditions []Condition
}
func (condition *conditionCombination) applyTo(qb *QueryBuilder, isRoot bool) (ConditionId, error) {
if len(condition.conditions) == 0 {
return 0, nil
} else if len(condition.conditions) == 1 {
return condition.conditions[0].applyTo(qb, isRoot)
}
ids := make([]ConditionId, len(condition.conditions))
var err error
for k, sub := range condition.conditions {
if ids[k], err = sub.applyTo(qb, false); err != nil {
return 0, err
}
}
if condition.or {
return qb.Any(ids)
} else if !isRoot {
// only necessary to use AND if it's not a root condition group
return qb.All(ids)
} else {
return 0, nil
}
}
// Any provides a way to combine multiple query conditions (equivalent to OR logical operator)
func Any(conditions ...Condition) Condition {
return &conditionCombination{
or: true,
conditions: conditions,
}
}
// All provides a way to combine multiple query conditions (equivalent to AND logical operator)
func All(conditions ...Condition) Condition {
return &conditionCombination{
conditions: conditions,
}
}