common/src/main/java/org/onap/so/client/graphinventory/GraphInventoryQueryClient.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 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
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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=========================================================
  19  */
  20
  21 package org.onap.so.client.graphinventory;
  22
  23 import java.util.Optional;
  24
  25 import org.onap.so.client.aai.entities.CustomQuery;
  26 import org.onap.so.client.graphinventory.entities.uri.GraphInventoryUri;
  27
  28 public abstract class GraphInventoryQueryClient<S> {
  29
  30         private Optional<String> depth = Optional.empty();
  31         private boolean nodesOnly = false;
  32         private Optional<GraphInventorySubgraphType> subgraph = Optional.empty();
  33         private GraphInventoryClient client;
  34
  35         public GraphInventoryQueryClient(GraphInventoryClient client) {
  36                 this.client = client;
  37         }
  38
  39         protected abstract GraphInventoryUri getQueryUri();
  40
  41         public String query(Format format, CustomQuery query) {
  42                 return client.createClient(setupQueryParams(getQueryUri().queryParam("format", format.toString()))).put(query, String.class);
  43         }
  44
  45         public S depth (String depth) {
  46                 this.depth = Optional.of(depth);
  47                 return (S) this;
  48         }
  49         public S nodesOnly() {
  50                 this.nodesOnly = true;
  51                 return (S) this;
  52         }
  53         public S subgraph(GraphInventorySubgraphType type){
  54
  55                 subgraph =  Optional.of(type);
  56
  57                 return (S) this;
  58         }
  59
  60         protected GraphInventoryUri setupQueryParams(GraphInventoryUri uri) {
  61                 GraphInventoryUri clone = uri.clone();
  62                 if (this.depth.isPresent()) {
  63                         clone.queryParam("depth", depth.get());
  64                 }
  65                 if (this.nodesOnly) {
  66                         clone.queryParam("nodesOnly", "");
  67                 }
  68                 if (this.subgraph.isPresent()) {
  69                         clone.queryParam("subgraph", this.subgraph.get().toString());
  70                 }
  71                 return clone;
  72         }
  73 }