alfresco-community-repo/source/java/org/alfresco/repo/avm/SimultaneousLoadTest.java

/*
 * Copyright (C) 2005-2008 Alfresco Software Limited.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.

 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 * As a special exception to the terms and conditions of version 2.0 of
 * the GPL, you may redistribute this Program in connection with Free/Libre
 * and Open Source Software ("FLOSS") applications as described in Alfresco's
 * FLOSS exception.  You should have recieved a copy of the text describing
 * the FLOSS exception, and it is also available here:
 * http://www.alfresco.com/legal/licensing" */

package org.alfresco.repo.avm;

import org.alfresco.repo.avm.util.BulkLoader;

/**
 * This tests loading content simultaneously.
 * @author britt
 */
public class SimultaneousLoadTest extends AVMServiceTestBase
{
    public void testSimulLoadA() throws Throwable
    {
        testSimultaneousLoad(1,1);
    }
    
    public void testSimulLoadB() throws Throwable
    {
        testSimultaneousLoad(5,3);
    }
    
    /**
     * Test loading content simultaneously.
     */
    private void testSimultaneousLoad(int n, int m) throws Throwable
    {
        try
        {
            fReaper.setActiveBaseSleep(60000);
            for (int i = 0; i < n; i++)
            {
                fService.createDirectory("main:/", "d" + i);
            }
            fService.createSnapshot("main", null, null);
            Thread [] threads = new Thread[n];
            for (int i = 0; i < n; i++)
            {
                //Loader loader = new Loader("/Users/britt/stuff/" + i, "main:/d" + i, m);
                Loader loader = new Loader("source/java/org/alfresco/repo/avm", "main:/d" + i, m);
                
                threads[i] = new Thread(loader);
                threads[i].start();
            }
            for (int i = 0; i < n; i++)
            {
                threads[i].join();
            }
        }
        catch (Exception e)
        {
            e.printStackTrace(System.err);
            throw e;
        }
    }
    
    private class Loader implements Runnable
    {
        /**
         * The BulkLoader.
         */
        private BulkLoader fLoader;
        
        /**
         * The source directory.
         */
        private String fSource;
        
        /**
         * The destination path.
         */
        private String fDestination;
        
        /**
         * The number of copies of stuff to make serially.
         */
        private int fCount;
        
        /**
         * Set up.
         * @param source Source directory.
         * @param destination Destination path.
         */
        public Loader(String source, String destination, int count)
        {
            fLoader = new BulkLoader();
            fLoader.setAvmService(fService);
            fSource = source;
            fDestination = destination;
            fCount = count;
        }
        
        public void run()
        {
            for (int i = 0; i < fCount; i++)
            {
                fService.createDirectory(fDestination, "" + i);
                fLoader.recursiveLoad(fSource, fDestination + "/" + i);
            }
        }
    }
}