5e4467e74b09fc181a8ca4804c12a70283789024
[ccsdk/features.git] /
1 /*
2  * ============LICENSE_START=======================================================
3  * ONAP : ccsdk features
4  * ================================================================================
5  * Copyright (C) 2019 highstreet technologies GmbH Intellectual Property.
6  * All rights reserved.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *     http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  *
21  */
22 package org.onap.ccsdk.features.sdnr.wt.common.database.requests;
23
24 import java.io.UnsupportedEncodingException;
25 import java.net.URLEncoder;
26 import java.nio.charset.StandardCharsets;
27
28 import org.elasticsearch.client.Request;
29 import org.json.JSONObject;
30 import org.onap.ccsdk.features.sdnr.wt.common.database.queries.QueryBuilder;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33
34 public abstract class BaseRequest {
35
36     private static final Logger LOG = LoggerFactory.getLogger(BaseRequest.class);
37
38     public static final int DEFAULT_RETRIES = 3;
39
40     protected final Request request;
41     private String query;
42     private final boolean refresh;
43
44     public BaseRequest(String method, String endpoint) {
45         LOG.debug("create request {} {}", method, endpoint);
46         this.refresh = false;
47         this.request = new Request(method, endpoint);
48         query = null;
49     }
50
51     public BaseRequest(String method, String endpoint, boolean refresh) {
52         LOG.debug("create request {} {} with refresh={}", method, endpoint, refresh);
53         this.refresh = refresh;
54         this.request = new Request(method, String.format("%s?refresh=%s", endpoint, String.valueOf(refresh)));
55         query = null;
56     }
57
58     public BaseRequest(String method, String endpoint, boolean refresh, int retries) {
59         LOG.debug("create request {} {} with refresh={}", method, endpoint, refresh);
60         this.refresh = refresh;
61         this.request = new Request(method,
62                 String.format("%s?refresh=%s&retry_on_conflict=%d", endpoint, String.valueOf(refresh), retries));
63         query = null;
64     }
65
66     public Request getInner() {
67
68         return this.request;
69     }
70
71     public static String urlEncodeValue(String value) {
72         if (value == null)
73             return null;
74         try {
75             return URLEncoder.encode(value, StandardCharsets.UTF_8.toString()).replace("+", "%20");
76         } catch (UnsupportedEncodingException ex) {
77             LOG.warn("encoding problem: {}", ex.getMessage());
78         }
79         return value;
80     }
81
82     @Override
83     public String toString() {
84         return this.request.getMethod() + " " + this.request.getEndpoint() + " : "
85                 + (this.query != null ? this.query : "no query");
86     }
87
88     protected void setQuery(QueryBuilder query) {
89         this.setQuery(query.toJSON());
90     }
91
92     public void setQuery(JSONObject o) {
93         this.setQuery(o.toString());
94     }
95
96     public void setQuery(String content) {
97         this.query = content;
98         LOG.trace("query={}", content);
99         this.request.setJsonEntity(this.query);
100     }
101
102     protected String getQuery() {
103         return this.query;
104     }
105
106     protected boolean doRefresh() {
107         return this.refresh;
108     }
109 }