addressed comments

This commit is contained in:
navis.ryu 2015-11-02 14:21:29 +09:00
parent 032c3e986d
commit 69c86716d6
9 changed files with 122 additions and 33 deletions

View File

@ -147,4 +147,13 @@ Search filters can be used to filter on partial string matches.
|property|description|required?|
|--------|-----------|---------|
|type|This String should always be "fragment".|yes|
|values|A JSON array of String values to run the search over. Case insensitive.|yes|
|values|A JSON array of String values to run the search over.|yes|
|caseSensitive|Whether strings should be compared as case sensitive or not. Default: false(insensitive)|no|
##### Contains
|property|description|required?|
|--------|-----------|---------|
|type|This String should always be "contains".|yes|
|value|A String value to run the search over.|yes|
|caseSensitive|Whether two string should be compared as case sensitive or not|yes|

View File

@ -19,11 +19,25 @@ If any part of a dimension value contains the value specified in this search que
FragmentSearchQuerySpec
-----------------------
If any part of a dimension value contains any of the values specified in this search query spec, regardless of case, a "match" occurs. The grammar is:
If any part of a dimension value contains all of the values specified in this search query spec, regardless of case by default, a "match" occurs. The grammar is:
```json
{
"type" : "fragment",
"case_sensitive" : false,
"values" : ["fragment1", "fragment2"]
}
```
ContainsSearchQuerySpec
----------------------------------
If any part of a dimension value contains the value specified in this search query spec, a "match" occurs. The grammar is:
```json
{
"type" : "contains",
"case_sensitive" : true,
"value" : "some_value"
}
```

View File

@ -75,6 +75,10 @@
<groupId>org.mapdb</groupId>
<artifactId>mapdb</artifactId>
</dependency>
<dependency>
<groupId>commons-lang</groupId>
<artifactId>commons-lang</artifactId>
</dependency>
<!-- Tests -->
<dependency>

View File

@ -18,6 +18,7 @@
package io.druid.query;
import com.google.common.base.Function;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import com.google.common.collect.Lists;
@ -683,25 +684,29 @@ public class Druids
public SearchQueryBuilder query(String q)
{
Preconditions.checkNotNull(q, "no value");
querySpec = new InsensitiveContainsSearchQuerySpec(q);
return this;
}
public SearchQueryBuilder query(Map<String, Object> q)
{
querySpec = new InsensitiveContainsSearchQuerySpec((String) q.get("value"));
String value = Preconditions.checkNotNull(q.get("value"), "no value").toString();
querySpec = new InsensitiveContainsSearchQuerySpec(value);
return this;
}
public SearchQueryBuilder query(String q, boolean caseSensitive)
{
Preconditions.checkNotNull(q, "no value");
querySpec = new ContainsSearchQuerySpec(q, caseSensitive);
return this;
}
public SearchQueryBuilder query(Map<String, Object> q, boolean caseSensitive)
{
querySpec = new ContainsSearchQuerySpec((String) q.get("value"), caseSensitive);
String value = Preconditions.checkNotNull(q.get("value"), "no value").toString();
querySpec = new ContainsSearchQuerySpec(value, caseSensitive);
return this;
}
@ -712,6 +717,7 @@ public class Druids
public SearchQueryBuilder fragments(List<String> q, boolean caseSensitive)
{
Preconditions.checkNotNull(q, "no value");
querySpec = new FragmentSearchQuerySpec(q, caseSensitive);
return this;
}

View File

