src/main/java/org/onap/dmaap/dbcapi/service/DR_NodeService.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.onap.dcae
   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 java.util.ArrayList;
  26 import java.util.List;
  27 import java.util.Map;
  28
  29 import javax.ws.rs.core.Response.Status;
  30
  31 import org.apache.log4j.Logger;
  32 import org.onap.dmaap.dbcapi.client.DrProvConnection;
  33 import org.onap.dmaap.dbcapi.database.DatabaseClass;
  34 import org.onap.dmaap.dbcapi.logging.BaseLoggingClass;
  35 import org.onap.dmaap.dbcapi.model.ApiError;
  36 import org.onap.dmaap.dbcapi.model.DR_Node;
  37 import org.onap.dmaap.dbcapi.model.DmaapObject.DmaapObject_Status;
  38 import org.onap.dmaap.dbcapi.util.DmaapConfig;
  39
  40
  41 public class DR_NodeService extends BaseLoggingClass {
  42         private  class DrProv {
  43                 String currentNodes;
  44                 String currentStaticNodes;
  45
  46                 private String getX( String X, ApiError apiError ) {
  47
  48                 logger.info( "templog:getX at" + " 12.10.10" );
  49                         DrProvConnection prov = new DrProvConnection();
  50                 logger.info( "templog:getX at" + " 12.10.12" );
  51                         prov.makeNodesConnection( X );
  52                 logger.info( "templog:getX at" + " 12.10.14" );
  53                         String resp  = prov.doGetNodes(  apiError );
  54                 logger.info( "templog:getX at" + " 12.10.16" );
  55                         logger.info( "rc=" + apiError.getCode() );
  56                 logger.info( "templog:getX at" + " 12.10.18" );
  57                         return resp;
  58                 }
  59
  60                 private void setX( String X, String list, ApiError apiError ) {
  61                         DrProvConnection prov = new DrProvConnection();
  62                         prov.makeNodesConnection( X, list );
  63                         prov.doPutNodes( apiError );
  64                 }
  65
  66                 private String removeFromList( String aNode, String aList ) {
  67                         String[] nodeList = aList.split("\\|");
  68                         StringBuilder res = new StringBuilder();
  69                         for ( String n: nodeList ) {
  70                                 logger.info( "compare existing node " + n + " vs " + aNode );
  71                                 if ( ! n.equals(aNode)) {
  72                                         if (res.length() > 0 ) {
  73                                                 res.append( "|" );
  74                                         }
  75                                         res.append(n);
  76                                 }
  77                         }
  78                         logger.info( "result=" + res.toString() );
  79                         return res.toString();
  80                 }
  81
  82                  boolean containsNode( String aNode , ApiError apiError ){
  83
  84                 logger.info( "templog:containsNode at" + " 12.10" );
  85                         //DrProvConnection prov = new DrProvConnection();
  86                         //prov.makeNodesConnection();
  87                         currentNodes = getX( "NODES", apiError );
  88                 logger.info( "templog:containsNode at" + " 12.12" );
  89                         if ( ! apiError.is2xx() || currentNodes == null ) {
  90                 logger.info( "templog:containsNode at" + " 12.14" );
  91                                 return false;
  92                         }
  93                 logger.info( "templog:containsNode at" + " 12.16" );
  94                         logger.info( "NODES now=" + currentNodes );
  95                         String[] nodeList = currentNodes.split("\\|");
  96                 logger.info( "templog:containsNode at" + " 12.17" );
  97                         for( String n: nodeList ) {
  98                 logger.info( "templog:containsNode at" + " 12.18" );
  99                                 logger.info( "compare existing node " + n + " vs " + aNode );
 100                                 if ( n.equals(aNode) ) {
 101                                         return true;
 102                                 }
 103                         }
 104                         return false;
 105                 }
 106
 107                  void addNode( String aNode, ApiError apiError ) {
 108
 109                         currentNodes = currentNodes + "|" + aNode;
 110                         setX( "NODES", currentNodes, apiError );
 111
 112
 113                 }
 114                 void removeNode( String aNode, ApiError apiError ) {
 115                         currentNodes = removeFromList( aNode, currentNodes );
 116                         setX( "NODES", currentNodes, apiError );
 117                 }
 118
 119                 public boolean containsStaticNode(String aNode, ApiError apiError) {
 120
 121                         //DrProvConnection prov = new DrProvConnection();
 122                         //prov.makeNodesConnection();
 123                         currentStaticNodes = getX( "STATIC_ROUTING_NODES", apiError );
 124                         if (! apiError.is2xx() || currentStaticNodes == null ) {
 125                                 return false;
 126                         }
 127                         logger.info( "STATIC_ROUTING_NODES now=" + currentNodes );
 128                         String[] nodeList = currentStaticNodes.split("\\|");
 129                         for( String n: nodeList ) {
 130                                 logger.info( "compare existing node " + n + " vs " + aNode );
 131                                 if ( n.equals(aNode) ) {
 132                                         return true;
 133                                 }
 134                         }
 135                         return false;
 136                 }
 137
 138
 139                 public void addStaticNode(String aNode, ApiError apiError) {
 140                         currentStaticNodes = currentStaticNodes + "|" + aNode;
 141                         setX( "STATIC_ROUTING_NODES", currentStaticNodes, apiError );
 142                 }
 143                 void removeStaticNode( String aNode, ApiError apiError ) {
 144                         currentStaticNodes = removeFromList( aNode, currentStaticNodes );
 145                         setX( "STATIC_ROUTING_NODES", currentStaticNodes, apiError );
 146                 }
 147         }
 148
 149         DmaapConfig p = (DmaapConfig)DmaapConfig.getConfig();
 150         String unit_test = p.getProperty( "UnitTest", "No" );
 151
 152         private Map<String, DR_Node> dr_nodes = DatabaseClass.getDr_nodes();
 153
 154         public Map<String, DR_Node> getDr_Nodes() {
 155                 return dr_nodes;
 156         }
 157
 158         public List<DR_Node> getAllDr_Nodes() {
 159                 return new ArrayList<DR_Node>(dr_nodes.values());
 160         }
 161
 162         public DR_Node getDr_Node( String fqdn, ApiError apiError ) {
 163                 DR_Node old = dr_nodes.get( fqdn );
 164                 if ( old == null ) {
 165                         apiError.setCode(Status.NOT_FOUND.getStatusCode());
 166                         apiError.setFields( "fqdn");
 167                         apiError.setMessage( "Node " + fqdn + " does not exist");
 168                         return null;
 169                 }
 170                 apiError.setCode(200);
 171                 return old;
 172         }
 173
 174         public DR_Node addDr_Node( DR_Node node, ApiError apiError ) {
 175                 String fqdn = node.getFqdn();
 176                 DR_Node old = dr_nodes.get( fqdn );
 177                 if ( old != null ) {
 178                         apiError.setCode(Status.CONFLICT.getStatusCode());
 179                         apiError.setFields( "fqdn");
 180                         apiError.setMessage( "Node " + fqdn + " already exists");
 181                         return null;
 182                 }
 183                 logger.info( "templog:addDr_Node at" + " 10" );
 184
 185                 DrProv drProv = new DrProv();
 186                 logger.info( "templog:addDr_Node at" + " 12" );
 187
 188                 if ( ! drProv.containsNode( node.getFqdn(), apiError ) && apiError.is2xx() ) {
 189                         logger.info( "templog:addDr_Node at" + " 15" );
 190                         drProv.addNode( node.getFqdn(), apiError );
 191                 }
 192                 logger.info( "templog:addDr_Node at" + " 20" );
 193                 if ( ! apiError.is2xx() && ! unit_test.equals( "Yes" ) ) {
 194                         return null;
 195                 }
 196                 logger.info( "templog:addDr_Node at" + " 30" );
 197                 DcaeLocationService locService = new DcaeLocationService();
 198                 if ( locService.isEdgeLocation( node.getDcaeLocationName()) && ! drProv.containsStaticNode( node.getFqdn(), apiError ) ) {
 199                         if ( apiError.is2xx() ) {
 200                                 drProv.addStaticNode( node.getFqdn(), apiError );
 201                         }
 202                 }
 203                 logger.info( "templog:addDr_Node at" + " 40" );
 204                 if ( ! apiError.is2xx() && ! unit_test.equals("Yes") ) {
 205                         return null;
 206                 }
 207
 208                 logger.info( "templog:addDr_Node at" + " 50" );
 209                 node.setLastMod();
 210                 node.setStatus(DmaapObject_Status.VALID);
 211                 dr_nodes.put( node.getFqdn(), node );
 212                 logger.info( "templog:addDr_Node at" + " 60" );
 213                 apiError.setCode(200);
 214                 return node;
 215         }
 216
 217         public DR_Node updateDr_Node( DR_Node node, ApiError apiError ) {
 218                 DR_Node old = dr_nodes.get( node.getFqdn() );
 219                 if ( old == null ) {
 220                         apiError.setCode(Status.NOT_FOUND.getStatusCode());
 221                         apiError.setFields( "fqdn");
 222                         apiError.setMessage( "Node " + node.getFqdn() + " does not exist");
 223                         return null;
 224                 }
 225                 node.setLastMod();
 226                 dr_nodes.put( node.getFqdn(), node );
 227                 apiError.setCode(200);
 228                 return node;
 229         }
 230
 231         public DR_Node removeDr_Node( String nodeName, ApiError apiError ) {
 232                 DR_Node old = dr_nodes.get( nodeName );
 233                 if ( old == null ) {
 234                         apiError.setCode(Status.NOT_FOUND.getStatusCode());
 235                         apiError.setFields( "fqdn");
 236                         apiError.setMessage( "Node " + nodeName + " does not exist");
 237                         return null;
 238                 }
 239
 240                 DrProv drProv = new DrProv();
 241                 if ( drProv.containsNode( old.getFqdn(), apiError ) && apiError.is2xx() ) {
 242                         drProv.removeNode( old.getFqdn(), apiError );
 243                 }
 244                 DcaeLocationService locService = new DcaeLocationService();
 245                 if ( locService.isEdgeLocation( old.getDcaeLocationName()) && drProv.containsStaticNode( old.getFqdn(), apiError ) ) {
 246                         if ( apiError.is2xx()) {
 247                                 drProv.removeStaticNode( old.getFqdn(), apiError );
 248                         }
 249
 250                 }
 251
 252                 apiError.setCode(200);
 253                 return dr_nodes.remove(nodeName);
 254         }
 255
 256         public String getNodePatternAtLocation( String loc, boolean allowMult ) {
 257                 logger.info( "loc=" + loc );
 258                 if ( loc == null ) {
 259                         return null;
 260                 }
 261                 StringBuilder str = new StringBuilder();
 262                 for( DR_Node node : dr_nodes.values() ) {
 263                         if ( loc.equals( node.getDcaeLocationName()) ) {
 264                                 if ( str.length() > 0 ) {
 265                                         str.append( ",");
 266                                 }
 267                                 str.append( node.getFqdn());
 268                                 if ( ! allowMult ) {
 269                                         break;
 270                                 }
 271                         }
 272                 }
 273                 logger.info( "returning " + str.toString() );
 274                 return str.toString();
 275         }
 276 }