PerCflowXAspectFactoryCompiler.java

/**************************************************************************************
 * Copyright (c) Jonas Bon�r, Alexandre Vasseur. All rights reserved.                 *
 * http://aspectwerkz.codehaus.org                                                    *
 * ---------------------------------------------------------------------------------- *
 * The software in this package is published under the terms of the LGPL license      *
 * a copy of which has been included with this distribution in the license.txt file.  *
 **************************************************************************************/
package org.codehaus.aspectwerkz.aspect.container;

import org.codehaus.aspectwerkz.definition.AspectDefinition;
import org.codehaus.aspectwerkz.aspect.management.NoAspectBoundException;
import org.codehaus.aspectwerkz.aspect.AspectContainer;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Label;
import org.objectweb.asm.Type;

/**
 * FIXME AVF review
 * @author <a href="mailto:alex AT gnilux DOT com">Alexandre Vasseur</a>
 */
public class PerCflowXAspectFactoryCompiler extends LazyPerXFactoryCompiler {

    public PerCflowXAspectFactoryCompiler(String uuid, String aspectClassName, String aspectQualifiedName, String containerClassName, String rawParameters, ClassLoader loader) {
        super(uuid, aspectClassName, aspectQualifiedName, containerClassName, rawParameters, loader);
    }

    protected String getXSignature() {
        return "Ljava/lang/Thread;";
    }

    /**
     * Overrides the aspectOf() method to not do lazy aspect instantiation since controlled by bind()/unbind()
     */
    protected void createAspectOf() {
        m_cw.visitField(
                ACC_PRIVATE + ACC_STATIC,
                "ASPECTS",
                MAP_CLASS_SIGNATURE,
                null,
                null
        );

        m_clinit.visitTypeInsn(NEW, MAP_CLASS_NAME);
        m_clinit.visitInsn(DUP);
        m_clinit.visitMethodInsn(INVOKESPECIAL, "java/util/WeakHashMap", INIT_METHOD_NAME, "()V");
        m_clinit.visitFieldInsn(PUTSTATIC, m_aspectFactoryClassName, "ASPECTS", MAP_CLASS_SIGNATURE);

        MethodVisitor cv = m_cw.visitMethod(
                ACC_PUBLIC + ACC_STATIC + ACC_FINAL,
                "aspectOf",
                "(" + getXSignature() + ")" + m_aspectClassSignature,
                null,
                null
        );

        cv.visitFieldInsn(GETSTATIC, m_aspectFactoryClassName, "ASPECTS", MAP_CLASS_SIGNATURE);
        cv.visitVarInsn(ALOAD, 0);//Thread
        cv.visitMethodInsn(INVOKEINTERFACE, MAP_CLASS_NAME, "get", "(Ljava/lang/Object;)Ljava/lang/Object;");
        cv.visitVarInsn(ASTORE, 1);
        cv.visitVarInsn(ALOAD, 1);
        Label ifBound = new Label();
        cv.visitJumpInsn(IFNONNULL, ifBound);
        cv.visitTypeInsn(NEW, Type.getInternalName(NoAspectBoundException.class));
        cv.visitInsn(DUP);
        cv.visitLdcInsn("Not bound");
        cv.visitLdcInsn(m_aspectQualifiedName);
        cv.visitMethodInsn(
                INVOKESPECIAL,
                NO_ASPECT_BOUND_EXCEPTION_CLASS_NAME,
                INIT_METHOD_NAME,
                "(Ljava/lang/String;Ljava/lang/String;)V"
        );
        cv.visitInsn(ATHROW);

        cv.visitLabel(ifBound);
        cv.visitVarInsn(ALOAD, 1);
        cv.visitTypeInsn(CHECKCAST, m_aspectClassName);
        cv.visitInsn(ARETURN);

        // create an implicit Thread.currentThread no arg aspectOf() method
        cv = m_cw.visitMethod(
                ACC_STATIC + ACC_PUBLIC + ACC_FINAL,
                "aspectOf",
                "()" + m_aspectClassSignature,
                null,
                null
        );
        cv.visitMethodInsn(
                INVOKESTATIC, Type.getInternalName(Thread.class), "currentThread", "()Ljava/lang/Thread;"
        );
        cv.visitMethodInsn(
                INVOKESTATIC, m_aspectFactoryClassName, "aspectOf", "(Ljava/lang/Thread;)" + getXSignature()
        );
        cv.visitInsn(ARETURN);
        cv.visitMaxs(0, 0);
    }

    protected void createOtherArtifacts() {
        createBindMethod();
        createUnbindMethod();
    }

    private void createBindMethod() {
        //FIXME AVF do not bind if already present
        MethodVisitor cv = m_cw.visitMethod(
                ACC_PUBLIC + ACC_STATIC + ACC_FINAL,
                "bind",
                "(" + getXSignature() + ")V",
                null,
                null
        );

        cv.visitFieldInsn(GETSTATIC, m_aspectFactoryClassName, "ASPECTS", MAP_CLASS_SIGNATURE);
        if (m_hasAspectContainer) {
            cv.visitFieldInsn(
                    GETSTATIC, m_aspectFactoryClassName, "CONTAINER", Type.getDescriptor(AspectContainer.class)
            );
            cv.visitMethodInsn(
                    INVOKEINTERFACE,
                    Type.getInternalName(AspectContainer.class),
                    "aspectOf",
                    "(" + getXSignature() + ")Ljava/lang/Object;"
            );
            cv.visitTypeInsn(CHECKCAST, m_aspectClassName);
        } else {
            cv.visitTypeInsn(NEW, m_aspectClassName);
            cv.visitInsn(DUP);
            cv.visitMethodInsn(INVOKESPECIAL, m_aspectClassName, INIT_METHOD_NAME, "()V");
        }
        cv.visitVarInsn(ALOAD, 0);//Thread
        cv.visitMethodInsn(
                INVOKEINTERFACE,
                MAP_CLASS_NAME,
                "put",
                "(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;"
        );
        cv.visitInsn(POP);
        cv.visitInsn(RETURN);
        cv.visitMaxs(0, 0);
    }

    private void createUnbindMethod() {
        MethodVisitor cv = m_cw.visitMethod(
                ACC_PUBLIC + ACC_STATIC + ACC_FINAL,
                "unbind",
                "(" + getXSignature() + ")V",
                null,
                null
        );

        cv.visitFieldInsn(GETSTATIC, m_aspectFactoryClassName, "ASPECTS", MAP_CLASS_SIGNATURE);
        cv.visitVarInsn(ALOAD, 0);//Thread
        cv.visitMethodInsn(INVOKEINTERFACE, MAP_CLASS_NAME, "remove", "(Ljava/lang/Object;)Ljava/lang/Object;");
        cv.visitInsn(POP);
        cv.visitInsn(RETURN);
        cv.visitMaxs(0, 0);
    }
}