Package org.jboss.aop.advice

Source Code of org.jboss.aop.advice.ClassifiedBindingAndPointcutCollection

/*
* JBoss, Home of Professional Open Source
* Copyright 2005, JBoss Inc., and individual contributors as indicated
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.aop.advice;

import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedHashMap;

import org.jboss.aop.AspectManager;
import org.jboss.aop.pointcut.Pointcut;
import org.jboss.aop.pointcut.PointcutExpression;
import org.jboss.aop.pointcut.PointcutInfo;
import org.jboss.aop.pointcut.PointcutStats;
import org.jboss.aop.util.BindingClassifier;
import org.jboss.aop.util.UnmodifiableEmptyCollections;
import org.jboss.aop.util.UnmodifiableLinkedHashMap;
import org.jboss.logging.Logger;

/**
* Manages the binding, pointcut and pointcutInfo collections contained in a domain. All entries
* contained in this collection are indexed according to their classification.
* <p>
* <i>For internal use only.</i>
*
* @author  <a href="flavia.rainone@jboss.com">Flavia Rainone</a>
* @author  <a href="kabir.khan@jboss.com">Kabir Khan</a>
*/
public class ClassifiedBindingAndPointcutCollection
{
   private static final Logger logger = Logger.getLogger(AspectManager.class);
  
   //private ReentrantReadWriteLock lock = new ReentrantReadWriteLock();
 
   //Collections of bindings
   private volatile LinkedHashMap<String, AdviceBinding> bindings;
   private volatile Collection<AdviceBinding> fieldReadBindings;
   private volatile Collection<AdviceBinding> fieldWriteBindings;
   private volatile Collection<AdviceBinding> constructionBindings;
   private volatile Collection<AdviceBinding> constructorExecutionBindings;
   private volatile Collection<AdviceBinding> methodExecutionBindings;
   private volatile Collection<AdviceBinding> constructorCallBindings;
   private volatile Collection<AdviceBinding> methodCallBindings;
  
   //Collections of pointcuts
   private volatile LinkedHashMap<String, Pointcut> pointcuts;
   private volatile Collection<Pointcut> fieldReadPointcuts;
   private volatile Collection<Pointcut> fieldWritePointcuts;
   private volatile Collection<Pointcut> constructionPointcuts;
   private volatile Collection<Pointcut> constructorExecutionPointcuts;
   private volatile Collection<Pointcut> methodExecutionPointcuts;
   private volatile Collection<Pointcut> constructorCallPointcuts;
   private volatile Collection<Pointcut> methodCallPointcuts;
  
  
   //Collections of pointcutInfos
   private volatile LinkedHashMap<String, PointcutInfo> pointcutInfos;
   private volatile Collection<PointcutInfo> fieldReadPointcutInfos;
   private volatile Collection<PointcutInfo> fieldWritePointcutInfos;
   private volatile Collection<PointcutInfo> constructionPointcutInfos;
   private volatile Collection<PointcutInfo> constructorExecutionPointcutInfos;
   private volatile Collection<PointcutInfo> methodExecutionPointcutInfos;
   private volatile Collection<PointcutInfo> constructorCallPointcutInfos;
   private volatile Collection<PointcutInfo> methodCallPointcutInfos;
  
   //Pointcut stats
   private boolean execution = false;
   private boolean construction = false;
   private boolean call = false;
   private boolean within = false;
   private boolean get = false;
   private boolean set = false;
   private boolean withincode = false;

   /**
    * Constructor.<p>
    */
   @SuppressWarnings("all")
   public ClassifiedBindingAndPointcutCollection()
   {
      bindings = UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP;
      this.fieldReadBindings = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.fieldWriteBindings = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructionBindings = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructorExecutionBindings = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.methodExecutionBindings = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructorCallBindings = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.methodCallBindings = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
     
      pointcuts = UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP;
      this.fieldReadPointcuts = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.fieldWritePointcuts = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructionPointcuts = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructorExecutionPointcuts = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.methodExecutionPointcuts = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructorCallPointcuts = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.methodCallPointcuts = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
     
      pointcutInfos = UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP;
      this.fieldReadPointcutInfos = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.fieldWritePointcutInfos = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructionPointcutInfos = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructorExecutionPointcutInfos = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.methodExecutionPointcutInfos = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.constructorCallPointcutInfos = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
      this.methodCallPointcutInfos = UnmodifiableEmptyCollections.EMPTY_ARRAYLIST;
   }
  
