/*
* Copyright 2010-2012 Luca Garulli (l.garulli--at--orientechnologies.com)
*
* Licensed 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 com.orientechnologies.orient.test.database.speed;
import org.testng.Assert;
import org.testng.annotations.Test;
import com.orientechnologies.common.test.SpeedTestMultiThreads;
import com.orientechnologies.orient.core.intent.OIntentMassiveInsert;
import com.orientechnologies.orient.core.metadata.schema.OClass;
import com.orientechnologies.orient.core.metadata.schema.OType;
import com.orientechnologies.orient.test.database.base.OrientMultiThreadTest;
import com.orientechnologies.orient.test.database.base.OrientThreadTest;
import com.tinkerpop.blueprints.impls.orient.OrientBaseGraph;
import com.tinkerpop.blueprints.impls.orient.OrientGraphFactory;
import com.tinkerpop.blueprints.impls.orient.OrientGraphNoTx;
import com.tinkerpop.blueprints.impls.orient.OrientVertexType;
@Test(enabled = false)
public class PLocalCreateVerticesMultiThreadSpeedTest extends OrientMultiThreadTest {
protected static final String URL = "plocal:target/databases/graphspeedtest";
protected final OrientGraphFactory factory = new OrientGraphFactory(URL);
protected long foundObjects;
@Test(enabled = false)
public static class CreateObjectsThread extends OrientThreadTest {
private OrientBaseGraph graph;
public CreateObjectsThread(final SpeedTestMultiThreads parent, final int threadId) {
super(parent, threadId);
}
@Override
public void init() {
OrientGraphFactory factory = new OrientGraphFactory(URL);
graph = factory.getNoTx();
factory.close();
graph.getRawGraph().declareIntent(new OIntentMassiveInsert());
}
public void cycle() {
graph
.addVertex("class:Client,cluster:client_" + currentThreadId(), "uid", "" + currentThreadId() + "_" + data.getCyclesDone());
}
@Override
public void deinit() throws Exception {
if (graph != null)
graph.shutdown();
super.deinit();
}
private int currentThreadId() {
return threadId;
}
}
public PLocalCreateVerticesMultiThreadSpeedTest() {
super(1000000, 4, CreateObjectsThread.class);
}
public static void main(String[] iArgs) throws InstantiationException, IllegalAccessException {
// System.setProperty("url", "memory:test");
PLocalCreateVerticesMultiThreadSpeedTest test = new PLocalCreateVerticesMultiThreadSpeedTest();
test.data.go(test);
}
@Override
public void init() {
final OrientGraphNoTx graph = factory.getNoTx();
try {
if (graph.getVertexType("Client") == null) {
final OrientVertexType clientType = graph.createVertexType("Client");
final OrientVertexType.OrientVertexProperty property = clientType.createProperty("uid", OType.STRING);
property.createIndex(OClass.INDEX_TYPE.UNIQUE_HASH_INDEX);
// CREATE ONE CLUSTER PER THREAD
for (int i = 0; i < getThreads(); ++i) {
System.out.println("Creating cluster: client_" + i + "...");
clientType.addCluster("client_" + i);
}
foundObjects = 0;
} else
foundObjects = graph.countVertices("Client");
} finally {
graph.shutdown();
}
}
@Override
public void deinit() {
final OrientGraphNoTx graph = factory.getNoTx();
try {
final long total = graph.countVertices("Client");
System.out.println("\nTotal objects in Client cluster after the test: " + total);
System.out.println("Created " + (total - foundObjects));
Assert.assertEquals(total - foundObjects, threadCycles);
final long indexedItems = graph.getRawGraph().getMetadata().getIndexManager().getIndex("Client.uid").getSize();
System.out.println("\nTotal indexed objects after the test: " + indexedItems);
} finally {
graph.shutdown();
}
}
}