| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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. |
| */ |
| package org.apache.unomi.router.core.processor; |
| |
| import org.apache.camel.Exchange; |
| import org.apache.camel.Processor; |
| import org.apache.unomi.router.api.ImportConfiguration; |
| import org.apache.unomi.router.api.ImportLineError; |
| import org.apache.unomi.router.api.ProfileToImport; |
| import org.apache.unomi.router.api.RouterConstants; |
| import org.apache.unomi.router.api.services.ImportExportConfigurationService; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import java.util.*; |
| |
| /** |
| * Created by amidani on 14/06/2017. |
| */ |
| public class ImportRouteCompletionProcessor implements Processor { |
| |
| private static final Logger logger = LoggerFactory.getLogger(ImportRouteCompletionProcessor.class.getName()); |
| private ImportExportConfigurationService<ImportConfiguration> importConfigurationService; |
| private int executionsHistorySize; |
| |
| @Override |
| public void process(Exchange exchange) throws Exception { |
| String importConfigId = null; |
| ImportConfiguration importConfigOneShot = (ImportConfiguration) exchange.getIn().getHeader(RouterConstants.HEADER_IMPORT_CONFIG_ONESHOT); |
| if (importConfigOneShot != null) { |
| importConfigId = importConfigOneShot.getItemId(); |
| } else { |
| importConfigId = exchange.getFromRouteId(); |
| } |
| ImportConfiguration importConfiguration = importConfigurationService.load(importConfigId); |
| long successCount = 0; |
| long failureCount = 0; |
| long ignoreCount = 0; |
| List<ImportLineError> errors = new ArrayList<ImportLineError>(); |
| |
| for (Object line : exchange.getIn().getBody(ArrayList.class)) { |
| if (line instanceof ProfileToImport) { |
| successCount++; |
| } else if (line instanceof ImportLineError) { |
| failureCount++; |
| errors.add(((ImportLineError) line)); |
| } else { |
| ignoreCount++; |
| } |
| } |
| |
| Map execution = new HashMap(); |
| execution.put("date", ((Date) exchange.getProperty("CamelCreatedTimestamp")).getTime()); |
| execution.put("totalLinesNb", exchange.getProperty("CamelSplitSize")); |
| execution.put("successCount", successCount); |
| execution.put("failureCount", failureCount); |
| execution.put("errors", errors); |
| |
| if (importConfiguration.getExecutions().size() >= executionsHistorySize) { |
| int oldestExecIndex = 0; |
| long oldestExecDate = (Long) importConfiguration.getExecutions().get(0).get("date"); |
| for (int i = 1; i < importConfiguration.getExecutions().size(); i++) { |
| if ((Long) importConfiguration.getExecutions().get(i).get("date") < oldestExecDate) { |
| oldestExecDate = (Long) importConfiguration.getExecutions().get(i).get("date"); |
| oldestExecIndex = i; |
| } |
| } |
| importConfiguration.getExecutions().remove(oldestExecIndex); |
| } |
| |
| importConfiguration.getExecutions().add(execution); |
| //Set running to false, route is complete |
| if (failureCount > 0 && successCount > 0) { |
| importConfiguration.setStatus(RouterConstants.CONFIG_STATUS_COMPLETE_WITH_ERRORS); |
| } else if (failureCount > 0 && successCount == 0) { |
| importConfiguration.setStatus(RouterConstants.CONFIG_STATUS_COMPLETE_ERRORS); |
| } else if (failureCount == 0 && successCount > 0) { |
| importConfiguration.setStatus(RouterConstants.CONFIG_STATUS_COMPLETE_SUCCESS); |
| } |
| importConfigurationService.save(importConfiguration); |
| logger.info("Processing route {} completed.", exchange.getFromRouteId()); |
| } |
| |
| public void setImportConfigurationService(ImportExportConfigurationService<ImportConfiguration> importConfigurationService) { |
| this.importConfigurationService = importConfigurationService; |
| } |
| |
| public void setExecutionsHistorySize(int executionsHistorySize) { |
| this.executionsHistorySize = executionsHistorySize; |
| } |
| } |