org.snapscript.core.ModuleRegistry Maven / Gradle / Ivy

Go to download
package org.snapscript.core;

import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicInteger;

import org.snapscript.core.extend.ModuleExtender;

public class ModuleRegistry {

   private final Map modules;
   private final List references;
   private final PathConverter converter;
   private final ModuleExtender extender;
   private final AtomicInteger counter;
   private final Context context;
   private final int limit;
   
   public ModuleRegistry(Context context) {
      this(context, 100000);
   }
   
   public ModuleRegistry(Context context, int limit) {
      this.modules = new ConcurrentHashMap();
      this.references = new CopyOnWriteArrayList();
      this.extender = new ModuleExtender(context);
      this.converter = new FilePathConverter();
      this.counter = new AtomicInteger(1);
      this.context = context;
      this.limit = limit;
   }
   
   public synchronized List getModules() {
      return references;
   }

   public synchronized Module getModule(String name) {
      if (name == null) {
         throw new InternalArgumentException("Module name was null");
      }
      return modules.get(name);
   }

   public synchronized Module addModule(String name) {
      if (name == null) {
         throw new InternalArgumentException("Module name was null");
      }
      Path path = converter.createPath(name);
      Module current = modules.get(name);
      
      if(current == null) {
         return addModule(name, path);
      }
      return current;
   }
   
   public synchronized Module addModule(String name, Path path) {
      if (name == null) {
         throw new InternalArgumentException("Module name was null");
      }
      Module current = modules.get(name);

      if (current == null) {
         int order = counter.getAndIncrement();
         
         if(order > limit) {
            throw new InternalStateException("Module limit of " + limit + " exceeded");
         }
         Module module = new ContextModule(context, path, name, order);

         modules.put(name, module);
         extender.extend(module);
         references.add(module);
         
         return module;
      }
      return current;
   }
}