   /**
    * Check if there are any pointcuts stored at this level
    */
   public boolean hasPointcuts()
   {
      return pointcuts.size() > 0;
   }
  
   /**
    * Returns only the bindings whose pointcuts may match successfully field read
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the bindings that may match field
    *         read pointcuts
    */
   public Collection<AdviceBinding> getFieldReadBindings()
   {
      return this.fieldReadBindings;
   }
  
   /**
    * Returns only the bindings whose pointcuts may match successfully field write
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the bindings that may match field
    *         write pointcuts
    */
   public Collection<AdviceBinding> getFieldWriteBindings()
   {
      return this.fieldWriteBindings;
   }
  
   /**
    * Returns only the bindings whose pointcuts may match successfully construction
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the bindings that may match
    *         construction pointcuts
    */
   public Collection<AdviceBinding> getConstructionBindings()
   {
      return this.constructionBindings;
   }
  
   /**
    * Returns only the bindings whose pointcuts may match successfully constructor
    * execution joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the bindings that may match
    *         constructor execution pointcuts
    */
   public Collection<AdviceBinding> getConstructorExecutionBindings()
   {
      return this.constructorExecutionBindings;
   }
  
   /**
    * Returns only the bindings whose pointcuts may match successfully method
    * execution joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the bindings that may match
    *         method execution pointcuts
    */
   public Collection<AdviceBinding> getMethodExecutionBindings()
   {
      return this.methodExecutionBindings;
   }
  
   /**
    * Returns only the bindings whose pointcuts may match successfully constructor
    * call joinpoints.
    * <p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the bindings that may match
    *         constructor call pointcuts
    */
   public Collection<AdviceBinding> getConstructorCallBindings()
   {
      return this.constructorCallBindings;
   }
  
   /**
    * Returns only the bindings whose pointcuts may match successfully method
    * call joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the bindings that may match
    *         method call pointcuts
    */
   public Collection<AdviceBinding> getMethodCallBindings()
   {
      return this.methodCallBindings;
   }

   /**
    * Returns only the pointcuts that may match successfully field read
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcuts that may match field
    *         read pointcuts
    */
   public Collection<Pointcut> getFieldReadPointcuts()
   {
      return this.fieldReadPointcuts;
   }
  
   /**
    * Returns only the pointcuts that may match successfully field write
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcuts that may match field
    *         write pointcuts
    */
   public Collection<Pointcut> getFieldWritePointcuts()
   {
      return this.fieldWritePointcuts;
   }
  
   /**
    * Returns only the pointcuts that may match successfully construction
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcuts that may match
    *         construction pointcuts
    */
   public Collection<Pointcut> getConstructionPointcuts()
   {
      return this.constructionPointcuts;
   }
  
   /**
    * Returns only the pointcuts that may match successfully constructor
    * execution joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcuts that may match
    *         constructor execution pointcuts
    */
   public Collection<Pointcut> getConstructorExecutionPointcuts()
   {
      return this.constructorExecutionPointcuts;
   }
  
   /**
    * Returns only the pointcuts that may match successfully method
    * execution joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcuts that may match
    *         method execution pointcuts
    */
   public Collection<Pointcut> getMethodExecutionPointcuts()
   {
      return this.methodExecutionPointcuts;
   }
  
   /**
    * Returns only the pointcuts that may match successfully constructor
    * call joinpoints.
    * <p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcuts that may match
    *         constructor call pointcuts
    */
   public Collection<Pointcut> getConstructorCallPointcuts()
   {
      return this.constructorCallPointcuts;
   }
  
   /**
    * Returns only the pointcuts that may match successfully method
    * call joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcuts that may match
    *         method call pointcuts
    */
   public Collection<Pointcut> getMethodCallPointcuts()
   {
      return this.methodCallPointcuts;
   }
  
   /**
    * Returns only the pointcut infos that may match successfully field read
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcut infos that may match field
    *         read pointcuts
    */
   public Collection<PointcutInfo> getFieldReadPointcutInfos()
   {
      return this.fieldReadPointcutInfos;
   }
  
   /**
    * Returns only the pointcut infos that may match successfully field write
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcut infos that may match field
    *         write pointcuts
    */
   public Collection<PointcutInfo> getFieldWritePointcutInfos()
   {
      return this.fieldWritePointcutInfos;
   }
  
