dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/service/MR_ClientService.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.onap.dmaap
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  *
   7  * Modifications Copyright (C) 2019 IBM.
   8  * ================================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *      http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.dmaap.dbcapi.service;
  24
  25 import org.onap.dmaap.dbcapi.aaf.AafService.ServiceType;
  26 import org.onap.dmaap.dbcapi.aaf.AafServiceFactory;
  27 import org.onap.dmaap.dbcapi.client.MrProvConnection;
  28 import org.onap.dmaap.dbcapi.database.DatabaseClass;
  29 import org.onap.dmaap.dbcapi.logging.BaseLoggingClass;
  30 import org.onap.dmaap.dbcapi.model.ApiError;
  31 import org.onap.dmaap.dbcapi.model.DcaeLocation;
  32 import org.onap.dmaap.dbcapi.model.DmaapObject.DmaapObject_Status;
  33 import org.onap.dmaap.dbcapi.model.MR_Client;
  34 import org.onap.dmaap.dbcapi.model.MR_Cluster;
  35 import org.onap.dmaap.dbcapi.model.Topic;
  36 import org.onap.dmaap.dbcapi.util.DmaapConfig;
  37
  38 import javax.ws.rs.core.Response.Status;
  39 import java.util.ArrayList;
  40 import java.util.List;
  41 import java.util.Map;
  42
  43 public class MR_ClientService extends BaseLoggingClass {
  44
  45     private static final String MR_CLIENT_ID = "mrClientId";
  46     private int deleteLevel;
  47     private Map<String, MR_Client> mr_clients = DatabaseClass.getMr_clients();
  48     private Map<String, MR_Cluster> clusters = DatabaseClass.getMr_clusters();
  49     private Map<String, DcaeLocation> locations = DatabaseClass.getDcaeLocations();
  50     private DmaapService dmaap = new DmaapService();
  51     private AafPermissionService aafPermissionService =
  52             new AafPermissionService(new AafServiceFactory().initAafService(ServiceType.AAF_TopicMgr), dmaap);
  53     private String centralCname;
  54
  55     public MR_ClientService() {
  56         DmaapConfig p = (DmaapConfig) DmaapConfig.getConfig();
  57
  58         centralCname = p.getProperty("MR.CentralCname", "MRcname.not.set");
  59         deleteLevel = Integer.valueOf(p.getProperty("MR.ClientDeleteLevel", "0"));
  60     }
  61
  62     public List<MR_Client> getAllMr_Clients() {
  63         return new ArrayList<>(mr_clients.values());
  64     }
  65
  66     List<MR_Client> getAllMrClients(String fqtn) {
  67         ArrayList<MR_Client> results = new ArrayList<>();
  68         for (Map.Entry<String, MR_Client> entry : mr_clients.entrySet()) {
  69             MR_Client client = entry.getValue();
  70             if (fqtn.equals(client.getFqtn())) {
  71                 results.add(client);
  72             }
  73         }
  74         return results;
  75     }
  76
  77     List<MR_Client> getClientsByLocation(String location) {
  78         List<MR_Client> results = new ArrayList<>();
  79         for (Map.Entry<String, MR_Client> entry : mr_clients.entrySet()) {
  80             MR_Client client = entry.getValue();
  81             if (location.equals(client.getDcaeLocationName())) {
  82                 results.add(client);
  83             }
  84         }
  85         return results;
  86     }
  87
  88     public MR_Client getMr_Client(String key, ApiError apiError) {
  89         MR_Client c = mr_clients.get(key);
  90         if (c == null) {
  91             apiError.setCode(Status.NOT_FOUND.getStatusCode());
  92             apiError.setFields(MR_CLIENT_ID);
  93             apiError.setMessage(MR_CLIENT_ID + " " + key + " not found");
  94         } else {
  95             apiError.setCode(200);
  96         }
  97         return c;
  98     }
  99
 100     public MR_Client addMr_Client(MR_Client client, Topic topic, ApiError err) {
 101         if (client.getDcaeLocationName().isEmpty()) {
 102             logger.info("Client  dcaeLocation that doesn't exist or not specified");
 103             return null;
 104         }
 105         // original style: clients specified Role.  This has precedence for backwards
 106         //                 compatibility.
 107         // ONAP style: clients specify Identity to be assigned to generated Role
 108         String role = client.getClientRole();
 109         if (role != null) {
 110             updateApiError(err, aafPermissionService.grantClientRolePerms(client));
 111         } else if (client.hasClientIdentity()) {
 112             if (client.isSubscriber()) {
 113                 role = topic.getSubscriberRole();
 114                 updateApiError(err, aafPermissionService.assignClientToRole(client, role));
 115             }
 116             if (client.isPublisher()) {
 117                 role = topic.getPublisherRole();
 118                 updateApiError(err, aafPermissionService.assignClientToRole(client, role));
 119             }
 120         }
 121         if (!client.isStatusValid()) {
 122             return null;
 123         }
 124         String centralFqdn = null;
 125         DcaeLocation candidate = locations.get(client.getDcaeLocationName());
 126
 127         MR_Cluster cluster = clusters.get(client.getDcaeLocationName());
 128         if (cluster != null && candidate != null) {
 129             if (candidate.isCentral() && !topic.getReplicationCase().involvesFQDN()) {
 130                 centralFqdn = centralCname;
 131             }
 132             client.setTopicURL(cluster.genTopicURL(centralFqdn, client.getFqtn()));
 133             if (centralFqdn == null) {
 134                 client.setStatus(addTopicToCluster(cluster, topic, err));
 135                 if (!err.is2xx() && err.getCode() != 409) {
 136                     topic.setFqtn(err.getMessage());
 137                     return null;
 138                 }
 139
 140             } else {
 141                 MR_ClusterService clusters = new MR_ClusterService();
 142                 //  MM should only exist for edge-to-central
 143                 //  we use a cname for the central target (default resiliency with no replicationGroup set)
 144                 // but still need to provision topics on all central MRs
 145                 for (MR_Cluster central : clusters.getCentralClusters()) {
 146                     client.setStatus(addTopicToCluster(central, topic, err));
 147                     if (!err.is2xx() && err.getCode() != 409) {
 148                         topic.setFqtn(err.getMessage());
 149                         return null;
 150                     }
 151                 }
 152             }
 153
 154         } else {
 155             logger.warn("Client references a dcaeLocation that doesn't exist:" + client.getDcaeLocationName());
 156             client.setStatus(DmaapObject_Status.STAGED);
 157         }
 158
 159         mr_clients.put(client.getMrClientId(), client);
 160
 161         err.setCode(200);
 162
 163         return client;
 164     }
 165
 166     private DmaapObject_Status addTopicToCluster(MR_Cluster cluster, Topic topic, ApiError err) {
 167
 168         MrProvConnection prov = new MrProvConnection();
 169         logger.info("POST topic " + topic.getFqtn() + " to cluster " + cluster.getFqdn() + " in loc " + cluster.getDcaeLocationName());
 170         if (prov.makeTopicConnection(cluster)) {
 171             prov.doPostTopic(topic, err);
 172             logger.info("response code: " + err.getCode());
 173             if (err.is2xx() || err.getCode() == 409) {
 174                 return DmaapObject_Status.VALID;
 175             }
 176         }
 177         return DmaapObject_Status.INVALID;
 178     }
 179
 180     public MR_Client updateMr_Client(MR_Client client, ApiError apiError) {
 181         MR_Client c = mr_clients.get(client.getMrClientId());
 182         if (c == null) {
 183             apiError.setCode(Status.NOT_FOUND.getStatusCode());
 184             apiError.setFields(MR_CLIENT_ID);
 185             apiError.setMessage("mrClientId " + client.getMrClientId() + " not found");
 186         } else {
 187             apiError.setCode(200);
 188         }
 189         mr_clients.put(client.getMrClientId(), client);
 190         return client;
 191     }
 192
 193     public void removeMr_Client(String key, boolean updateTopicView, ApiError apiError) {
 194         MR_Client client = mr_clients.get(key);
 195         if (client == null) {
 196             apiError.setCode(Status.NOT_FOUND.getStatusCode());
 197             apiError.setFields(MR_CLIENT_ID);
 198             apiError.setMessage("mrClientId " + key + " not found");
 199             return;
 200         } else {
 201             apiError.setCode(200);
 202         }
 203
 204         if (updateTopicView) {
 205
 206             TopicService topics = new TopicService();
 207
 208             Topic t = topics.getTopic(client.getFqtn(), apiError);
 209             if (t != null) {
 210                 List<MR_Client> tc = t.getClients();
 211                 for (MR_Client c : tc) {
 212                     if (c.getMrClientId().equals(client.getMrClientId())) {
 213                         tc.remove(c);
 214                         break;
 215                     }
 216                 }
 217                 t.setClients(tc);
 218                 topics.updateTopic(t, apiError);
 219             }
 220
 221         }
 222
 223         // remove from DB
 224         if (deleteLevel >= 1) {
 225             mr_clients.remove(key);
 226         }
 227     }
 228
 229     private void updateApiError(ApiError err, ApiError permissionServiceError) {
 230         err.setCode(permissionServiceError.getCode());
 231         err.setMessage(permissionServiceError.getMessage());
 232         err.setFields(permissionServiceError.getFields());
 233     }
 234 }