2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.openecomp.core.utilities.file;
23 import org.apache.commons.collections4.MapUtils;
25 import java.io.ByteArrayInputStream;
26 import java.io.InputStream;
27 import java.util.HashMap;
31 public class FileContentHandler {
33 private Map<String, byte[]> files = new HashMap<>();
38 * @param fileName the file name
39 * @return the file content
41 public InputStream getFileContent(String fileName) {
43 byte[] content = files.get(fileName);
44 if (content == null || content.length == 0) {
48 ByteArrayInputStream is = new ByteArrayInputStream(content);
52 public void addFile(String fileName, byte[] contect) {
53 files.put(fileName, contect);
56 public void addFile(String fileName, InputStream is) {
58 files.put(fileName, FileUtils.toByteArray(is));
61 public Map<String, byte[]> getFiles() {
65 public void setFiles(FileContentHandler extFiles) {
66 extFiles.getFileList().stream()
67 .forEach(fileName -> this.addFile(fileName, extFiles.getFileContent(fileName)));
70 public Set<String> getFileList() {
71 return files.keySet();
74 public void putAll(Map<String, byte[]> files) {
78 public void addAll(FileContentHandler other) {
79 this.files.putAll(other.files);
82 public boolean isEmpty() {
83 return MapUtils.isEmpty(this.files);
86 public void remove(String fileName) {
87 files.remove(fileName);
90 public boolean containsFile(String fileName) {
91 return files.containsKey(fileName);