StressTest.java

/*
 *  eXist Open Source Native XML Database
 *  Copyright (C) 2001-04 The eXist Project
 *  http://exist-db.org
 *  
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
 *  
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *  
 *  $Id$
 */
package org.exist.xupdate;

import java.util.Random;

import junit.framework.TestCase;
import junit.textui.TestRunner;

import org.exist.storage.DBBroker;
import org.exist.xmldb.concurrent.DBUtils;
import org.xmldb.api.base.Collection;
import org.xmldb.api.base.Resource;
import org.xmldb.api.base.ResourceSet;
import org.xmldb.api.modules.CollectionManagementService;
import org.xmldb.api.modules.XMLResource;
import org.xmldb.api.modules.XPathQueryService;
import org.xmldb.api.modules.XUpdateQueryService;

/**
 * @author wolf
 *
 */
public class StressTest extends TestCase {

    public static void main(String[] args) {
        TestRunner.run(StressTest.class);
    }
    
    private final static String XML = "<root><a/><b/><c/></root>";
    
    private final static String URI = "xmldb:exist://" + DBBroker.ROOT_COLLECTION;
    
    private final static int RUNS = 1000;
    
    private Collection rootCol;
    private Collection testCol;
    private Random rand = new Random();
    
    private String[] tags;
    
    public void test() throws Exception {
        try {
            insertTags();
            removeTags();
            fetchDb();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    private void insertTags() throws Exception {
        XUpdateQueryService service = (XUpdateQueryService)
            testCol.getService("XUpdateQueryService", "1.0");
        XPathQueryService xquery = (XPathQueryService)
            testCol.getService("XPathQueryService", "1.0");
        
        String[] tagsWritten = new String[RUNS];
        for (int i = 0; i < RUNS; i++) {
            String tag = tags[i];
            String parent;
            if (i > 0 && rand.nextInt(100) < 70) {
                parent = "//" + tagsWritten[rand.nextInt(i) / 2];
            } else
                parent = "/root";
            String xupdate = "<xupdate:modifications version=\"1.0\" xmlns:xupdate=\"http://www.xmldb.org/xupdate\">" +
                "<xupdate:append select=\"" + parent + "\">" +
                "<xupdate:element name=\"" + tag + "\"/>" +
                "</xupdate:append>" +
                "</xupdate:modifications>";
            
            long mods = service.updateResource("test.xml", xupdate);
            System.out.println("Inserted " + tag + ": " + mods + " ; parent = " + parent);
            assertEquals(mods, 1);
            
            tagsWritten[i] = tag;
            String query = "//" + tagsWritten[rand.nextInt(i + 1)];
            ResourceSet result = xquery.query(query);
            assertEquals(result.getSize(), 1);
            
            System.out.println(result.getResource(0).getContent());
        }
        
        XMLResource res = (XMLResource) testCol.getResource("test.xml");
        assertNotNull(res);
        System.out.println(res.getContent());
    }
    
    private void removeTags() throws Exception {
        XUpdateQueryService service = (XUpdateQueryService)
            testCol.getService("XUpdateQueryService", "1.0");
        int start = rand.nextInt(RUNS / 4);
        for (int i = start; i < RUNS; i++) {
            String xupdate = "<xupdate:modifications version=\"1.0\" xmlns:xupdate=\"http://www.xmldb.org/xupdate\">" +
            "<xupdate:remove select=\"//" + tags[i] + "\"/>" +
            "</xupdate:modifications>";
            
            long mods = service.updateResource("test.xml", xupdate);
            System.out.println("Removed: " + tags[i]);
            
            i += rand.nextInt(3);
        }
    }
    
    private void fetchDb() throws Exception {
        XPathQueryService xquery = (XPathQueryService)
            testCol.getService("XPathQueryService", "1.0");
        ResourceSet result = xquery.query("for $n in collection('" + DBBroker.ROOT_COLLECTION + "/test')//* return local-name($n)");
        
        for (int i = 0; i < result.getSize(); i++) {
            Resource r = result.getResource(i);
            String tag = r.getContent().toString();
            System.out.println("Retrieving " + tag);
            
            ResourceSet result2 = xquery.query("//" + tag);
            assertEquals(result2.getSize(), 1);
            
            System.out.println(result2.getResource(0).getContent());
        }
    }
    
    protected void setUp() throws Exception {
        rootCol = DBUtils.setupDB(URI);
        

        testCol = rootCol.getChildCollection(DBBroker.ROOT_COLLECTION + "/test");
        if(testCol != null) {
            CollectionManagementService mgr = DBUtils.getCollectionManagementService(rootCol);
            mgr.removeCollection(DBBroker.ROOT_COLLECTION + "/test");
        }
        
        testCol = DBUtils.addCollection(rootCol, "test");
        assertNotNull(testCol);
        
        System.out.println("Generating " + RUNS + " tags ...");
        
        tags = new String[RUNS];
        for (int i = 0; i < RUNS; i++) {
            tags[i] = "TAG" + i;
        }
        
        
        DBUtils.addXMLResource(testCol, "test.xml", XML);
    }
    
    protected void tearDown() throws Exception {
//        DBUtils.shutdownDB(URI);
    }
}