   /**
    * Returns only the pointcut infos that may match successfully construction
    * joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcut infos that may match
    *         construction pointcuts
    */
   public Collection<PointcutInfo> getConstructionPointcutInfos()
   {
      return this.constructionPointcutInfos;
   }
  
   /**
    * Returns only the pointcut infos that may match successfully constructor
    * execution joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcut infos that may match
    *         constructor execution pointcuts
    */
   public Collection<PointcutInfo> getConstructorExecutionPointcutInfos()
   {
      return this.constructorExecutionPointcutInfos;
   }
  
   /**
    * Returns only the pointcut infos that may match successfully method
    * execution joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcut infos that may match
    *         method execution pointcuts
    */
   public Collection<PointcutInfo> getMethodExecutionPointcutInfos()
   {
      return this.methodExecutionPointcutInfos;
   }
  
   /**
    * Returns only the pointcut infos that may match successfully constructor
    * call joinpoints.
    * <p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcut infos that may match
    *         constructor call pointcuts
    */
   public Collection<PointcutInfo> getConstructorCallPointcutInfos()
   {
      return this.constructorCallPointcutInfos;
   }
  
   /**
    * Returns only the pointcut infos that may match successfully method
    * call joinpoints.<p>
    * <b>Attention:</b> this collection is not supposed to be edited.
    *
    * @return a collection containing exclusively the pointcut infos that may match
    *         method call pointcuts
    */
   public Collection<PointcutInfo> getMethodCallPointcutInfos()
   {
      return this.methodCallPointcutInfos;
   }

   /**
    * Indicate whether this collection is empty.
    */
   public boolean isEmpty()
   {
      return this.bindings.isEmpty();
   }
  
   /**
    * Returns the bindings map.
    * @return an unmodifiable map containing all the bindings
    */
   public LinkedHashMap<String, AdviceBinding> getBindings()
   {
      return new UnmodifiableLinkedHashMap<String, AdviceBinding>(bindings);
   }
  
   /**
    * Returns the bindings map. This method is only for internal use, hence the @Deprecated
    * @return a map containing all the bindings
    */
   @Deprecated
   public LinkedHashMap<String, AdviceBinding> getBindingsInternal()
   {
      return new UnmodifiableLinkedHashMap<String, AdviceBinding>(bindings);
   }
  
   /**
    * Returns the pointcuts map.
    * @return an modifiable map containing all the pointcuts
    */
   public LinkedHashMap<String, Pointcut> getPointcuts()
   {
      return pointcuts;
   }
  
   /**
    * Returns the pointcuts map. This method is only for internal use, hence the @Deprecated
    * @return a map containing all the pointcuts
    */
   @Deprecated
   public LinkedHashMap<String, Pointcut> getPointcutsInternal()
   {
      return pointcuts;
   }
  
   /**
    * Returns the pointcutInfos map.
    * @return an unmodifiable map containing all the pointcutInfos
    */
   public LinkedHashMap<String, PointcutInfo> getPointcutInfos()
   {
      return new UnmodifiableLinkedHashMap<String, PointcutInfo>(pointcutInfos);
   }
  
   /**
    * Returns the pointcutInfos map. This method is only for internal use, hence the @Deprecated
    * @return an unmodifiable map containing all the pointcutInfos
    */
   @Deprecated
   public LinkedHashMap<String, PointcutInfo> getPointcutInfosInternal()
   {
      return pointcutInfos;
   }
  
   /**
    * Adds a binding to this collection.
    */
   public void add(AdviceBinding binding, AspectManager manager)
   {
      if (bindings == UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP)
      {
         bindings = new LinkedHashMap<String, AdviceBinding>();
      }
      bindings.put(binding.getName(), binding);
     
      if (pointcuts == UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP)
      {
         pointcuts = new LinkedHashMap<String, Pointcut>();
      }
      pointcuts.put(binding.getPointcut().getName(), binding.getPointcut());
     
      // create the pointcutinfo outside of methods to avoid duplicate creation
      // of pointcut infos
      PointcutInfo pointcutInfo = new PointcutInfo(binding.getPointcut(), binding,
            AspectManager.hasTransformationStarted());
      if (pointcutInfos == UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP)
      {
         pointcutInfos = new LinkedHashMap<String, PointcutInfo>();
      }
      pointcutInfos.put(pointcutInfo.getPointcut().getName(), pointcutInfo);
     
      addGet(binding, pointcutInfo);
      addSet(binding, pointcutInfo);
      addConstruction(binding, pointcutInfo);
      addConstructorExecution(binding, pointcutInfo);
      addMethodExecution(binding, pointcutInfo);
      addConstructorCall(binding, pointcutInfo);
      addMethodCall(binding, pointcutInfo);
      updatePointcutStats(binding.getPointcut(), manager);
   }

