(no commit message)
[utils] / wicket / components / src / main / java / org / wamblee / wicket / behavior / FlushEntityManagerBehavior.java
1 /*
2  * Copyright 2005-2010 the original author or authors.
3  * 
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  * 
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  * 
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.wamblee.wicket.behavior;
17
18 import java.util.logging.Level;
19 import java.util.logging.Logger;
20
21 import javax.annotation.Resource;
22 import javax.persistence.EntityManager;
23 import javax.transaction.Status;
24 import javax.transaction.SystemException;
25 import javax.transaction.UserTransaction;
26
27 import org.apache.wicket.markup.html.WebPage;
28
29 /**
30  * Behavior to flush the entity manager after rendering of the page.
31  * 
32  * @author Erik Brakkee
33  * 
34  */
35 public class FlushEntityManagerBehavior extends AbstractPageBehavior {
36
37     private static final Logger LOGGER = Logger
38         .getLogger(FlushEntityManagerBehavior.class.getName());
39
40     private EntityManager entityManager;
41
42     /**
43      * Constructs the behavior.
44      * 
45      * @param aEntityManager
46      *            Contextual reference to an entitymanager.
47      */
48     public FlushEntityManagerBehavior(EntityManager aEntityManager) {
49         entityManager = aEntityManager;
50     }
51
52     @Override
53     public void onAfterRender(WebPage aPage) {
54         try {
55             if (entityManager.isOpen()) {
56                 entityManager.flush();
57             }
58         } catch (Exception e) { // TODO catch the Entity manager exceptions explicitly and rethrow, but log
59                                  // the other onees.
60             LOGGER.log(Level.WARNING, "Could not flush entitymanager", e);
61         }
62     }
63
64 }