2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ============LICENSE_END=========================================================
21 package org.onap.aai.stress;
23 import java.util.HashSet;
26 import org.apache.commons.lang3.RandomStringUtils;
27 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
28 import org.apache.tinkerpop.gremlin.structure.Vertex;
29 import org.janusgraph.core.JanusGraphTransaction;
30 import org.junit.Before;
31 import org.junit.Ignore;
32 import org.junit.Test;
33 import org.onap.aai.AAISetup;
34 import org.onap.aai.dbmap.AAIGraph;
35 import org.slf4j.Logger;
36 import org.slf4j.LoggerFactory;
37 import org.springframework.test.annotation.DirtiesContext;
39 @Ignore("Run this only to test indexes limit")
40 @DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_CLASS)
41 public class IndexStressTest extends AAISetup {
43 private static final Logger LOGGER = LoggerFactory.getLogger(IndexStressTest.class);
47 AAIGraph.getInstance().getGraph();
51 public void testIndexStress() {
52 JanusGraphTransaction tx = AAIGraph.getInstance().getGraph().newTransaction();
54 GraphTraversalSource g = tx.traversal();
56 Set<String> linkNameSet = new HashSet<>();
57 Set<String> aaiUriSet = new HashSet<>();
59 int TOTAL_LINKS = 101000;
61 for (int i = 0; i < TOTAL_LINKS; i++) {
63 String linkName = generateName(linkNameSet);
64 aaiUriSet.add("/network/logical-links/logical-link/" + linkName);
66 Vertex v = g.addV().property("aai-node-type", "logical-link").property("link-name", linkName)
67 .property("aai-uri", "/network/logical-links/logical-link/" + linkName).next();
70 LOGGER.debug("Committing up to index {}", i);
72 tx = AAIGraph.getInstance().getGraph().newTransaction();
79 tx = AAIGraph.getInstance().getGraph().newTransaction();
83 int totalLinksWithNodeType = 0;
84 int totalLinksUsingUri = 0;
87 for (String linkName : linkNameSet) {
89 if (g.V().has("aai-node-type", "logical-link").has("link-name", linkName).hasNext()) {
90 totalLinksWithNodeType++;
93 if (g.V().has("link-name", linkName).hasNext()) {
97 if (g.V().has("aai-uri", "/network/logical-links/logical-link/" + linkName).hasNext()) {
103 if (index % 1000 == 0) {
104 LOGGER.debug("Processed {} many queries and has {} many to go", index, (TOTAL_LINKS - index));
105 LOGGER.debug("Total links using linkname found: {}", totalLinks);
106 LOGGER.debug("Total links using nodetype and linkname found: {}", totalLinksWithNodeType);
107 LOGGER.debug("Total links using uri found: {}", totalLinksUsingUri);
113 LOGGER.debug("Total links using linkname found: {}", totalLinks);
114 LOGGER.debug("Total links using nodetype and linkname found: {}", totalLinksWithNodeType);
115 LOGGER.debug("Total links using uri found: {}", totalLinksUsingUri);
118 String generateName(Set<String> uniqueKeys) {
121 String data = RandomStringUtils.randomAlphabetic(20);
122 if (!uniqueKeys.contains(data)) {
123 uniqueKeys.add(data);