openecomp-be/lib/openecomp-core-lib/openecomp-utilities-lib/src/main/java/org/openecomp/core/utilities/file/FileContentHandler.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   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.core.utilities.file;
  22
  23 import org.apache.commons.collections4.MapUtils;
  24
  25 import java.io.ByteArrayInputStream;
  26 import java.io.InputStream;
  27 import java.util.HashMap;
  28 import java.util.Map;
  29 import java.util.Set;
  30
  31 public class FileContentHandler {
  32
  33   private Map<String, byte[]> files = new HashMap<>();
  34
  35   /**
  36    * Gets file content.
  37    *
  38    * @param fileName the file name
  39    * @return the file content
  40    */
  41   public InputStream getFileContent(String fileName) {
  42
  43     byte[] content = files.get(fileName);
  44     if (content == null || content.length == 0) {
  45       return null;
  46     }
  47
  48     ByteArrayInputStream is = new ByteArrayInputStream(content);
  49     return is;
  50   }
  51
  52   public void addFile(String fileName, byte[] contect) {
  53     files.put(fileName, contect);
  54   }
  55
  56   public void addFile(String fileName, InputStream is) {
  57
  58     files.put(fileName, FileUtils.toByteArray(is));
  59   }
  60
  61   public Map<String, byte[]> getFiles() {
  62     return files;
  63   }
  64
  65   public void setFiles(FileContentHandler extFiles) {
  66     extFiles.getFileList().stream()
  67         .forEach(fileName -> this.addFile(fileName, extFiles.getFileContent(fileName)));
  68   }
  69
  70   public Set<String> getFileList() {
  71     return files.keySet();
  72   }
  73
  74   public void putAll(Map<String, byte[]> files) {
  75     this.files = files;
  76   }
  77
  78   public void addAll(FileContentHandler other) {
  79     this.files.putAll(other.files);
  80   }
  81
  82   public boolean isEmpty() {
  83     return MapUtils.isEmpty(this.files);
  84   }
  85
  86   public void remove(String fileName) {
  87     files.remove(fileName);
  88   }
  89
  90   public boolean containsFile(String fileName) {
  91     return files.containsKey(fileName);
  92   }
  93 }