summaryrefslogtreecommitdiffstats
path: root/src/main/java/org/openecomp/sparky/dal/elasticsearch/ElasticSearchAdapter.java
blob: 99f09a973c4a94de86e7c8b7df2857d5d95a8ad0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
 * Copyright © 2017 Amdocs
 * ================================================================================
 * 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=========================================================
 *
 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
 */
package org.openecomp.sparky.dal.elasticsearch;

import org.openecomp.sparky.dal.elasticsearch.config.ElasticSearchConfig;
import org.openecomp.sparky.dal.rest.HttpMethod;
import org.openecomp.sparky.dal.rest.OperationResult;
import org.openecomp.sparky.dal.rest.RestDataProvider;
import org.openecomp.sparky.dal.rest.RestfulDataAccessor;

/**
 * The Class ElasticSearchAdapter.
 *
 * @author davea.
 */
public class ElasticSearchAdapter implements ElasticSearchDataProvider {

  private static final String BULK_IMPORT_INDEX_TEMPLATE =
      "{\"index\":{\"_index\":\"%s\",\"_type\":\"%s\",\"_id\":\"%s\", \"_version\":\"%s\"}}\n";

  private final RestDataProvider restDataProvider;
  private final ElasticSearchConfig esConfig;

  /**
   * Instantiates a new elastic search adapter.
   *
   * @param provider the provider
   */
  public ElasticSearchAdapter(RestDataProvider provider, ElasticSearchConfig esConfig) {
    this.restDataProvider = provider;
    this.esConfig = esConfig;
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#doGet(java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doGet(String url, String acceptContentType) {
    return restDataProvider.doGet(url, acceptContentType);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#doDelete(java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doDelete(String url, String acceptContentType) {
    return restDataProvider.doDelete(url, acceptContentType);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#doPost(java.lang.String, java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doPost(String url, String jsonPayload, String acceptContentType) {
    return restDataProvider.doPost(url, jsonPayload, acceptContentType);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#doPut(java.lang.String, java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doPut(String url, String jsonPayload, String acceptContentType) {
    return restDataProvider.doPut(url, jsonPayload, acceptContentType);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#doPatch(java.lang.String, java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doPatch(String url, String jsonPayload, String acceptContentType) {
    return restDataProvider.doPatch(url, jsonPayload, acceptContentType);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#doHead(java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doHead(String url, String acceptContentType) {
    return restDataProvider.doHead(url, acceptContentType);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#clearCache()
   */
  @Override
  public void clearCache() {
    restDataProvider.clearCache();
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.elasticsearch.ElasticSearchDataProvider#doBulkOperation(java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doBulkOperation(String url, String payload) {

    return doRestfulOperation(HttpMethod.PUT, url, payload,
        RestfulDataAccessor.APPLICATION_X_WWW_FORM_URL_ENCODED,
        RestfulDataAccessor.APPLICATION_JSON);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.elasticsearch.ElasticSearchDataProvider#shutdown()
   */
  @Override
  public void shutdown() {
    restDataProvider.shutdown();
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.rest.RestDataProvider#doRestfulOperation(org.openecomp.sparky.dal.rest.HttpMethod, java.lang.String, java.lang.String, java.lang.String, java.lang.String)
   */
  @Override
  public OperationResult doRestfulOperation(HttpMethod method, String url, String payload,
      String payloadType, String acceptContentType) {
    return restDataProvider.doRestfulOperation(method, url, payload, payloadType,
        acceptContentType);
  }

  /* (non-Javadoc)
   * @see org.openecomp.sparky.dal.elasticsearch.ElasticSearchDataProvider#buildBulkImportOperationRequest(java.lang.String, java.lang.String, java.lang.String, java.lang.String, java.lang.String)
   */
  @Override
  public String buildBulkImportOperationRequest(String index, String type, String id,
      String version, String payload) {

    StringBuilder requestPayload = new StringBuilder(128);

    requestPayload.append(String.format(BULK_IMPORT_INDEX_TEMPLATE, index, type, id, version));
    requestPayload.append(payload).append("\n");

    return requestPayload.toString();

  }

  @Override
  public OperationResult retrieveEntityById(String entityId) throws Exception {

    String url = esConfig.getElasticFullUrl("/" +entityId);
    return doGet( url, "application/json");
  }

}