org.reflections.Store Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rapidpm-dynamic-cdi-modules-reflection Show documentation
The newest version!
/**
 * Copyright © 2013 Sven Ruppert ([email protected])
 *
 * 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 org.reflections;

import repacked.com.google.common.collect.Iterables;
import repacked.com.google.common.collect.Multimap;
import repacked.com.google.common.collect.Multimaps;

import java.util.*;

/**
 * stores metadata information in multimaps
 * use the different query methods (getXXX) to query the metadata
 * 
the query methods are string based, and does not cause the class loader to define the types
 * use {@link org.reflections.Reflections#getStore()} to access this store
 */
public class Store {

  private final Map> storeMap;
  private transient boolean                           concurrent;

  //used via reflection
  @SuppressWarnings("UnusedDeclaration")
  protected Store() {
    storeMap = new HashMap<>();
    concurrent = false;
  }

  public Store(Configuration configuration) {
    storeMap = new HashMap<>();
    concurrent = configuration.getExecutorService() != null;
  }


  public Set keySet() {
    return storeMap.keySet();
  }


  public Multimap getOrCreate(String index) {
    Multimap mmap = storeMap.get(index);
    if (mmap == null) {
      Multimap multimap =
          Multimaps.newSetMultimap(new HashMap>() , HashSet::new);
      mmap = concurrent ? Multimaps.synchronizedSetMultimap(multimap) : multimap;
      storeMap.put(index , mmap);
    }
    return mmap;
  }


  public Multimap get(String index) {
    Multimap mmap = storeMap.get(index);
    if (mmap == null) {
      throw new ReflectionsException("Scanner " + index + " was not configured");
    }
    return mmap;
  }


  public Iterable get(String index , String... keys) {
    return get(index , Arrays.asList(keys));
  }


  public Iterable get(String index , Iterable keys) {
    Multimap mmap   = get(index);
    IterableChain    result = new IterableChain<>();
    for (String key : keys) {
      result.addAll(mmap.get(key));
    }
    return result;
  }


  private Iterable getAllIncluding(String index , Iterable keys , IterableChain result) {
    result.addAll(keys);
    for (String key : keys) {
      Iterable values = get(index , key);
      if (values.iterator().hasNext()) {
        getAllIncluding(index , values , result);
      }
    }
    return result;
  }


  public Iterable getAll(String index , String key) {
    return getAllIncluding(index , get(index , key) , new IterableChain<>());
  }


  public Iterable getAll(String index , Iterable keys) {
    return getAllIncluding(index , get(index , keys) , new IterableChain<>());
  }

  private static class IterableChain implements Iterable {
    private final List> chain = new ArrayList<>();

    private void addAll(Iterable iterable) {
      chain.add(iterable);
    }

    public Iterator iterator() {
      return Iterables.concat(chain).iterator();
    }
  }
}