@Test public void basicTransactionRollbackTest() { Environment env = createEnvironment(context); //TransactionManager txm = (TransactionManager) env.get( EnvironmentName.TRANSACTION_MANAGER ); javax.transaction.TransactionManager tm = (javax.transaction.TransactionManager) env.get( EnvironmentName.TRANSACTION_MANAGER ); TransactionManager txm = new JtaTransactionManager( env.get( EnvironmentName.TRANSACTION ), env.get( EnvironmentName.TRANSACTION_SYNCHRONIZATION_REGISTRY ), tm ); // Create linked transactionTestObjects TransactionTestObject mainObject = new TransactionTestObject(); mainObject.setName("main"); TransactionTestObject subObject = new TransactionTestObject(); subObject.setName("sub"); mainObject.setSubObject(subObject); Cache<Serializable, Object> cache = cm.getCache("jbpm-configured-cache"); try { boolean notTxOwner = txm.begin(); Serializable s = generateId(mainObject); mainObject.setId(Long.valueOf(s.toString())); cache.put(s, mainObject); txm.rollback(notTxOwner); } catch ( Exception e ) { fail("There should not be an exception thrown here: " + e.getMessage()); } }
@Override public void set(String name, Object value, Document document, LuceneOptions luceneOptions) { if (value != null) { try { TransactionTestObject obj = (TransactionTestObject) value; Long objId = obj.getId(); String objName = obj.getName(); ByteArrayOutputStream baout = new ByteArrayOutputStream(); ObjectOutputStream oout = new ObjectOutputStream(baout); oout.writeObject(objId); oout.writeUTF(objName); Field field = new Field(name, baout.toByteArray()); field.setBoost(luceneOptions.getBoost()); document.add(field); } catch (Exception e) { throw new RuntimeException("problem bridging SessionInfo", e); } } }
@Test public void testSingleSessionCommandServiceAndJtaTransactionManagerTogether() { // Initialize drools environment stuff Environment env = createEnvironment(context); KieBase kbase = initializeKnowledgeBase(simpleRule); StatefulKnowledgeSession commandKSession = InfinispanKnowledgeService.newStatefulKnowledgeSession( kbase, null, env ); PersistableRunner commandService = (PersistableRunner) ((CommandBasedStatefulKnowledgeSession) commandKSession).getRunner(); InfinispanPersistenceContextManager jpm = (InfinispanPersistenceContextManager) getValueOfField("jpm", commandService); jpm.getApplicationScopedPersistenceContext(); @SuppressWarnings("unchecked") Cache<String, EntityHolder> cache = (Cache<String, EntityHolder>) getValueOfField("appScopedCache", jpm); TransactionTestObject mainObject = new TransactionTestObject(); mainObject.setName("mainCommand"); TransactionTestObject subObject = new TransactionTestObject(); subObject.setName("subCommand"); mainObject.setSubObject(subObject); HashMap<String, Object> emEnv = new HashMap<String, Object>(); emEnv.put(COMMAND_ENTITY_MANAGER_FACTORY, cm); emEnv.put(COMMAND_ENTITY_MANAGER, cache); TransactionTestCommand txTestCmd = new TransactionTestCommand(mainObject, subObject, emEnv); commandKSession.execute(txTestCmd); }
public Void execute(Context context ) { //cache.joinTransaction();TODO TransactionTestObject obj = (TransactionTestObject) mainObject; if (obj.getId() == null) { obj.setId(generateId()); } cache.put(InfinispanPlaceholderResolverStrategy.getClassIdValue(mainObject), mainObject); if( subObject != null ) { KieSession ksession = ((RegistryContext) context).lookup( KieSession.class ); // THe following 3 lines are the important ones! (See below for an explanation) InternalKnowledgeBase cleanKBase = KnowledgeBaseFactory.newKnowledgeBase(); cleanKBase.addPackages(((InternalKnowledgeBase)ksession.getKieBase()).getKiePackages()); StatefulKnowledgeSession commandKSession = InfinispanKnowledgeService.newStatefulKnowledgeSession( cleanKBase, null, initializeEnvironment() ); /** * Here's what's going on: * If the SingleSessionCommandService (SSCS) & JtaTransactionManager (JTM) were _not_ aware of transactions, * -> then inserting the mainObject _before_ having inserted the subObject * would cause the operation/persist to fail and the transaction to fail. * - This is because the mainObject contains a foreign key referring to the subObject. * * However, the SSCS & JTM check whether or not they're owners of the transaction * when starting and when committing the transaction they use. * -> So that when we insert the mainObject here (via a _new_ CommandBasedStatefulKnowledgeSession), * it does _not_ mess with the transaction state and the operation succeeds. */ TransactionTestCommand transactionTestSubCommand = new TransactionTestCommand(this.subObject, getPersistenceEnvironment()); commandKSession.execute(transactionTestSubCommand); } return null; }
TransactionTestObject mainObject = new TransactionTestObject(); mainObject.setName("main" + testName); TransactionTestObject subObject = new TransactionTestObject(); subObject.setName("sub" + testName); mainObject.setSubObject(subObject); mainObject.setId(Long.valueOf(s.toString())); cache.put(s, mainObject); txm.commit(notTxOwner); subObject.setId(Long.valueOf(s2.toString())); cache.put(s2, subObject); txm.commit(txOwner);