aboutsummaryrefslogtreecommitdiffstats
path: root/src/main/java/org/onap/aai/modelloader/entity/model/NamedQueryArtifact.java
blob: f3269806bebde58ae573089e7b174d8f396fa7a5 (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
/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017 AT&T Intellectual Property.
 * Copyright © 2017 Amdocs
 * 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=========================================================
 *
 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
 */
package org.onap.aai.modelloader.entity.model;

import java.util.List;

import javax.ws.rs.core.Response;

import org.onap.aai.modelloader.config.ModelLoaderConfig;
import org.onap.aai.modelloader.entity.ArtifactType;
import org.onap.aai.modelloader.restclient.AaiRestClient;
import org.onap.aai.modelloader.service.ModelLoaderMsgs;
import org.onap.aai.cl.api.Logger;
import org.onap.aai.cl.eelf.LoggerFactory;

import com.sun.jersey.api.client.ClientResponse;

public class NamedQueryArtifact extends AbstractModelArtifact {
		
  private Logger logger = LoggerFactory.getInstance().getLogger(NamedQueryArtifact.class.getName());
  
	private String namedQueryUuid; 
	
	public NamedQueryArtifact() {
	  super(ArtifactType.NAMED_QUERY);
	}
	
	public String getNamedQueryUuid() {
		return namedQueryUuid;
	}
	
	public void setNamedQueryUuid(String namedQueryUuid) {
		this.namedQueryUuid = namedQueryUuid;
	}
	
  @Override
  public String getUniqueIdentifier() {
    return getNamedQueryUuid();
  }	

  @Override
  public boolean push(AaiRestClient aaiClient, ModelLoaderConfig config, String distId, List<AbstractModelArtifact> addedModels) {
    ClientResponse getResponse  = aaiClient.getResource(getNamedQueryUrl(config), distId, AaiRestClient.MimeType.XML);
    if ( (getResponse == null) || (getResponse.getStatus() != Response.Status.OK.getStatusCode()) ) {
      // Only attempt the PUT if the model doesn't already exist
      ClientResponse putResponse = aaiClient.putResource(getNamedQueryUrl(config), getPayload(), distId, AaiRestClient.MimeType.XML);
      if ( (putResponse != null) && (putResponse.getStatus() == Response.Status.CREATED.getStatusCode()) ) {
        addedModels.add(this);
        logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, getType().toString() + " " + getUniqueIdentifier() + " successfully ingested.");
      }
      else {
        logger.error(ModelLoaderMsgs.DISTRIBUTION_EVENT_ERROR, "Ingestion failed for " + getType().toString() + " " + getUniqueIdentifier() +
            ". Rolling back distribution.");
        return false;
      }
    }
    else {
      logger.info(ModelLoaderMsgs.DISTRIBUTION_EVENT, getType().toString() + " " + getUniqueIdentifier() + " already exists.  Skipping ingestion.");
    }
    
    return true;
  }

  @Override
  public void rollbackModel(AaiRestClient aaiClient, ModelLoaderConfig config, String distId) {
    // Best effort to delete.  Nothing we can do in the event this fails.
    aaiClient.getAndDeleteResource(getNamedQueryUrl(config), distId);
  }

  private String getNamedQueryUrl(ModelLoaderConfig config) {
    String baseURL = config.getAaiBaseUrl().trim();
    String subURL = null;
    String instance = null;

    subURL = config.getAaiNamedQueryUrl(getModelNamespaceVersion()).trim();
    instance = this.getNamedQueryUuid();

    if ( (!baseURL.endsWith("/")) && (!subURL.startsWith("/")) ) {
      baseURL = baseURL + "/";
    }

    if ( baseURL.endsWith("/") && subURL.startsWith("/") ) {
      baseURL = baseURL.substring(0, baseURL.length()-1);
    }

    if (!subURL.endsWith("/")) {
      subURL = subURL + "/";
    }

    String url = baseURL + subURL + instance;
    return url;
  }
}