alfresco-community-repo/source/test-java/org/alfresco/traitextender/TestService.java

package org.alfresco.traitextender;

import java.util.List;

import org.alfresco.traitextender.AJExtender;
import org.alfresco.traitextender.Extend;
import org.alfresco.traitextender.ExtendedTrait;
import org.alfresco.traitextender.Extensible;
import org.alfresco.traitextender.Trait;

public class TestService implements Extensible
{

    private String psmv2;

    private final ExtendedTrait<TestTrait> testTrait;

    public TestService(String psmv2)
    {
        super();
        this.testTrait = new ExtendedTrait<TestTrait>(AJProxyTrait.create(createTestTrait(),
                                                                          TestTrait.class));
        this.psmv2 = psmv2;
    }

    @Extend(traitAPI = TestTrait.class, extensionAPI = TestExtension.class)
    private String privateServiceMethod1(String s)
    {
        return "TestService.privateServiceMethod1(" + s + ")";
    }

    public String publicServiceMethod1(String s)
    {
        return privateServiceMethod1(s);
    }

    @Extend(traitAPI = TestTrait.class, extensionAPI = TestExtension.class)
    public String publicServiceMethod3(String s)
    {
        return "PSM3" + privateServiceMethod1(s);
    }

    @Extend(traitAPI = TestTrait.class, extensionAPI = TestExtension.class)
    public String publicServiceMethod2(String s)
    {
        return this.psmv2;
    }

    @Extend(traitAPI = TestTrait.class, extensionAPI = TestExtension.class)
    public void publicServiceMethod3(TestService s, List<Integer> traitIdentities)
    {
        traitIdentities.add(System.identityHashCode(this));
    }

    @Override
    public <M extends Trait> ExtendedTrait<M> getTrait(Class<? extends M> traitAPI)
    {
        return (ExtendedTrait<M>) testTrait;
    }

    private TestTrait createTestTrait()
    {
        return new TestTrait()
        {

            @Override
            public String traitImplOf_privateServiceMethod1(final String s)
            {
                return AJExtender.run(new AJExtender.ExtensionBypass<String>()
                {

                    @Override
                    public String run()
                    {
                        return privateServiceMethod1(s);
                    };
                });
            }

            @Override
            public String traitImplOf_publicServiceMethod2(final String s)
            {
                return AJExtender.run(new AJExtender.ExtensionBypass<String>()
                {

                    @Override
                    public String run()
                    {
                        return publicServiceMethod2(s);
                    };
                });
            }

            @Override
            public String publicServiceMethod3(final String s)
            {
                return publicServiceMethod3(s);
            }

            @Override
            public void traitImplOf_publicServiceMethod3(final TestService s, final List<Integer> traitIdentities)
            {
                AJExtender.run(new AJExtender.ExtensionBypass<Void>()
                {

                    @Override
                    public Void run()
                    {
                        TestService.this.publicServiceMethod3(s,
                                                              traitIdentities);
                        return null;
                    };
                });
            }

        };
    }
}