/*
* Copyright (C) 2009 eXo Platform SAS.
*
* This 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.1 of
* the License, or (at your option) any later version.
*
* This software 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 software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.exoplatform.services.jcr.lab.cluster.test;
import junit.framework.AssertionFailedError;
import org.exoplatform.services.jcr.JcrAPIBaseTest;
import org.exoplatform.services.jcr.core.CredentialsImpl;
import org.exoplatform.services.jcr.impl.core.SessionImpl;
import org.exoplatform.services.jcr.lab.cluster.prepare.TestLoadIndexerWriter;
import org.exoplatform.services.log.ExoLogger;
import org.exoplatform.services.log.Log;
import java.util.Random;
import javax.jcr.Node;
import javax.jcr.NodeIterator;
import javax.jcr.Property;
import javax.jcr.PropertyIterator;
import javax.jcr.RepositoryException;
import javax.jcr.query.Query;
import javax.jcr.query.QueryManager;
import javax.jcr.query.QueryResult;
/**
* @author <a href="mailto:nikolazius@gmail.com">Nikolay Zamosenchuk</a>
* @version $Id: TestLoadIndexerQuery.java 34360 2009-07-22 23:58:59Z nzamosenchuk $
*
*/
public class TestLoadPredictableIndexerQuery extends JcrAPIBaseTest
{
protected final Log log = ExoLogger.getLogger("exo.jcr.component.core.TestLoadPredictableIndexerQuery");
private boolean stop = false;
private int queryThreadCount = 20;
private int writerThreadCount = 3;
public void testQuery()
{
log.info("Creating threads...");
try
{
for (int i = 0; i < queryThreadCount; i++)
{
// create new thread and start it
new Thread(new ReadTask()).start();
log.info("Thread#" + i + " created and started.");
}
// wait 4 minutes
try
{
Thread.sleep(60000 * 60);
}
catch (InterruptedException e)
{
log.error(e);
}
stop = true;
}
catch (Exception e)
{
// TODO Auto-generated catch block
e.printStackTrace();
}
System.exit(0);
}
private class ReadTask implements Runnable
{
private SessionImpl sessionLocal;
private Node rootLocal;
private Random random;
public ReadTask() throws RepositoryException
{
// login
CredentialsImpl credentials = new CredentialsImpl("admin", "admin".toCharArray());
sessionLocal = (SessionImpl)repository.login(credentials, "ws");
// prepare nodes
rootLocal = sessionLocal.getRootNode();
random = new Random();
}
/**
* @see java.lang.Runnable#run()
*/
public void run()
{
try
{
while (!stop)
{
Node threadNode = getRandomChild(rootLocal, "Thread*");
if (threadNode != null)
{
// StringBuffer sb = new StringBuffer();
// dump(threadNode, sb);
// log.info(sb.toString());
Node statisticNode = threadNode.getNode(TestLoadIndexerWriter.STATISTIC);
Node contentNode = threadNode.getNode(TestLoadIndexerWriter.CONTENT);
Node wordNode = getRandomStatisticChild(statisticNode);
if (wordNode != null)
{
String word = wordNode.getName();
Long count = wordNode.getProperty(TestLoadIndexerWriter.COUNT).getLong();
try
{
Thread.sleep(1);
}
catch (InterruptedException e1)
{
}
QueryManager qman = sessionLocal.getWorkspace().getQueryManager();
Query q =
qman.createQuery("SELECT * FROM nt:base WHERE jcr:path LIKE '" + contentNode.getPath()
+ "/%' and fn:name() = '" + word + "'", Query.SQL);
QueryResult res = q.execute();
long sqlsize = res.getNodes().getSize();
System.out.println("Exp: " + count + "\t found:" + sqlsize);
try
{
//assertTrue("Exp: "+count+"\t found:"+sqlsize, sqlsize >= count);
}
catch (AssertionFailedError e)
{
System.out.println("-" + e.getMessage());
}
}
}
}
}
catch (RepositoryException e)
{
log.error(e);
}
}
private void dump(Node nodesRoot, StringBuffer buffer) throws RepositoryException
{
PropertyIterator iter = nodesRoot.getProperties();
while (iter.hasNext())
{
Property prop = iter.nextProperty();
buffer.append(prop.getName()).append("=").append(prop.getString()).append("\n");
}
for (NodeIterator children = nodesRoot.getNodes(); children.hasNext();)
{
Node node = children.nextNode();
buffer.append(node.getPath()).append("\n");
dump(node, buffer);
}
}
private Node getRandomStatisticChild(Node parent) throws RepositoryException
{
// NodeIterator iterator = parent.getNodes(pattern);
// if (iterator.getSize() < 1)
// {
// return null;
// }
// int i = random.nextInt((int)iterator.getSize());
// iterator.skip(i);
// return iterator.nextNode();
int i = random.nextInt(TestLoadIndexerWriter.WORDS.length);
Node result = null;
for (int j = 0; j < writerThreadCount; j++)
{
try
{
result = parent.getNode(TestLoadIndexerWriter.WORDS[i] + j);
if (result != null)
break;
}
catch (Exception e)
{
// TODO Auto-generated catch block
// e.printStackTrace();
}
}
return result;
}
private Node getRandomChild(Node parent, String pattern) throws RepositoryException
{
// NodeIterator iterator = parent.getNodes(pattern);
// if (iterator.getSize() < 1)
// {
// return null;
// }
// int i = random.nextInt((int)iterator.getSize());
// iterator.skip(i);
// return iterator.nextNode();
int i = random.nextInt(writerThreadCount);
Node result = null;
try
{
result = parent.getNode("Thread" + i);
}
catch (Exception e)
{
// TODO Auto-generated catch block
// e.printStackTrace();
}
return result;
}
}
}