deleted file mode 100644| .. | .. |
|---|
| 1 | | -package net.curisit.securis.ioc; |
|---|
| 2 | | - |
|---|
| 3 | | -import java.io.IOException; |
|---|
| 4 | | -import java.lang.reflect.Method; |
|---|
| 5 | | - |
|---|
| 6 | | -import javax.annotation.Priority; |
|---|
| 7 | | -import javax.interceptor.InvocationContext; |
|---|
| 8 | | -import javax.persistence.EntityManager; |
|---|
| 9 | | -import javax.persistence.EntityManagerFactory; |
|---|
| 10 | | -import javax.ws.rs.Priorities; |
|---|
| 11 | | -import javax.ws.rs.container.ContainerRequestContext; |
|---|
| 12 | | -import javax.ws.rs.container.ContainerRequestFilter; |
|---|
| 13 | | -import javax.ws.rs.container.ContainerResponseContext; |
|---|
| 14 | | -import javax.ws.rs.container.ContainerResponseFilter; |
|---|
| 15 | | -import javax.ws.rs.core.Response.Status; |
|---|
| 16 | | -import javax.ws.rs.ext.Provider; |
|---|
| 17 | | - |
|---|
| 18 | | -import org.apache.logging.log4j.LogManager; |
|---|
| 19 | | -import org.apache.logging.log4j.Logger; |
|---|
| 20 | | -import org.jboss.resteasy.core.ResourceMethodInvoker; |
|---|
| 21 | | -import org.jboss.resteasy.spi.ResteasyProviderFactory; |
|---|
| 22 | | - |
|---|
| 23 | | -@Provider |
|---|
| 24 | | -@Priority(Priorities.AUTHENTICATION) |
|---|
| 25 | | -public class TransactionsManager implements ContainerRequestFilter, ContainerResponseFilter { |
|---|
| 26 | | - |
|---|
| 27 | | - private static final Logger LOG = LogManager.getLogger(TransactionsManager.class); |
|---|
| 28 | | - |
|---|
| 29 | | - private EntityManagerFactory entityManagerFactory = javax.persistence.Persistence.createEntityManagerFactory("localdb"); |
|---|
| 30 | | - |
|---|
| 31 | | - private EntityManager getEM() { |
|---|
| 32 | | - return entityManagerFactory.createEntityManager(); |
|---|
| 33 | | - } |
|---|
| 34 | | - |
|---|
| 35 | | - // @Inject |
|---|
| 36 | | - // private ThreadLocal<EntityManager> threadLocal; |
|---|
| 37 | | - |
|---|
| 38 | | - // @AroundInvoke |
|---|
| 39 | | - public Object invoke(InvocationContext context) throws Exception { |
|---|
| 40 | | - EntityManager em = getEM(); |
|---|
| 41 | | - if (!em.getTransaction().isActive()) { |
|---|
| 42 | | - em.getTransaction().begin(); |
|---|
| 43 | | - LOG.info("INIT trans"); |
|---|
| 44 | | - ResteasyProviderFactory.pushContext(EntityManager.class, em); |
|---|
| 45 | | - Object result = context.proceed(); |
|---|
| 46 | | - |
|---|
| 47 | | - em.getTransaction().commit(); |
|---|
| 48 | | - LOG.info("COMMIT trans"); |
|---|
| 49 | | - return result; |
|---|
| 50 | | - } |
|---|
| 51 | | - LOG.info("No trans"); |
|---|
| 52 | | - Object result = context.proceed(); |
|---|
| 53 | | - return result; |
|---|
| 54 | | - } |
|---|
| 55 | | - |
|---|
| 56 | | - @Override |
|---|
| 57 | | - public void filter(ContainerRequestContext requestContext) throws IOException { |
|---|
| 58 | | - ResourceMethodInvoker methodInvoker = (ResourceMethodInvoker) requestContext.getProperty("org.jboss.resteasy.core.ResourceMethodInvoker"); |
|---|
| 59 | | - Method method = methodInvoker.getMethod(); |
|---|
| 60 | | - EntityManager em = getEM(); |
|---|
| 61 | | - |
|---|
| 62 | | - ResteasyProviderFactory.pushContext(EntityManager.class, em); |
|---|
| 63 | | - |
|---|
| 64 | | - if (method.isAnnotationPresent(EnsureTransaction.class)) { |
|---|
| 65 | | - LOG.info("WE need transaction!!!"); |
|---|
| 66 | | - |
|---|
| 67 | | - em.getTransaction().begin(); |
|---|
| 68 | | - } |
|---|
| 69 | | - LOG.info("Sent to context, em: {}, {}?", em, method.toGenericString()); |
|---|
| 70 | | - } |
|---|
| 71 | | - |
|---|
| 72 | | - @Override |
|---|
| 73 | | - public void filter(ContainerRequestContext requestContext, ContainerResponseContext responseContext) throws IOException { |
|---|
| 74 | | - // TODO Auto-generated method stub |
|---|
| 75 | | - // ResteasyProviderFactory.pushContext(EntityManager.class, em); |
|---|
| 76 | | - EntityManager em = (EntityManager) ResteasyProviderFactory.getContextData(EntityManager.class); |
|---|
| 77 | | - LOG.info("Operating on em: {}", em); |
|---|
| 78 | | - |
|---|
| 79 | | - if (em != null && em.getTransaction().isActive()) { |
|---|
| 80 | | - LOG.info("There is transaction go ahead..."); |
|---|
| 81 | | - if (responseContext.getStatus() == Status.OK.getStatusCode()) { |
|---|
| 82 | | - em.getTransaction().commit(); |
|---|
| 83 | | - LOG.info("COMMIT"); |
|---|
| 84 | | - } else { |
|---|
| 85 | | - em.getTransaction().rollback(); |
|---|
| 86 | | - LOG.info("ROLLBACK"); |
|---|
| 87 | | - } |
|---|
| 88 | | - } else { |
|---|
| 89 | | - LOG.info("There is NO transaction"); |
|---|
| 90 | | - |
|---|
| 91 | | - } |
|---|
| 92 | | - } |
|---|
| 93 | | - |
|---|
| 94 | | -} |
|---|