001 /** 002 * Licensed to the Apache Software Foundation (ASF) under one or more 003 * contributor license agreements. See the NOTICE file distributed with 004 * this work for additional information regarding copyright ownership. 005 * The ASF licenses this file to You under the Apache License, Version 2.0 006 * (the "License"); you may not use this file except in compliance with 007 * the License. You may obtain a copy of the License at 008 * 009 * http://www.apache.org/licenses/LICENSE-2.0 010 * 011 * Unless required by applicable law or agreed to in writing, software 012 * distributed under the License is distributed on an "AS IS" BASIS, 013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 014 * See the License for the specific language governing permissions and 015 * limitations under the License. 016 */ 017 package org.apache.camel.processor.idempotent.jpa; 018 019 import java.util.List; 020 021 import javax.persistence.EntityManagerFactory; 022 import javax.persistence.Persistence; 023 024 import org.apache.camel.processor.idempotent.MessageIdRepository; 025 026 import org.springframework.orm.jpa.JpaTemplate; 027 import org.springframework.orm.jpa.JpaTransactionManager; 028 import org.springframework.transaction.TransactionDefinition; 029 import org.springframework.transaction.TransactionStatus; 030 import org.springframework.transaction.support.TransactionCallback; 031 import org.springframework.transaction.support.TransactionTemplate; 032 033 /** 034 * @version $Revision: 1.1 $ 035 */ 036 public class JpaMessageIdRepository implements MessageIdRepository { 037 protected static final String QUERY_STRING = "select x from " + MessageProcessed.class.getName() + " x where x.processorName = ?1 and x.messageId = ?2"; 038 private JpaTemplate jpaTemplate; 039 private String processorName; 040 private TransactionTemplate transactionTemplate; 041 042 public JpaMessageIdRepository(JpaTemplate template, String processorName) { 043 this(template, createTransactionTemplate(template), processorName); 044 } 045 046 public JpaMessageIdRepository(JpaTemplate template, TransactionTemplate transactionTemplate, String processorName) { 047 this.jpaTemplate = template; 048 this.processorName = processorName; 049 this.transactionTemplate = transactionTemplate; 050 } 051 052 public static JpaMessageIdRepository jpaMessageIdRepository(String persistenceUnit, String processorName) { 053 EntityManagerFactory entityManagerFactory = Persistence.createEntityManagerFactory(persistenceUnit); 054 return jpaMessageIdRepository(new JpaTemplate(entityManagerFactory), processorName); 055 } 056 057 public static JpaMessageIdRepository jpaMessageIdRepository(JpaTemplate jpaTemplate, String processorName) { 058 return new JpaMessageIdRepository(jpaTemplate, processorName); 059 } 060 061 private static TransactionTemplate createTransactionTemplate(JpaTemplate jpaTemplate) { 062 TransactionTemplate transactionTemplate = new TransactionTemplate(); 063 transactionTemplate.setTransactionManager(new JpaTransactionManager(jpaTemplate.getEntityManagerFactory())); 064 transactionTemplate.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED); 065 return transactionTemplate; 066 } 067 068 public boolean contains(final String messageId) { 069 // Run this in single transaction. 070 Boolean rc = (Boolean)transactionTemplate.execute(new TransactionCallback() { 071 public Object doInTransaction(TransactionStatus arg0) { 072 073 List list = jpaTemplate.find(QUERY_STRING, processorName, messageId); 074 if (list.isEmpty()) { 075 MessageProcessed processed = new MessageProcessed(); 076 processed.setProcessorName(processorName); 077 processed.setMessageId(messageId); 078 jpaTemplate.persist(processed); 079 jpaTemplate.flush(); 080 return Boolean.FALSE; 081 } else { 082 return Boolean.TRUE; 083 } 084 } 085 }); 086 return rc.booleanValue(); 087 } 088 }