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   private Map<String, byte[]> files = new HashMap<>();
  33
  34   /**
  35    * Gets file content.
  36    *
  37    * @param fileName the file name
  38    * @return the file content
  39    */
  40   public InputStream getFileContent(String fileName) {
  41
  42     byte[] content = files.get(fileName);
  43     if (content == null || content.length == 0) {
  44       return null;
  45     }
  46
  47     ByteArrayInputStream is = new ByteArrayInputStream(content);
  48     return is;
  49   }
  50
  51   public void addFile(String fileName, byte[] contect) {
  52     files.put(fileName, contect);
  53   }
  54
  55   public void addFile(String fileName, InputStream is) {
  56
  57     files.put(fileName, FileUtils.toByteArray(is));
  58   }
  59
  60   public void setFiles(FileContentHandler extFiles) {
  61     extFiles.getFileList().stream()
  62         .forEach(fileName -> this.addFile(fileName, extFiles.getFileContent(fileName)));
  63   }
  64
  65   public Set<String> getFileList() {
  66     return files.keySet();
  67   }
  68
  69   public void putAll(Map<String, byte[]> files) {
  70     this.files = files;
  71   }
  72
  73   public void addAll(FileContentHandler other) {
  74     this.files.putAll(other.files);
  75   }
  76
  77   public boolean isEmpty() {
  78     return MapUtils.isEmpty(this.files);
  79   }
  80
  81   public void remove(String fileName) {
  82     files.remove(fileName);
  83   }
  84
  85   public boolean containsFile(String fileName) {
  86     return files.containsKey(fileName);
  87   }
  88 }