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 org.apache.commons.lang.RandomStringUtils;
24 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
25 import org.apache.tinkerpop.gremlin.structure.Vertex;
26 import org.janusgraph.core.JanusGraphTransaction;
27 import org.junit.Before;
28 import org.junit.Test;
29 import org.junit.Ignore;
30 import org.onap.aai.AAISetup;
31 import org.onap.aai.dbmap.AAIGraph;
32 import org.slf4j.Logger;
33 import org.slf4j.LoggerFactory;
34 import org.springframework.test.annotation.DirtiesContext;
36 import java.util.HashSet;
37 import java.util.LinkedHashSet;
40 @Ignore("Run this only to test indexes limit")
41 @DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_CLASS)
42 public class IndexStressTest extends AAISetup {
44 private static final Logger LOGGER = LoggerFactory.getLogger(IndexStressTest.class);
48 AAIGraph.getInstance().getGraph();
52 public void testIndexStress(){
53 JanusGraphTransaction tx = AAIGraph.getInstance().getGraph().newTransaction();
55 GraphTraversalSource g = tx.traversal();
57 Set<String> linkNameSet = new HashSet<>();
58 Set<String> aaiUriSet = new HashSet<>();
60 int TOTAL_LINKS = 101000;
62 for(int i = 0; i <TOTAL_LINKS; i++){
64 String linkName = generateName(linkNameSet);
65 aaiUriSet.add("/network/logical-links/logical-link/" + linkName);
68 .property("aai-node-type", "logical-link")
69 .property("link-name", linkName)
70 .property("aai-uri", "/network/logical-links/logical-link/" + linkName)
74 LOGGER.debug("Committing up to index {}", i);
76 tx = AAIGraph.getInstance().getGraph().newTransaction();
83 tx = AAIGraph.getInstance().getGraph().newTransaction();
87 int totalLinksWithNodeType = 0;
88 int totalLinksUsingUri = 0;
91 for (String linkName : linkNameSet) {
93 if(g.V().has("aai-node-type", "logical-link").has("link-name", linkName).hasNext()){
94 totalLinksWithNodeType++;
97 if(g.V().has("link-name", linkName).hasNext()){
101 if(g.V().has("aai-uri", "/network/logical-links/logical-link/" + linkName).hasNext()){
102 totalLinksUsingUri++;
108 LOGGER.debug("Processed {} many queries and has {} many to go", index, (TOTAL_LINKS-index));
109 LOGGER.debug("Total links using linkname found: {}", totalLinks);
110 LOGGER.debug("Total links using nodetype and linkname found: {}", totalLinksWithNodeType);
111 LOGGER.debug("Total links using uri found: {}", totalLinksUsingUri);
117 LOGGER.debug("Total links using linkname found: {}", totalLinks);
118 LOGGER.debug("Total links using nodetype and linkname found: {}", totalLinksWithNodeType);
119 LOGGER.debug("Total links using uri found: {}", totalLinksUsingUri);
122 String generateName(Set<String> uniqueKeys){
125 String data = RandomStringUtils.randomAlphabetic(20);
126 if (!uniqueKeys.contains(data)){
127 uniqueKeys.add(data);