AggregatingClassfileLoader.java

  1. /*
  2.  *  Copyright (c) 2001-2024, Jean Tessier
  3.  *  All rights reserved.
  4.  *  
  5.  *  Redistribution and use in source and binary forms, with or without
  6.  *  modification, are permitted provided that the following conditions
  7.  *  are met:
  8.  *  
  9.  *      * Redistributions of source code must retain the above copyright
  10.  *        notice, this list of conditions and the following disclaimer.
  11.  *  
  12.  *      * Redistributions in binary form must reproduce the above copyright
  13.  *        notice, this list of conditions and the following disclaimer in the
  14.  *        documentation and/or other materials provided with the distribution.
  15.  *  
  16.  *      * Neither the name of Jean Tessier nor the names of his contributors
  17.  *        may be used to endorse or promote products derived from this software
  18.  *        without specific prior written permission.
  19.  *  
  20.  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21.  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22.  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23.  *  A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR
  24.  *  CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  25.  *  EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  26.  *  PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  27.  *  PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  28.  *  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  29.  *  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  30.  *  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31.  */

  32. package com.jeantessier.classreader;

  33. import com.jeantessier.classreader.impl.DefaultClassfileFactory;

  34. import java.io.DataInput;
  35. import java.io.IOException;
  36. import java.util.Collection;
  37. import java.util.Collections;
  38. import java.util.Map;
  39. import java.util.TreeMap;

  40. public class AggregatingClassfileLoader extends ClassfileLoaderEventSource {
  41.     private final Map<String, Classfile> classfiles = new TreeMap<>();

  42.     public AggregatingClassfileLoader() {
  43.         this(new DefaultClassfileFactory());
  44.     }

  45.     private AggregatingClassfileLoader(ClassfileFactory factory) {
  46.         super(factory);
  47.     }

  48.     public AggregatingClassfileLoader(ClassfileLoaderDispatcher dispatcher) {
  49.         this(new DefaultClassfileFactory(), dispatcher);
  50.     }

  51.     private AggregatingClassfileLoader(ClassfileFactory factory, ClassfileLoaderDispatcher dispatcher) {
  52.         super(factory, dispatcher);
  53.     }

  54.     public Classfile getClassfile(String name) {
  55.         return classfiles.get(name);
  56.     }

  57.     public Collection<Classfile> getAllClassfiles() {
  58.         return Collections.unmodifiableCollection(classfiles.values());
  59.     }

  60.     public Collection<String> getAllClassNames() {
  61.         return Collections.unmodifiableCollection(classfiles.keySet());
  62.     }

  63.     protected Classfile load(DataInput in) throws IOException {
  64.         var result = getFactory().create(this, in);

  65.         classfiles.put(result.getClassName(), result);

  66.         return result;
  67.     }
  68. }