appc-directed-graph/dg-loader/provider/src/main/java/org/openecomp/sdnc/dg/loader/DGXMLLoadNActivate.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APP-C
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property.  All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdnc.dg.loader;
  22
  23 import java.io.File;
  24 import java.util.ArrayList;
  25 import java.util.List;
  26
  27 import org.apache.commons.io.FileUtils;
  28 import org.apache.commons.lang.StringUtils;
  29 import org.slf4j.Logger;
  30 import org.slf4j.LoggerFactory;
  31
  32 import org.openecomp.sdnc.sli.SvcLogicException;
  33 import org.openecomp.sdnc.sli.SvcLogicGraph;
  34 import org.openecomp.sdnc.sli.SvcLogicParser;
  35 import org.openecomp.sdnc.sli.SvcLogicStore;
  36 import org.openecomp.sdnc.sli.SvcLogicStoreFactory;
  37
  38 public class DGXMLLoadNActivate {
  39         private final static Logger logger = LoggerFactory.getLogger(DGXMLLoadNActivate.class);
  40         private final SvcLogicStore store;
  41         public static String STRING_ENCODING = "utf-8";
  42
  43         public DGXMLLoadNActivate(String propfile) throws Exception{
  44                 if(StringUtils.isBlank(propfile)){
  45                         throw new Exception(propfile + " Profile file is not defined");
  46                 }
  47                 this.store = SvcLogicStoreFactory.getSvcLogicStore(propfile);
  48         }
  49
  50         public void loadDGXMLFile(String dgXMLpath) throws SvcLogicException{
  51                 if(dgXMLpath != null ){
  52                         SvcLogicParser.load(dgXMLpath, this.store);
  53                 }
  54         }
  55
  56         private void loadDGXMLDir(String xmlPath) throws Exception {
  57                 try {
  58                         logger.info("******************** Loading DG into Database *****************************");
  59                         List<String> errors = new ArrayList<String>();
  60                         if(this.store != null){
  61                                 File xmlDir = new File(xmlPath);
  62                                 if(xmlDir != null && xmlDir.isDirectory()){
  63                                         String[] extensions = new String[] { "xml", "XML" };
  64                                         List<File> files = (List<File>) FileUtils.listFiles(xmlDir, extensions, true);
  65                                         for (File file : files) {
  66                                                 logger.info("Loading DG XML file :" + file.getCanonicalPath());
  67                                                 try{
  68                                                         SvcLogicParser.load(file.getCanonicalPath(), this.store);
  69                                                 }catch (Exception e) {
  70                                                         errors.add("Failed to load XML "+file.getCanonicalPath() + ", Exception : "+e.getMessage());
  71                                                 }
  72                                         }
  73                                 }else{
  74                                         throw new Exception(xmlPath + " is not a valid XML Directory");
  75                                 }
  76                         }else{
  77                                 throw new Exception("Failed to initialise SvcLogicStore");
  78                         }
  79
  80                         if(errors.size() > 0){
  81                                 throw new Exception(errors.toString());
  82                         }
  83                 } catch (Exception e) {
  84                         logger.error(e.getMessage());
  85                 }
  86         }
  87
  88         public void activateDg(String activateFilePath) throws Exception {
  89                 logger.info("******************** Activating DG into Database *****************************");
  90                 try {
  91                         List<String> errors = new ArrayList<String>();
  92                         if(this.store != null){
  93                                 File activateFile = new File(activateFilePath);
  94                                 if(activateFile != null && activateFile.isFile()){
  95                                         List<String> fileLines = FileUtils.readLines(activateFile,STRING_ENCODING);
  96                                         if(fileLines != null ){
  97                                                 for (String line : fileLines) {
  98                                                         if(line != null && ! line.trim().startsWith("#")){
  99                                                                 String lineArray[] = line.trim().split(":");
 100                                                                 try {
 101                                                                         if(lineArray != null && lineArray.length >= 4){
 102                                                                                 String module = lineArray[0];
 103                                                                                 String rpc = lineArray[1];
 104                                                                                 String version = lineArray[2];
 105                                                                                 String mode = lineArray[3];
 106                                                                                 if(StringUtils.isNotBlank(module) && StringUtils.isNotBlank(rpc)
 107                                                                                                 && StringUtils.isNotBlank(version) && StringUtils.isNotBlank(mode)){
 108                                                                                         logger.info("Activating DG :" + line);
 109                                                                                         SvcLogicGraph graph = this.store.fetch(module, rpc, version, mode);
 110                                                                                         if(graph != null){
 111                                                                                                 logger.info("Found Graph :" + line + " Activating ...");
 112                                                                                                 this.store.activate(graph);
 113                                                                                         }else{
 114                                                                                                 throw new Exception("Failed to fetch from Database");
 115                                                                                         }
 116                                                                                 }
 117                                                                         }
 118                                                                 } catch (Exception e) {
 119                                                                         e.printStackTrace();
 120                                                                         errors.add("Failed to Activate "+line + ", "+e.getMessage());
 121                                                                 }
 122                                                         }
 123                                                 }
 124                                         }
 125                                 }else{
 126                                         throw new Exception(activateFile + " is not a valid Activate file Path");
 127                                 }
 128                         }else{
 129                                 throw new Exception("Failed to initialise SvcLogicStore");
 130                         }
 131
 132                         if(errors.size() > 0){
 133                                 throw new Exception(errors.toString());
 134                         }
 135                 } catch (Exception e) {
 136                         logger.error(e.getMessage());
 137                 }
 138         }
 139
 140
 141         public static void main(String[] args) {
 142                 try {
 143                         String xmlPath = null;
 144                         String propertyPath = null;
 145                         String activateFile = null;
 146
 147                         if(args != null && args.length >= 3){
 148                                 xmlPath = args[0];
 149                                 activateFile = args[1];
 150                                 propertyPath = args[2];
 151                         }else{
 152                                 throw new Exception("Sufficient inputs for DGXMLLoadNActivate are missing <xmlpath> <activatefile> <dbPropertyfile>");
 153                         }
 154
 155                         //propertyPath = "/Users/bs2796/0Source/ecomp/bvc-3.2.2/others/properties/dblib.properties";
 156                         //xmlPath = DGXMLLoadNActivate.class.getClassLoader().getResource(".").getPath() +"/xml" ;
 157
 158                         DGXMLLoadNActivate dgXMLLoadDB = new DGXMLLoadNActivate(propertyPath);
 159                         dgXMLLoadDB.loadDGXMLDir(xmlPath);
 160                         dgXMLLoadDB.activateDg(activateFile);
 161                 } catch (Exception e) {
 162                         e.printStackTrace();
 163                 }finally {
 164                         System.exit(1);
 165                 }
 166         }
 167
 168 }