/*-
* ============LICENSE_START=======================================================
* SDC
* ================================================================================
* Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
* ================================================================================
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* ============LICENSE_END=========================================================
*/
package org.openecomp.core.validation.types;
import org.apache.commons.collections4.CollectionUtils;
import org.openecomp.sdc.logging.api.Logger;
import org.openecomp.sdc.logging.api.LoggerFactory;
import org.openecomp.sdc.datatypes.error.ErrorLevel;
import org.openecomp.sdc.logging.context.MdcUtil;
import org.openecomp.sdc.logging.types.LoggerConstants;
import org.openecomp.sdc.logging.types.LoggerErrorCode;
import java.io.InputStream;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import java.util.function.BiPredicate;
import java.util.stream.Collectors;
public class GlobalValidationContext {
private static Logger logger = (Logger) LoggerFactory.getLogger(GlobalValidationContext.class);
private Map<String, FileValidationContext> fileContextMap = new HashMap<>();
private Map<String, MessageContainer> messageContainerMap = new HashMap<>();
/**
* Add message.
*
* @param fileName the file name
* @param level the level
* @param message the message
* @param targetServiceName the target service name
* @param errorDescription the error details
*/
public void addMessage(String fileName, ErrorLevel level, String message,
String targetServiceName, String errorDescription) {
printLog(fileName, message, level, targetServiceName, errorDescription);
if (fileContextMap.containsKey(fileName)) {
fileContextMap.get(fileName).getMessageContainer().getMessageBuilder()
.setMessage(level.toString() + ": " + message).setLevel(level).create();
} else {
// if (CommonMethods.isEmpty(fileName)) {
// fileName = SdcCommon.UPLOAD_FILE;
// }
MessageContainer messageContainer;
synchronized (this) {
messageContainer = messageContainerMap.get(fileName);
if (messageContainer == null) {
messageContainer = new MessageContainer();
messageContainerMap.put(fileName, messageContainer);
}
}
messageContainer.getMessageBuilder().setMessage(level.toString() + ": " + message)
.setLevel(level).create();
}
}
/**
* Gets file content.
*
* @param fileName the file name
* @return the file content
*/
public Optional<InputStream> getFileContent(String fileName) {
FileValidationContext fileContext = fileContextMap.get(fileName);
if (fileContext == null || fileContext.isEmpty()) {
return Optional.empty();
}
return Optional.of(fileContext.getContent());
}
public void addFileContext(String fileName, byte[] fileContent) {
fileContextMap.put(fileName, new FileValidationContext(fileName, fileContent));
}
/**
* Gets context message containers.
*
* @return the context message containers
*/
public Map<String, MessageContainer> getContextMessageContainers() {
Map<String, MessageContainer> contextMessageContainer = new HashMap<>();
fileContextMap.entrySet().stream().filter(entry -> CollectionUtils
.isNotEmpty(entry.getValue().getMessageContainer().getErrorMessageList())).forEach(
entry -> contextMessageContainer.put(
entry.getKey(), entry.getValue()
.getMessageContainer()));
messageContainerMap.entrySet().stream()
.filter(entry -> CollectionUtils.isNotEmpty(entry.getValue().getErrorMessageList()))
.forEach(entry -> contextMessageContainer.put(entry.getKey(), entry.getValue()));
return contextMessageContainer;
}
public Map<String, FileValidationContext> getFileContextMap() {
return fileContextMap;
}
private void printLog(String fileName, String message, ErrorLevel level, String targetServiceName,
String errorDescription) {
String messageToPrint = message + " in file[" + fileName + "]";
MdcUtil.setValuesForMdc(LoggerConstants.TARGET_ENTITY_API, targetServiceName, level.name(),
LoggerErrorCode.DATA_ERROR.getErrorCode(), errorDescription);
switch (level) {
case ERROR:
logger.error(messageToPrint);
break;
case WARNING:
logger.warn(messageToPrint);
break;
case INFO:
logger.info(messageToPrint);
break;
default:
break;
}
}
public Collection<String> files(BiPredicate<String, GlobalValidationContext> func) {
return fileContextMap.keySet().stream().filter(t -> func.test(t, this))
.collect(Collectors.toList());
}
public Collection<String> getFiles() {
return this.getFileContextMap().keySet();
}
}