@ -1,3 +1,20 @@
/*
* Druid - a distributed column store.
* Copyright 2012 - 2015 Metamarkets Group Inc.
*
* 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 io.druid.query.search.search;
import com.fasterxml.jackson.annotation.JsonCreator;
@ -16,8 +33,6 @@ public class ContainsSearchQuerySpec implements SearchQuerySpec
private final String value;
private final boolean caseSensitive;
private final String target;
@JsonCreator
public ContainsSearchQuerySpec(
@JsonProperty("value") String value,
@ -26,7 +41,6 @@ public class ContainsSearchQuerySpec implements SearchQuerySpec
{
this.value = value;
this.caseSensitive = caseSensitive;
this.target = value == null || caseSensitive ? value : value.toLowerCase();
}
@JsonProperty
@ -44,21 +58,29 @@ public class ContainsSearchQuerySpec implements SearchQuerySpec
@Override
public boolean accept(String dimVal)
{
if (dimVal == null) {
if (dimVal == null || value == null) {
return false;
}
final String input = caseSensitive ? dimVal : dimVal.toLowerCase();
return input.contains(target);
if (caseSensitive) {
return dimVal.contains(value);
}
return org.apache.commons.lang.StringUtils.containsIgnoreCase(dimVal, value);
}
@Override
public byte[] getCacheKey()
{
if (value == null) {
return ByteBuffer.allocate(2)
.put(CACHE_TYPE_ID)
.put(caseSensitive ? (byte) 1 : 0).array();
}
byte[] valueBytes = StringUtils.toUtf8(value);
return ByteBuffer.allocate(2 + valueBytes.length)
.put(caseSensitive ? (byte)1 : 0)
.put(CACHE_TYPE_ID)
.put(caseSensitive ? (byte) 1 : 0)
.put(valueBytes)
.array();
}
@ -71,7 +93,7 @@ public class ContainsSearchQuerySpec implements SearchQuerySpec
"}";
}
@Override
@Override
public boolean equals(Object o)
{
if (this == o) {

View File

@ -40,7 +40,8 @@ public class FragmentSearchQuerySpec implements SearchQuerySpec
@JsonCreator
public FragmentSearchQuerySpec(
@JsonProperty("values") List<String> values
) {
)
{
this(values, false);
}
@ -55,7 +56,7 @@ public class FragmentSearchQuerySpec implements SearchQuerySpec
Set<String> set = new HashSet();
if (values != null) {
for (String value : values) {
set.add(caseSensitive ? value : value.toLowerCase());
set.add(value);
}
}
target = set.toArray(new String[set.size()]);
@ -76,10 +77,22 @@ public class FragmentSearchQuerySpec implements SearchQuerySpec
@Override
public boolean accept(String dimVal)
{
if (dimVal == null) {
if (dimVal == null || values == null) {
return false;
}
final String input = caseSensitive ? dimVal : dimVal.toLowerCase();
if (caseSensitive) {
return containsAny(target, dimVal);
}
for (String search : target) {
if (!org.apache.commons.lang.StringUtils.containsIgnoreCase(dimVal, search)) {
return false;
}
}
return true;
}
private boolean containsAny(String[] target, String input)
{
for (String value : target) {
if (!input.contains(value)) {
return false;
@ -91,6 +104,12 @@ public class FragmentSearchQuerySpec implements SearchQuerySpec
@Override
public byte[] getCacheKey()
{
if (values == null) {
return ByteBuffer.allocate(2)
.put(CACHE_TYPE_ID)
.put(caseSensitive ? (byte) 1 : 0).array();
}
final byte[][] valuesBytes = new byte[values.size()][];
int valuesBytesSize = 0;
int index = 0;
@ -101,8 +120,8 @@ public class FragmentSearchQuerySpec implements SearchQuerySpec
}
final ByteBuffer queryCacheKey = ByteBuffer.allocate(2 + valuesBytesSize)
.put(caseSensitive ? (byte) 1 : 0)
.put(CACHE_TYPE_ID);
.put(CACHE_TYPE_ID)
.put(caseSensitive ? (byte) 1 : 0);
for (byte[] bytes : valuesBytes) {
queryCacheKey.put(bytes);

View File

@ -329,10 +329,19 @@ public class QueryRunnerTestHelper
QueryRunnerFactory<T, QueryType> factory,
Segment adapter
)
{
return makeQueryRunner(factory, segmentId, adapter);
}
public static <T, QueryType extends Query<T>> QueryRunner<T> makeQueryRunner(
QueryRunnerFactory<T, QueryType> factory,
String segmentId,
Segment adapter
)
{
return new FinalizeResultsQueryRunner<T>(
new BySegmentQueryRunner<T>(
adapter.getIdentifier(), adapter.getDataInterval().getStart(),
segmentId, adapter.getDataInterval().getStart(),
factory.createRunner(adapter)
),
(QueryToolChest<T, Query<T>>)factory.getToolchest()

View File

@ -68,7 +68,8 @@ public class SearchQueryRunnerWithCaseTest
"2011-01-12T00:00:00.000Z\tspot\tAutoMotive\tPREFERRED\ta\u0001preferred\t100.000000\n" +
"2011-01-12T00:00:00.000Z\tSPot\tbusiness\tpreferred\tb\u0001Preferred\t100.000000\n" +
"2011-01-12T00:00:00.000Z\tspot\tentertainment\tPREFERRed\te\u0001preferred\t100.000000\n" +
"2011-01-13T00:00:00.000Z\tspot\tautomotive\tpreferred\ta\u0001preferred\t94.874713");
"2011-01-13T00:00:00.000Z\tspot\tautomotive\tpreferred\ta\u0001preferred\t94.874713"
);
IncrementalIndex index1 = TestIndex.makeRealtimeIndex(input, true);
IncrementalIndex index2 = TestIndex.makeRealtimeIndex(input, false);
@ -76,12 +77,14 @@ public class SearchQueryRunnerWithCaseTest
QueryableIndex index3 = TestIndex.persistRealtimeAndLoadMMapped(index1);
QueryableIndex index4 = TestIndex.persistRealtimeAndLoadMMapped(index2);
return transformToConstructionFeeder(Arrays.asList(
makeQueryRunner(factory, new IncrementalIndexSegment(index1, "index1")),
makeQueryRunner(factory, new IncrementalIndexSegment(index2, "index2")),
makeQueryRunner(factory, new QueryableIndexSegment("index3", index3)),
makeQueryRunner(factory, new QueryableIndexSegment("index4", index4))
));
return transformToConstructionFeeder(
Arrays.asList(
makeQueryRunner(factory, "index1", new IncrementalIndexSegment(index1, "index1")),
makeQueryRunner(factory, "index2", new IncrementalIndexSegment(index2, "index2")),
makeQueryRunner(factory, "index3", new QueryableIndexSegment("index3", index3)),
makeQueryRunner(factory, "index4", new QueryableIndexSegment("index4", index4))
)
);
}
private final QueryRunner runner;
@ -93,11 +96,12 @@ public class SearchQueryRunnerWithCaseTest
this.runner = runner;
}
private Druids.SearchQueryBuilder testBuilder() {
private Druids.SearchQueryBuilder testBuilder()
{
return Druids.newSearchQueryBuilder()
.dataSource(dataSource)
.granularity(allGran)
.intervals(fullOnInterval);
.dataSource(dataSource)
.granularity(allGran)
.intervals(fullOnInterval);
}
@Test
@ -157,7 +161,7 @@ public class SearchQueryRunnerWithCaseTest
private void checkSearchQuery(SearchQuery searchQuery, Map<String, Set<String>> expectedResults)
{
HashMap<String,List> context = new HashMap<>();
HashMap<String, List> context = new HashMap<>();
Iterable<Result<SearchResultValue>> results = Sequences.toList(
runner.run(searchQuery, context),
Lists.<Result<SearchResultValue>>newArrayList()

View File

@ -162,7 +162,8 @@ public class TestIndex
}
}
private static IncrementalIndex makeRealtimeIndex(final String resourceFilename, final boolean useOffheap) {
private static IncrementalIndex makeRealtimeIndex(final String resourceFilename, final boolean useOffheap)
{
final URL resource = TestIndex.class.getClassLoader().getResource(resourceFilename);
if (resource == null) {
throw new IllegalArgumentException("cannot find resource " + resourceFilename);
@ -198,7 +199,8 @@ public class TestIndex
int lineCount;
try {
lineCount = source.readLines(
new LineProcessor<Integer>() {
new LineProcessor<Integer>()
{
StringInputRowParser parser = new StringInputRowParser(
new DelimitedParseSpec(
new TimestampSpec("ts", "iso", null),