diff options
Diffstat (limited to 'common/src/main/java/org/openecomp/mso/client/aai/AAIQueryClient.java')
-rw-r--r-- | common/src/main/java/org/openecomp/mso/client/aai/AAIQueryClient.java | 93 |
1 files changed, 93 insertions, 0 deletions
diff --git a/common/src/main/java/org/openecomp/mso/client/aai/AAIQueryClient.java b/common/src/main/java/org/openecomp/mso/client/aai/AAIQueryClient.java new file mode 100644 index 0000000000..f8707dec3e --- /dev/null +++ b/common/src/main/java/org/openecomp/mso/client/aai/AAIQueryClient.java @@ -0,0 +1,93 @@ +/*- + * ============LICENSE_START======================================================= + * ONAP - SO + * ================================================================================ + * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved. + * ================================================================================ + * 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. + * ============LICENSE_END========================================================= + */ + +package org.openecomp.mso.client.aai; + +import java.util.Optional; +import java.util.UUID; + +import org.openecomp.mso.client.aai.entities.CustomQuery; +import org.openecomp.mso.client.aai.entities.uri.AAIUri; +import org.openecomp.mso.client.aai.entities.uri.AAIUriFactory; +import org.openecomp.mso.client.policy.RestClient; +import org.openecomp.mso.client.aai.AAIVersion; + +public class AAIQueryClient extends AAIClient { + + + private final AAIVersion version; + private Optional<String> depth = Optional.empty(); + private boolean nodesOnly = false; + private Optional<AAISubgraphType> subgraph = Optional.empty(); + + public AAIQueryClient() { + super(UUID.randomUUID()); + this.version = super.getVersion(); + } + + public AAIQueryClient(AAIVersion version, UUID requestId) { + super(requestId); + this.version = version; + } + + public AAIQueryClient(AAIVersion version) { + this(version, UUID.randomUUID()); + } + + public String query(Format format, CustomQuery query) { + return this.createClient(AAIUriFactory.createResourceUri(AAIObjectType.CUSTOM_QUERY).queryParam("format", format.toString())) + .addRequestId(requestId).put(query, String.class); + } + + @Override + protected AAIVersion getVersion() { + return this.version; + } + + public AAIQueryClient depth (String depth) { + this.depth = Optional.of(depth); + return this; + } + public AAIQueryClient nodesOnly() { + this.nodesOnly = true; + return this; + } + public AAIQueryClient subgraph(AAISubgraphType type){ + + subgraph = Optional.of(type); + + return this; + } + + @Override + public RestClient createClient(AAIUri uri) { + AAIUri clone = uri.clone(); + if (this.depth.isPresent()) { + clone.queryParam("depth", depth.get()); + } + if (this.nodesOnly) { + clone.queryParam("nodesOnly", ""); + } + if (this.subgraph.isPresent()) { + clone.queryParam("subgraph", this.subgraph.get().toString()); + } + return super.createClient(clone); + } +}
\ No newline at end of file |