xref: /OpenGrok/opengrok-indexer/src/main/java/org/opengrok/indexer/search/context/TokenSetMatcher.java (revision c6f0939b1c668e9f8e1e276424439c3106b3a029)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * See LICENSE.txt included in this distribution for the specific
9  * language governing permissions and limitations under the License.
10  *
11  * When distributing Covered Code, include this CDDL HEADER in each
12  * file and include the License file at LICENSE.txt.
13  * If applicable, add the following below this CDDL HEADER, with the
14  * fields enclosed by brackets "[]" replaced with your own identifying
15  * information: Portions Copyright [yyyy] [name of copyright owner]
16  *
17  * CDDL HEADER END
18  */
19 
20 /*
21  * Copyright (c) 2005, 2021, Oracle and/or its affiliates. All rights reserved.
22  */
23 package org.opengrok.indexer.search.context;
24 
25 import java.util.Set;
26 import java.util.TreeSet;
27 
28 public class TokenSetMatcher extends LineMatcher {
29     private final Set<String> tokenSet;
30 
TokenSetMatcher(Set<String> tokenSet, boolean caseInsensitive)31     public TokenSetMatcher(Set<String> tokenSet, boolean caseInsensitive) {
32         super(caseInsensitive);
33 
34         // Use a TreeSet with an explicit comparator to allow for case
35         // insensitive lookups in the set if this is a case insensitive
36         // matcher.
37         this.tokenSet = new TreeSet<>(this::compareStrings);
38 
39         this.tokenSet.addAll(tokenSet);
40     }
41 
42     @Override
match(String token)43     public int match(String token) {
44         return tokenSet.contains(token) ? MATCHED : NOT_MATCHED;
45     }
46 }
47