001// Copyright 2011 The Apache Software Foundation 002// 003// Licensed under the Apache License, Version 2.0 (the "License"); 004// you may not use this file except in compliance with the License. 005// You may obtain a copy of the License at 006// 007// http://www.apache.org/licenses/LICENSE-2.0 008// 009// Unless required by applicable law or agreed to in writing, software 010// distributed under the License is distributed on an "AS IS" BASIS, 011// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 012// See the License for the specific language governing permissions and 013// limitations under the License. 014 015package org.apache.tapestry5.internal.jpa; 016 017import org.apache.tapestry5.jpa.EntityManagerManager; 018import org.apache.tapestry5.jpa.annotations.CommitAfter; 019import org.apache.tapestry5.model.MutableComponentModel; 020import org.apache.tapestry5.plastic.MethodAdvice; 021import org.apache.tapestry5.plastic.PlasticClass; 022import org.apache.tapestry5.plastic.PlasticMethod; 023import org.apache.tapestry5.services.transform.ComponentClassTransformWorker2; 024import org.apache.tapestry5.services.transform.TransformationSupport; 025 026import javax.persistence.PersistenceContext; 027 028public class CommitAfterWorker implements ComponentClassTransformWorker2 029{ 030 031 private final MethodAdvice shared; 032 033 private final EntityManagerManager manager; 034 035 public CommitAfterWorker(EntityManagerManager manager) 036 { 037 this.manager = manager; 038 039 shared = new CommitAfterMethodAdvice(manager, null); 040 } 041 042 public void transform(PlasticClass plasticClass, TransformationSupport support, MutableComponentModel model) 043 { 044 for (final PlasticMethod method : plasticClass 045 .getMethodsWithAnnotation(CommitAfter.class)) 046 { 047 PersistenceContext annotation = method.getAnnotation(PersistenceContext.class); 048 049 MethodAdvice advice = annotation == null ? shared : new CommitAfterMethodAdvice(manager, annotation); 050 051 method.addAdvice(advice); 052 } 053 } 054}