| /* |
| * Copyright 2000-2014 JetBrains s.r.o. |
| * |
| * 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 com.intellij.util; |
| |
| import com.intellij.util.containers.ContainerUtil; |
| import gnu.trove.TObjectHashingStrategy; |
| import org.jetbrains.annotations.NotNull; |
| |
| import java.util.List; |
| |
| /** |
| * @author max |
| */ |
| public class QueryFactory<Result, Parameters> { |
| private final List<QueryExecutor<Result, Parameters>> myExecutors = ContainerUtil.createLockFreeCopyOnWriteList(); |
| |
| public void registerExecutor(@NotNull QueryExecutor<Result, Parameters> executor) { |
| myExecutors.add(executor); |
| } |
| |
| public void unregisterExecutor(@NotNull QueryExecutor<Result, Parameters> executor) { |
| myExecutors.remove(executor); |
| } |
| |
| /** |
| * @return query to perform the search. @param parameters of the search |
| */ |
| @NotNull |
| public final Query<Result> createQuery(@NotNull Parameters parameters) { |
| return new ExecutorsQuery<Result, Parameters>(parameters, getExecutors()); |
| } |
| |
| |
| @NotNull |
| protected List<QueryExecutor<Result, Parameters>> getExecutors() { |
| return myExecutors; |
| } |
| |
| public boolean hasAnyExecutors() { |
| return !getExecutors().isEmpty(); |
| } |
| |
| /** |
| * @param parameters of the search |
| * @return query to perform the search. Obtained results are automatically filtered wrt. equals() relation. |
| */ |
| @NotNull |
| public final Query<Result> createUniqueResultsQuery(@NotNull Parameters parameters) { |
| return new UniqueResultsQuery<Result, Result>(createQuery(parameters)); |
| } |
| |
| /** |
| * @param parameters of the search |
| * @param hashingStrategy strategy to factor results |
| * @return query to perform the search. Obtained results are automatically filtered wrt. equals() relation. |
| */ |
| @NotNull |
| public final Query<Result> createUniqueResultsQuery(@NotNull Parameters parameters, |
| @NotNull TObjectHashingStrategy<Result> hashingStrategy) { |
| return new UniqueResultsQuery<Result, Result>(createQuery(parameters), hashingStrategy); |
| } |
| |
| /** |
| * @param parameters of the search |
| * @param hashingStrategy strategy to factor results |
| * @param mapper function that maps results to their mapping counterparts. |
| * @return query to perform the search. Obtained results are mapped to whatever objects that are automatically filtered wrt. equals() |
| * relation. Storing mapped objects instead of original elements may be wise wrt to memory consumption. |
| */ |
| @NotNull |
| public final <T> Query<Result> createUniqueResultsQuery(@NotNull Parameters parameters, |
| @NotNull TObjectHashingStrategy<T> hashingStrategy, |
| @NotNull Function<Result, T> mapper) { |
| return new UniqueResultsQuery<Result, T>(createQuery(parameters), hashingStrategy, mapper); |
| } |
| } |