mirror of https://github.com/apache/activemq.git
Parsing message selector for wildcard matches for ignoreWildcardSelectors feature
This commit is contained in:
parent
25ff5699f1
commit
278d29dadc
|
@ -27,6 +27,8 @@ import java.util.HashSet;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
import java.util.concurrent.ConcurrentMap;
|
import java.util.concurrent.ConcurrentMap;
|
||||||
|
import java.util.regex.Matcher;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
import javax.management.JMException;
|
import javax.management.JMException;
|
||||||
import javax.management.ObjectName;
|
import javax.management.ObjectName;
|
||||||
|
@ -165,9 +167,9 @@ public class SubQueueSelectorCacheBroker extends BrokerFilter implements Runnabl
|
||||||
return super.addConsumer(context, info);
|
return super.addConsumer(context, info);
|
||||||
}
|
}
|
||||||
|
|
||||||
// trivial check for SQL92/selector wildcards
|
|
||||||
private boolean hasWildcards(String selector) {
|
static boolean hasWildcards(String selector) {
|
||||||
return selector.contains("%") || selector.contains("_");
|
return new WildcardFinder(selector).hasWildcards();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -306,5 +308,54 @@ public class SubQueueSelectorCacheBroker extends BrokerFilter implements Runnabl
|
||||||
public void setIgnoreWildcardSelectors(boolean ignoreWildcardSelectors) {
|
public void setIgnoreWildcardSelectors(boolean ignoreWildcardSelectors) {
|
||||||
this.ignoreWildcardSelectors = ignoreWildcardSelectors;
|
this.ignoreWildcardSelectors = ignoreWildcardSelectors;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// find wildcards inside like operator arguments
|
||||||
|
static class WildcardFinder {
|
||||||
|
|
||||||
|
private static final Pattern LIKE_PATTERN=Pattern.compile(
|
||||||
|
"\\bLIKE\\s+'(?<like>([^']|'')+)'(\\s+ESCAPE\\s+'(?<escape>.)')?",
|
||||||
|
Pattern.CASE_INSENSITIVE);
|
||||||
|
|
||||||
|
private static final String REGEX_SPECIAL = ".+?*(){}[]\\-";
|
||||||
|
private final Matcher matcher;
|
||||||
|
|
||||||
|
WildcardFinder(String selector) {
|
||||||
|
this.matcher = LIKE_PATTERN.matcher(selector);
|
||||||
|
}
|
||||||
|
|
||||||
|
private String getLike() {
|
||||||
|
return matcher.group("like");
|
||||||
|
}
|
||||||
|
|
||||||
|
private String getEscape() {
|
||||||
|
String escapeChar = matcher.group("escape");
|
||||||
|
if (escapeChar == null) {
|
||||||
|
return null;
|
||||||
|
} else if (REGEX_SPECIAL.contains(escapeChar)) {
|
||||||
|
escapeChar = "\\"+escapeChar;
|
||||||
|
}
|
||||||
|
return escapeChar;
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean hasLikeOperator() {
|
||||||
|
return matcher.find();
|
||||||
|
}
|
||||||
|
|
||||||
|
boolean hasWildcardInCurrentMatch() {
|
||||||
|
String wildcards = "[_%]";
|
||||||
|
if (getEscape() != null) {
|
||||||
|
wildcards = "(^|[^" + getEscape() + "])" + wildcards;
|
||||||
|
}
|
||||||
|
return Pattern.compile(wildcards).matcher(getLike()).find();
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean hasWildcards() {
|
||||||
|
while(hasLikeOperator()) {
|
||||||
|
if (hasWildcardInCurrentMatch())
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,60 @@
|
||||||
|
/**
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You 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 org.apache.activemq.plugin;
|
||||||
|
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.junit.Assert.assertTrue;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Tests that presence of wildcard characters is correctly identified by SubQueueSelectorCacheBroker
|
||||||
|
*/
|
||||||
|
public class SubQueueSelectorCacheBrokerWildcardTest {
|
||||||
|
@Test
|
||||||
|
public void testSimpleWildcardEvaluation() {
|
||||||
|
assertWildcard(true, "modelInstanceId = '170' AND modelClassId LIKE 'com.whatever.something.%'");
|
||||||
|
assertWildcard(true, "JMSMessageId LIKE '%'");
|
||||||
|
assertWildcard(false, "modelClassId = 'com.whatever.something.%'");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testEscapedWildcardEvaluation() {
|
||||||
|
assertWildcard(true, "foo LIKE '!_%' ESCAPE '!'");
|
||||||
|
assertWildcard(false, "_foo__ LIKE '!_!%' ESCAPE '!'");
|
||||||
|
assertWildcard(true, "_foo_ LIKE '_%' ESCAPE '.'");
|
||||||
|
assertWildcard(true, "JMSMessageId LIKE '%' ESCAPE '.'");
|
||||||
|
assertWildcard(false, "_foo_ LIKE '\\_\\%' ESCAPE '\\'");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNonWildard() {
|
||||||
|
assertWildcard(false, "type = 'UPDATE_ENTITY'");
|
||||||
|
assertWildcard(false, "a_property = 1");
|
||||||
|
assertWildcard(false, "percentage = '100%'");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testApostrophes() {
|
||||||
|
assertWildcard(true, "quote LIKE '''In G_d We Trust'''");
|
||||||
|
assertWildcard(true, "quote LIKE '''In Gd We Trust''' OR quote not like '''In G_d We Trust'''");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void assertWildcard(boolean expected, String selector) {
|
||||||
|
assertEquals("Wildcard should "+(!expected ? " NOT ":"")+" be found in "+selector, expected, SubQueueSelectorCacheBroker.hasWildcards(selector));
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue