f8f977d11776821fbe51c0625d362086f6ef59de
[so.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP - SO
4  * ================================================================================
5  * Copyright (C) 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
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.aaiclient.client.graphinventory;
22
23 import java.net.URI;
24 import java.util.Map;
25 import org.onap.aaiclient.client.graphinventory.entities.uri.GraphInventoryUri;
26 import org.onap.aaiclient.client.graphinventory.entities.uri.HttpAwareUri;
27 import org.onap.so.client.RestClient;
28 import org.onap.so.client.RestProperties;
29 import org.onap.so.client.RestPropertiesLoader;
30 import com.google.common.collect.ImmutableMap;
31
32 public abstract class GraphInventoryClient {
33
34     private RestProperties props;
35     protected final Map<String, String> additionalHeaders;
36
37     protected GraphInventoryClient(Class<? extends RestProperties> propertiesClass,
38             Map<String, String> additionalHeaders) {
39
40         RestProperties props = RestPropertiesLoader.getInstance().getNewImpl(propertiesClass);
41         this.props = props;
42         this.additionalHeaders = additionalHeaders;
43     }
44
45     protected abstract URI constructPath(URI uri);
46
47     protected abstract RestClient createClient(URI uri);
48
49     public RestClient createClient(GraphInventoryUri uri) {
50         final URI result;
51         if (uri instanceof HttpAwareUri) {
52             result = ((HttpAwareUri) uri).locateAndBuild();
53         } else {
54             result = uri.build();
55         }
56
57         return createClient(result);
58
59     }
60
61
62     public <T extends RestProperties> T getRestProperties() {
63         if (props == null) {
64             throw new IllegalStateException("No RestProperty implementation found on classpath");
65         }
66         return (T) props;
67     }
68
69     public abstract GraphInventoryVersion getVersion();
70
71     public abstract String getGraphDBName();
72
73     public Map<String, String> getAdditionalHeaders() {
74         return ImmutableMap.copyOf(this.additionalHeaders);
75     }
76 }