   /**
    * Adds a pointcut to this collection
    */
   public void add(Pointcut pointcut, AspectManager manager)
   {
      removePointcut(pointcut.getName());
      addPointcut(pointcut);

      updatePointcutStats(pointcut, manager);
   }
  
   /**
    * Removes the binding named {@code name}.
    *
    * @param name name of the binding to be removed.
    * @return the removed binding. If {@code null}, indicates that there is no
    *         binding with name equal to {@code name} in this collection.
    */
   public AdviceBinding removeBinding(String name)
   {
      AdviceBinding binding = bindings.remove(name);
      if (binding != null)
      {
         this.removePointcut(binding.getPointcut());
         this.fieldReadBindings.remove(binding);
         this.fieldWriteBindings.remove(binding);
         this.constructionBindings.remove(binding);
         this.constructorExecutionBindings.remove(binding);
         this.methodExecutionBindings.remove(binding);
         this.constructorCallBindings.remove(binding);
         this.methodCallBindings.remove(binding);
      }
      return binding;
   }
     
   /**
    * Removes the pointcut and pointcutInfo named {@code name}
    * @param name the name of the pointcut to be removed
    */
   public void removePointcut(String name)
   {
      Pointcut pc = pointcuts.get(name);
      if (pc != null)
      {
         this.removePointcut(pc);
      }
   }
  
   /**
    * Gets the pointcut named {@code name}
    * @param name the name of the pointcut to get
    * @return the pointcut
    */
   public Pointcut getPointcut(String name)
   {
      return pointcuts.get(name);
   }
  
   /**
    * Removes all bindings whose names are contained in {@code names}.
    *
    * @param names names of all bindings to be removed
    * @return the collection of the removed bindings
    */
   public ArrayList<AdviceBinding> removeBindings(ArrayList<String> names)
   {
      ArrayList<AdviceBinding> removedBindings = new ArrayList<AdviceBinding>();
      for (String name: names)
      {
         AdviceBinding binding = this.removeBinding(name);
         if (binding == null)
         {
            logger.debug("ClassifiedBindingCollection.removeBindings() no binding found with name " + name);
            continue;
         }
         removedBindings.add(binding);
      }
      return removedBindings;
   }

   public boolean isExecution()
   {
      return execution;
   }

   public boolean isConstruction()
   {
      return construction;
   }

   public boolean isCall()
   {
      return call;
   }

   public boolean isWithin()
   {
      return within;
   }

   public boolean isWithincode()
   {
      return withincode;
   }

   public boolean isGet()
   {
      return get;
   }

   public boolean isSet()
   {
      return set;
   }
  
   private void addGet(AdviceBinding binding, PointcutInfo pointcutInfo)
   {
      if (BindingClassifier.isGet(binding))
      {
         if (fieldReadBindings == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            fieldReadBindings = new ArrayList<AdviceBinding>();
         }
         this.fieldReadBindings.add(binding);
         this.addFieldReadPointcut(pointcutInfo.getPointcut(), pointcutInfo);
      }
   }
  
   private void addSet(AdviceBinding binding, PointcutInfo pointcutInfo)
   {
      if (BindingClassifier.isSet(binding))
      {
         if (fieldWriteBindings == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            fieldWriteBindings = new ArrayList<AdviceBinding>();
         }
         this.fieldWriteBindings.add(binding);
         this.addFieldWritePointcut(pointcutInfo.getPointcut(), pointcutInfo);
      }
   }
  
   private void addConstruction(AdviceBinding binding, PointcutInfo pointcutInfo)
   {
      if (BindingClassifier.isConstruction(binding))
      {
         if (constructionBindings == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructionBindings = new ArrayList<AdviceBinding>();
         }
         this.constructionBindings.add(binding);
         this.addConstructionPointcut(pointcutInfo.getPointcut(), pointcutInfo);
      }
   }
  
