/* * Copyright (C) 2005 Alfresco, Inc. * * Licensed under the Mozilla Public License version 1.1 * with a permitted attribution clause. You may obtain a * copy of the License at * * http://www.alfresco.org/legal/license.txt * * 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.alfresco.repo.node; import java.io.InputStream; import java.io.Serializable; import java.util.HashMap; import java.util.Map; import junit.framework.TestCase; import org.alfresco.model.ContentModel; import org.alfresco.repo.dictionary.DictionaryComponent; import org.alfresco.repo.dictionary.DictionaryDAO; import org.alfresco.repo.dictionary.M2Model; import org.alfresco.repo.node.integrity.IntegrityChecker; import org.alfresco.repo.transaction.AlfrescoTransactionSupport; import org.alfresco.repo.transaction.TransactionUtil; import org.alfresco.repo.transaction.TransactionUtil.TransactionWork; import org.alfresco.service.cmr.dictionary.DictionaryService; import org.alfresco.service.cmr.repository.ChildAssociationRef; import org.alfresco.service.cmr.repository.ContentService; import org.alfresco.service.cmr.repository.ContentWriter; import org.alfresco.service.cmr.repository.NodeRef; import org.alfresco.service.cmr.repository.NodeService; import org.alfresco.service.cmr.repository.StoreRef; import org.alfresco.service.namespace.QName; import org.alfresco.service.transaction.TransactionService; import org.alfresco.util.ApplicationContextHelper; import org.springframework.context.ApplicationContext; /** * PerformanceNodeServiceTest */ public class PerformanceNodeServiceTest extends TestCase { public static final String NAMESPACE = "http://www.alfresco.org/test/BaseNodeServiceTest"; public static final String TEST_PREFIX = "test"; public static final QName TYPE_QNAME_TEST = QName.createQName(NAMESPACE, "multiprop"); public static final QName PROP_QNAME_NAME = QName.createQName(NAMESPACE, "name"); public static final QName ASSOC_QNAME_CHILDREN = QName.createQName(NAMESPACE, "child"); private int flushCount = Integer.MAX_VALUE; private int testDepth = 3; private int testChildCount = 5; private int testStringPropertyCount = 10; private int testContentPropertyCount = 10; private static ApplicationContext applicationContext = ApplicationContextHelper.getApplicationContext(); protected DictionaryService dictionaryService; protected NodeService nodeService; private ContentService contentService; private TransactionService txnService; private int nodeCount = 0; private long startTime; /** populated during setup */ protected NodeRef rootNodeRef; @Override protected void setUp() throws Exception { DictionaryDAO dictionaryDao = (DictionaryDAO) applicationContext.getBean("dictionaryDAO"); // load the system model ClassLoader cl = PerformanceNodeServiceTest.class.getClassLoader(); InputStream modelStream = cl.getResourceAsStream("alfresco/model/contentModel.xml"); assertNotNull(modelStream); M2Model model = M2Model.createModel(modelStream); dictionaryDao.putModel(model); // load the test model modelStream = cl.getResourceAsStream("org/alfresco/repo/node/BaseNodeServiceTest_model.xml"); assertNotNull(modelStream); model = M2Model.createModel(modelStream); dictionaryDao.putModel(model); DictionaryComponent dictionary = new DictionaryComponent(); dictionary.setDictionaryDAO(dictionaryDao); dictionaryService = loadModel(applicationContext); nodeService = (NodeService) applicationContext.getBean("nodeService"); txnService = (TransactionService) applicationContext.getBean("transactionComponent"); contentService = (ContentService) applicationContext.getBean("contentService"); // create a first store directly TransactionWork createStoreWork = new TransactionWork() { public NodeRef doWork() { StoreRef storeRef = nodeService.createStore( StoreRef.PROTOCOL_WORKSPACE, "Test_" + System.nanoTime()); return nodeService.getRootNode(storeRef); } }; rootNodeRef = TransactionUtil.executeInUserTransaction(txnService, createStoreWork); } @Override protected void tearDown() { } /** * Loads the test model required for building the node graphs */ public static DictionaryService loadModel(ApplicationContext applicationContext) { DictionaryDAO dictionaryDao = (DictionaryDAO) applicationContext.getBean("dictionaryDAO"); // load the system model ClassLoader cl = PerformanceNodeServiceTest.class.getClassLoader(); InputStream modelStream = cl.getResourceAsStream("alfresco/model/contentModel.xml"); assertNotNull(modelStream); M2Model model = M2Model.createModel(modelStream); dictionaryDao.putModel(model); // load the test model modelStream = cl.getResourceAsStream("org/alfresco/repo/node/BaseNodeServiceTest_model.xml"); assertNotNull(modelStream); model = M2Model.createModel(modelStream); dictionaryDao.putModel(model); DictionaryComponent dictionary = new DictionaryComponent(); dictionary.setDictionaryDAO(dictionaryDao); return dictionary; } public void testSetUp() throws Exception { assertNotNull("StoreService not set", nodeService); assertNotNull("NodeService not set", nodeService); assertNotNull("rootNodeRef not created", rootNodeRef); } public void testPerformanceNodeService() throws Exception { startTime = System.currentTimeMillis(); // ensure that we execute the node tree building in a transaction TransactionWork buildChildrenWork = new TransactionWork() { public Object doWork() { IntegrityChecker.setWarnInTransaction(); buildNodeChildren(rootNodeRef, 1, testDepth, testChildCount); return null; } }; TransactionUtil.executeInUserTransaction(txnService, buildChildrenWork); long endTime = System.currentTimeMillis(); System.out.println("Test completed: \n" + " Built " + nodeCount + " nodes in " + (endTime-startTime) + "ms \n" + " Depth: " + testDepth + "\n" + " Child count: " + testChildCount); } public void buildNodeChildren(NodeRef parent, int level, int maxLevel, int childCount) { for (int i=0; i < childCount; i++) { ChildAssociationRef assocRef = this.nodeService.createNode( parent, ASSOC_QNAME_CHILDREN, QName.createQName(NAMESPACE, "child" + i), TYPE_QNAME_TEST); nodeCount++; NodeRef childRef = assocRef.getChildRef(); this.nodeService.setProperty(childRef, ContentModel.PROP_NAME, "node" + level + "_" + i); Map properties = new HashMap(17); for (int j = 0; j < testStringPropertyCount; j++) { properties.put( QName.createQName(NAMESPACE, "string" + j), level + "_" + i + "_" + j); } this.nodeService.setProperties(childRef, properties); for (int j = 0; j < testContentPropertyCount; j++) { ContentWriter writer = this.contentService.getWriter( childRef, QName.createQName(NAMESPACE, "content" + j), true); writer.setMimetype("text/plain"); writer.putContent( level + "_" + i + "_" + j ); } long currentTime = System.currentTimeMillis(); long diffTime = (currentTime - startTime); if (nodeCount % flushCount == 0) { System.out.println("Flushing transaction cache at nodecount: " + nodeCount); System.out.println("At time index " + diffTime + "ms"); AlfrescoTransactionSupport.flush(); } if (nodeCount % 100 == 0) { System.out.println("Interim summary: \n" + " nodes: " + nodeCount + "\n" + " time: " + (double)diffTime/1000.0/60.0 + " minutes \n" + " average: " + (double)nodeCount/(double)diffTime*1000.0 + " nodes/s"); } if (level < maxLevel) { buildNodeChildren(childRef, level + 1, maxLevel, childCount); } } } /** * Runs a test with more depth */ public static void main(String[] args) { try { PerformanceNodeServiceTest test = new PerformanceNodeServiceTest(); test.setUp(); test.testChildCount = 5; test.testDepth = 6; test.flushCount = 1000; test.testPerformanceNodeService(); test.tearDown(); } catch (Throwable e) { e.printStackTrace(); System.exit(1); } System.exit(0); } }