| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.openjpa.persistence.datacache; |
| |
| import java.util.Collection; |
| import java.util.HashMap; |
| import java.util.Iterator; |
| import java.util.Map; |
| |
| import org.apache.openjpa.datacache.DataCache; |
| import org.apache.openjpa.event.TCPRemoteCommitProvider; |
| import org.apache.openjpa.lib.conf.Configurations; |
| import org.apache.openjpa.persistence.OpenJPAEntityManager; |
| import org.apache.openjpa.persistence.OpenJPAEntityManagerFactory; |
| import org.apache.openjpa.persistence.OpenJPAEntityManagerFactorySPI; |
| import org.apache.openjpa.persistence.common.utils.AbstractTestCase; |
| import org.apache.openjpa.persistence.datacache.common.apps.RuntimeTest1; |
| import org.apache.openjpa.persistence.datacache.common.apps.RuntimeTest2; |
| import org.apache.openjpa.persistence.test.AllowFailure; |
| import org.apache.openjpa.util.Id; |
| |
| @AllowFailure(message="surefire excluded") |
| public class TestDistributedKodoDataCache extends AbstractTestCase { |
| |
| // We want more than 1 instance of each type of class. |
| private static final int NUM_OBJECTS = 4; |
| |
| private Object[] _runtime1sOids; |
| |
| private Object[] _runtime2sOids; |
| |
| private Object spec_oid; |
| |
| public TestDistributedKodoDataCache(String test) { |
| super(test, "datacachecactusapp"); |
| } |
| |
| @Override |
| public void setUp() { |
| deleteAll(RuntimeTest1.class); |
| deleteAll(RuntimeTest2.class); |
| } |
| |
| private interface ChangeOperation { |
| |
| String getName(); |
| |
| void operation(OpenJPAEntityManagerFactory kpmf, |
| boolean asLarge); |
| } |
| |
| private class performAsModify implements ChangeOperation { |
| |
| @Override |
| public String getName() { |
| return "Modify SpecialRuntimeTest1"; |
| } |
| |
| @Override |
| public void operation(OpenJPAEntityManagerFactory kpmf, |
| boolean asLarge) { |
| OpenJPAEntityManager pm; |
| pm = (OpenJPAEntityManager) kpmf.createEntityManager(); |
| if (asLarge) |
| //pm.setLargeTransaction(true); |
| pm.setTrackChangesByType(true); |
| startTx(pm); |
| RuntimeTest1 special = pm.find(RuntimeTest1.class, spec_oid); |
| assertNotNull(special); |
| special.setStringField("SpeicalRuntimeTest1_MODIFIED"); |
| |
| endTx(pm); |
| endEm(pm); |
| } |
| } |
| |
| private class performAsDelete implements ChangeOperation { |
| |
| @Override |
| public String getName() { |
| return "Delete SpecialRuntimeTest1"; |
| } |
| |
| @Override |
| public void operation(OpenJPAEntityManagerFactory kpmf, |
| boolean asLarge) { |
| OpenJPAEntityManager pm; |
| pm = (OpenJPAEntityManager) kpmf.createEntityManager(); |
| if (asLarge) |
| //pm.setLargeTransaction(true); |
| pm.setTrackChangesByType(true); |
| startTx(pm); |
| RuntimeTest1 specialObj = pm.find(RuntimeTest1.class, spec_oid); |
| assertNotNull(specialObj); |
| |
| pm.remove(specialObj); |
| |
| endTx(pm); |
| endEm(pm); |
| } |
| } |
| |
| public void testNormalTransAndDataCacheDelete() { |
| coreTestTransAndChange(new performAsDelete(), false, true); |
| } |
| |
| public void testNormalTransAndDataCacheModify() { |
| coreTestTransAndChange(new performAsModify(), false, false); |
| } |
| |
| public void testLargeTransAndDataCacheDelete() { |
| coreTestTransAndChange(new performAsDelete(), true, true); |
| } |
| |
| public void testLargeTransAndDataCacheModify() { |
| coreTestTransAndChange(new performAsModify(), true, false); |
| } |
| |
| public void coreTestTransAndChange(ChangeOperation changeOperation, |
| boolean asLargeTransaction, boolean isDelete) { |
| // Create two pmfs in a cluster that are using RCPTCP. |
| OpenJPAEntityManagerFactory pmfSender = createDistinctFactory( |
| TCPRemoteCommitProvider.class, |
| "Port=5636, Addresses=127.0.0.1:6636"); |
| OpenJPAEntityManagerFactory pmfReceiver = createDistinctFactory( |
| TCPRemoteCommitProvider.class, |
| "Port=6636, Addresses=127.0.0.1:5636"); |
| // Get the datacaches from each pmf |
| DataCache dcSender = |
| ((OpenJPAEntityManagerFactorySPI) pmfSender).getConfiguration() |
| .getDataCacheManagerInstance().getSystemDataCache(); |
| DataCache dcReceiver = |
| ((OpenJPAEntityManagerFactorySPI) pmfReceiver).getConfiguration() |
| .getDataCacheManagerInstance().getSystemDataCache(); |
| |
| deleteAll(RuntimeTest1.class); |
| deleteAll(RuntimeTest2.class); |
| |
| _runtime1sOids = null; |
| _runtime2sOids = null; |
| |
| String transType = "normal"; |
| String rcpType = "OIDs"; |
| if (asLargeTransaction) { |
| transType = "large"; |
| rcpType = "classes"; |
| } |
| System.out.println("-------------------"); |
| System.out.println( |
| "2 PMFs created, acting as a cluster using ports 5636 and 6636"); |
| System.out.println("Testing scenario:"); |
| System.out |
| .println(" 1 Seed datastore with instances of RuntimeTest1 AND " |
| + "RuntimeTest2 objs.\n" |
| + " 2 pmf2 fills its cache with both sets of objs.\n" |
| + " 3 pmf1 does a " |
| + transType |
| + " tx that invokes an operation of " |
| + changeOperation.getName() |
| + " to affect a single Runtime1 \n" |
| + " assert that both pmf1 and pmf2's data caches dropped all" |
| + "Runtime1s"); |
| System.out.println("Remote commit event is transmiting " + rcpType); |
| |
| // Create runtimes1s and 2s. |
| // Noraml transaction |
| OpenJPAEntityManager pmSender = (OpenJPAEntityManager) pmfSender |
| .createEntityManager(); |
| seedDataStore(pmSender, NUM_OBJECTS); |
| endEm(pmSender); |
| |
| // wait a bit so they get stored |
| pause(1); |
| |
| OpenJPAEntityManager pm2; |
| pm2 = (OpenJPAEntityManager) pmfReceiver.createEntityManager(); |
| performLoadAll(pm2); |
| endEm(pm2); |
| |
| pmSender = (OpenJPAEntityManager) pmfSender.createEntityManager(); |
| performLoadAll(pmSender); |
| endEm(pmSender); |
| |
| // assert that pmf2's data cache now has all the Runtime1 and 2s. |
| for (Object o : _runtime1sOids) { |
| assertTrue(dcReceiver.contains( |
| Id.newInstance(RuntimeTest1.class, o))); |
| } |
| for (Object oid : _runtime2sOids) { |
| assertTrue(dcReceiver.contains( |
| Id.newInstance(RuntimeTest2.class, oid))); |
| } |
| |
| // Modify or delete exactly 1 RuntimeTest1 object during a |
| // largeTransaction |
| changeOperation.operation(pmfSender, asLargeTransaction); |
| |
| // assert that pmf1's data cache now only has Runtime2 objects |
| if (asLargeTransaction) { |
| for (Object runtime1sOid : _runtime1sOids) { |
| assertFalse(dcSender.contains( |
| Id.newInstance(RuntimeTest1.class, runtime1sOid))); |
| } |
| } else { |
| // Normal transaction |
| for (int i = 0; i < _runtime1sOids.length; i++) { |
| if (isDelete && i == 0) { |
| assertFalse(dcSender.contains( |
| Id.newInstance(RuntimeTest1.class, _runtime1sOids[i]))); |
| } else { |
| // modified the first elemnt, which just updated it. |
| // (for Kodo's data cache). |
| assertTrue(dcSender.contains( |
| Id.newInstance(RuntimeTest1.class, _runtime1sOids[i]))); |
| } |
| } |
| } |
| for (Object sOid : _runtime2sOids) { |
| assertTrue(dcSender.contains( |
| Id.newInstance(RuntimeTest2.class, sOid))); |
| } |
| // wait a tiny bit so the rce propagates |
| pause(2); |
| // assert the pmf2's data cache also now only has Runtime2 objects |
| if (asLargeTransaction) { |
| for (Object runtime1sOid : _runtime1sOids) { |
| assertFalse(dcReceiver.contains(Id.newInstance( |
| RuntimeTest1.class, runtime1sOid))); //failing here |
| } |
| } else { |
| for (int i = 0; i < _runtime1sOids.length; i++) { |
| if (i == 0) { |
| assertFalse(dcReceiver.contains(Id.newInstance( |
| RuntimeTest1.class, |
| _runtime1sOids[i]))); //failing here |
| } else { |
| assertTrue(dcReceiver.contains( |
| Id.newInstance(RuntimeTest1.class, _runtime1sOids[i]))); |
| } |
| } |
| } |
| for (Object runtime2sOid : _runtime2sOids) { |
| assertTrue(dcReceiver.contains( |
| Id.newInstance(RuntimeTest2.class, runtime2sOid))); |
| } |
| |
| // shutdown |
| pmfSender.close(); |
| pmfReceiver.close(); |
| } |
| |
| protected void performLoadAll(OpenJPAEntityManager pm) |
| // load in (and thus cache) all the 1s and 2s |
| { |
| startTx(pm); |
| RuntimeTest1 temp1; |
| Collection runtime1s = (Collection) pm |
| .createQuery("SELECT a FROM RuntimeTest1 a").getResultList(); |
| |
| for (Object runtime1 : runtime1s) temp1 = (RuntimeTest1) runtime1; |
| RuntimeTest2 temp2; |
| Collection runtime2s = (Collection) pm |
| .createQuery("SELECT a FROM RuntimeTest2 a").getResultList(); |
| |
| for (Object runtime2 : runtime2s) temp2 = (RuntimeTest2) runtime2; |
| endTx(pm); |
| } |
| |
| protected void seedDataStore(OpenJPAEntityManager pm, int numObjects) { |
| startTx(pm); |
| // create objects |
| RuntimeTest1[] persistables = new RuntimeTest1[numObjects]; |
| _runtime1sOids = new Object[numObjects]; |
| for (int i = 0; i < persistables.length; i++) { |
| persistables[i] = new RuntimeTest1("foo #" + i, i); |
| pm.persist(persistables[i]); |
| _runtime1sOids[i] = pm.getObjectId(persistables[i]); |
| if (i == 0) { |
| persistables[i].setStringField("SpecialRuntimeTest1"); |
| spec_oid = pm.getObjectId(persistables[i]); |
| } |
| } |
| RuntimeTest2[] persistables2 = new RuntimeTest2[numObjects]; |
| _runtime2sOids = new Object[numObjects]; |
| for (int i = 0; i < persistables2.length; i++) { |
| persistables2[i] = new RuntimeTest2("bar #" + i, i); |
| pm.persist(persistables2[i]); |
| _runtime2sOids[i] = pm.getObjectId(persistables2[i]); |
| } |
| endTx(pm); |
| } |
| |
| static int _fetchGroupSerial = 0; |
| |
| protected OpenJPAEntityManagerFactory createDistinctFactory( |
| Class providerClass, String classProps1) { |
| Map propsMap; |
| |
| if (providerClass != null) { |
| // This test is for the combination of RCP, largeTrans, |
| // and Kodo's builtin DataCache. |
| // use this property to differentiate the factory |
| |
| propsMap = new HashMap(); |
| propsMap.put("openjpa.DataCache", "lru"); |
| propsMap.put("openjpa.RemoteCommitProvider", |
| Configurations.getPlugin(providerClass.getName(), classProps1)); |
| propsMap.put("openjpa.FetchGroups", "differentiatingFetchGroup" |
| + _fetchGroupSerial); |
| } else { |
| // No RCP |
| propsMap = new HashMap(); |
| propsMap.put("openjpa.RemoteCommitProvider", "sjvm"); |
| propsMap.put("openjpa.FetchGroups", "differentiatingFetchGroup" |
| + _fetchGroupSerial); |
| } |
| _fetchGroupSerial += 1; |
| |
| return (OpenJPAEntityManagerFactory) getEmf(propsMap); |
| } |
| |
| private void pause(double seconds) { |
| try { |
| Thread.currentThread(); |
| Thread.yield(); |
| Thread.currentThread(); |
| Thread.sleep((int) seconds * 1000); |
| } |
| catch (Exception e) { |
| } |
| } |
| } |
| |
| /* |
| remove all Runteim1, and 2 |
| create a few 1 and 2s |
| get them into the cache, assert |
| large transaction, and modify a runtime2, assert cache only has runtime1s |
| |
| |
| remove all Runteim1, and 2 |
| create a few 1 and 2s |
| get them into the cache, assert |
| large transaction, delete a runtime2, assert cache only has runteime1s |
| */ |
| |