   private void addConstructorExecution(AdviceBinding binding, PointcutInfo pointcutInfo)
   {
      if (BindingClassifier.isConstructorExecution(binding))
      {
         if (constructorExecutionBindings == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructorExecutionBindings = new ArrayList<AdviceBinding>();
         }
         this.constructorExecutionBindings.add(binding);
         this.addConstructorExecutionPointcut(pointcutInfo.getPointcut(), pointcutInfo);
      }
   }
  
   private void addMethodExecution(AdviceBinding binding, PointcutInfo pointcutInfo)
   {
      if (BindingClassifier.isMethodExecution(binding))
      {
         if (methodExecutionBindings == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            methodExecutionBindings = new ArrayList<AdviceBinding>();
         }
         this.methodExecutionBindings.add(binding);
         this.addMethodExecutionPointcut(pointcutInfo.getPointcut(), pointcutInfo);
      }
   }
  
   private void addMethodCall(AdviceBinding binding, PointcutInfo pointcutInfo)
   {
      if (BindingClassifier.isMethodCall(binding))
      {
         if (methodCallBindings == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            methodCallBindings = new ArrayList<AdviceBinding>();
         }
         this.methodCallBindings.add(binding);
         this.addMethodCallPointcut(pointcutInfo.getPointcut(), pointcutInfo);
      }
   }
  
   private void addConstructorCall(AdviceBinding binding, PointcutInfo pointcutInfo)
   {
      if (BindingClassifier.isConstructorCall(binding))
      {
         if (constructorCallBindings == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructorCallBindings = new ArrayList<AdviceBinding>();
         }
         this.constructorCallBindings.add(binding);
         this.addConstructorCallPointcut(pointcutInfo.getPointcut(), pointcutInfo);
      }
   }
  
   private void addPointcut(Pointcut pointcut)
   {
      if (pointcuts == UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP)
      {
         pointcuts = new LinkedHashMap<String, Pointcut>();
      }
      pointcuts.put(pointcut.getName(), pointcut);

      if (pointcutInfos == UnmodifiableEmptyCollections.EMPTY_LINKED_HASHMAP)
      {
         pointcutInfos = new LinkedHashMap<String, PointcutInfo>();
      }
      PointcutInfo info = new PointcutInfo(pointcut, AspectManager.hasTransformationStarted());
      pointcutInfos.put(pointcut.getName(), info);

      addFieldReadPointcut(pointcut, info);
      addFieldWritePointcut(pointcut, info);
      addConstructionPointcut(pointcut, info);
      addConstructorExecutionPointcut(pointcut, info);
      addMethodExecutionPointcut(pointcut, info);
      addConstructorCallPointcut(pointcut, info);
      addMethodCallPointcut(pointcut, info);
   }

   private void addFieldReadPointcut(Pointcut pointcut, PointcutInfo info)
   {
      if (BindingClassifier.isGet(pointcut))
      {
         if (fieldReadPointcuts == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            fieldReadPointcuts = new ArrayList<Pointcut>();
         }
         fieldReadPointcuts.add(pointcut);
  
         if (fieldReadPointcutInfos == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            fieldReadPointcutInfos = new ArrayList<PointcutInfo>();
         }
         fieldReadPointcutInfos.add(info);
      }
   }
  
   private void addFieldWritePointcut(Pointcut pointcut, PointcutInfo info)
   {
      if (BindingClassifier.isSet(pointcut))
      {
         if (fieldWritePointcuts == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            fieldWritePointcuts = new ArrayList<Pointcut>();
         }
         fieldWritePointcuts.add(pointcut);
  
         if (fieldWritePointcutInfos == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            fieldWritePointcutInfos = new ArrayList<PointcutInfo>();
         }
         fieldWritePointcutInfos.add(info);
      }
   }
  
   private void addConstructionPointcut(Pointcut pointcut, PointcutInfo info)
   {
      if (BindingClassifier.isConstruction(pointcut))
      {
         if (constructionPointcuts == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructionPointcuts = new ArrayList<Pointcut>();
         }
         constructionPointcuts.add(pointcut);
  
         if (constructionPointcutInfos == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructionPointcutInfos = new ArrayList<PointcutInfo>();
         }
         constructionPointcutInfos.add(info);
      }
   }

   private void addConstructorExecutionPointcut(Pointcut pointcut, PointcutInfo info)
   {
      if (BindingClassifier.isConstructorExecution(pointcut))
      {
         if (constructorExecutionPointcuts == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructorExecutionPointcuts = new ArrayList<Pointcut>();
         }
         constructorExecutionPointcuts.add(pointcut);
  
         if (constructorExecutionPointcutInfos == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructorExecutionPointcutInfos = new ArrayList<PointcutInfo>();
         }
         constructorExecutionPointcutInfos.add(info);
      }
   }

   private void addMethodExecutionPointcut(Pointcut pointcut, PointcutInfo info)
   {
      if (BindingClassifier.isMethodExecution(pointcut))
      {
         if (methodExecutionPointcuts == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            methodExecutionPointcuts = new ArrayList<Pointcut>();
         }
         methodExecutionPointcuts.add(pointcut);
  
         if (methodExecutionPointcutInfos == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            methodExecutionPointcutInfos = new ArrayList<PointcutInfo>();
         }
         methodExecutionPointcutInfos.add(info);
      }
   }

   private void addConstructorCallPointcut(Pointcut pointcut, PointcutInfo info)
   {
      if (BindingClassifier.isConstructorCall(pointcut))
      {
         if (constructorCallPointcuts == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructorCallPointcuts = new ArrayList<Pointcut>();
         }
         constructorCallPointcuts.add(pointcut);
  
         if (constructorCallPointcutInfos == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            constructorCallPointcutInfos = new ArrayList<PointcutInfo>();
         }
         constructorCallPointcutInfos.add(info);
      }
   }

   private void addMethodCallPointcut(Pointcut pointcut, PointcutInfo info)
   {
      if (BindingClassifier.isMethodCall(pointcut))
      {
         if (methodCallPointcuts == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            methodCallPointcuts = new ArrayList<Pointcut>();
         }
         methodCallPointcuts.add(pointcut);
  
         if (methodCallPointcutInfos == UnmodifiableEmptyCollections.EMPTY_ARRAYLIST)
         {
            methodCallPointcutInfos = new ArrayList<PointcutInfo>();
         }
         methodCallPointcutInfos.add(info);
      }
   }
  
   /**
    * Removes the pointcut and the equivalent pointcutInfo.
    * @param pc the pointcut to be removed
    */
   public void removePointcut(Pointcut pc)
   {
      this.pointcuts.remove(pc.getName());
      this.fieldReadPointcuts.remove(pc);
      this.fieldWritePointcuts.remove(pc);
      this.constructionPointcuts.remove(pc);
      this.constructorExecutionPointcuts.remove(pc);
      this.methodExecutionPointcuts.remove(pc);
      this.constructorCallPointcuts.remove(pc);
      this.methodCallPointcuts.remove(pc);
     
      PointcutInfo info = pointcutInfos.remove(pc.getName());
      this.fieldReadPointcutInfos.remove(info);
      this.fieldWritePointcutInfos.remove(info);
      this.constructionPointcutInfos.remove(info);
      this.constructorExecutionPointcutInfos.remove(info);
      this.methodExecutionPointcutInfos.remove(info);
      this.constructorCallPointcutInfos.remove(info);
      this.methodCallPointcutInfos.remove(info);
   }

   private void updatePointcutStats(Pointcut pointcut, AspectManager manager)
   {
      // the following is for performance reasons.
      if (pointcut instanceof PointcutExpression)
      {
         PointcutExpression expr = (PointcutExpression) pointcut;
         expr.setManager(manager);
         PointcutStats stats = expr.getStats();
         updateStats(stats);
      }
      else
      {
         // can't be sure so set all
         execution = true;
         construction = true;
         call = true;
         within = true;
         get = true;
         set = true;
         withincode = true;
      }
   }

   public void updateStats(PointcutStats stats)
   {
      if (stats != null)
      {
         construction |= stats.isConstruction();
         execution |= stats.isExecution();
         call |= stats.isCall();
         within |= stats.isWithin();
         get |= stats.isGet();
         set |= stats.isSet();
         withincode |= stats.isWithincode();
      }
      else
      {
         if (AspectManager.verbose && logger.isDebugEnabled()) logger.debug("Setting all pointcut stats to true");
         // can't be sure so set all
         execution = true;
         construction = true;
         call = true;
         within = true;
         get = true;
         set = true;
         withincode = true;
      }
   }

}
TOP

Related Classes of org.jboss.aop.advice.ClassifiedBindingAndPointcutCollection

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.