001/* 002 * (C) Copyright 2012 Nuxeo SA (http://nuxeo.com/) and others. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 * 016 * Contributors: 017 * Thomas Roger 018 */ 019 020package org.nuxeo.ecm.csv; 021 022import java.io.File; 023import java.util.ArrayList; 024import java.util.Arrays; 025import java.util.Collections; 026import java.util.List; 027 028import org.nuxeo.ecm.core.api.CoreSession; 029import org.nuxeo.ecm.core.work.api.Work; 030import org.nuxeo.ecm.core.work.api.Work.State; 031import org.nuxeo.ecm.core.work.api.WorkManager; 032import org.nuxeo.runtime.api.Framework; 033 034/** 035 * @since 5.7 036 */ 037public class CSVImporterImpl implements CSVImporter { 038 039 @Override 040 public String launchImport(CoreSession session, String parentPath, File csvFile, String csvFileName, 041 CSVImporterOptions options) { 042 CSVImporterWork work = new CSVImporterWork(session.getRepositoryName(), parentPath, session.getPrincipal() 043 .getName(), csvFile, 044 csvFileName, options); 045 WorkManager workManager = Framework.getLocalService(WorkManager.class); 046 workManager.schedule(work, WorkManager.Scheduling.IF_NOT_RUNNING_OR_SCHEDULED); 047 return work.getId(); 048 } 049 050 @Override 051 public CSVImportStatus getImportStatus(String id) { 052 WorkManager workManager = Framework.getLocalService(WorkManager.class); 053 State state = workManager.getWorkState(id); 054 if (state == null) { 055 return null; 056 } else if (state == State.COMPLETED) { 057 return new CSVImportStatus(CSVImportStatus.State.COMPLETED); 058 } else if (state == State.SCHEDULED) { 059 String queueId = workManager.getCategoryQueueId(CSVImporterWork.CATEGORY_CSV_IMPORTER); 060 int queueSize = workManager.getQueueSize(queueId, State.SCHEDULED); 061 return new CSVImportStatus(CSVImportStatus.State.SCHEDULED, 0, queueSize); 062 } else { // RUNNING 063 return new CSVImportStatus(CSVImportStatus.State.RUNNING); 064 } 065 } 066 067 @Override 068 public List<CSVImportLog> getImportLogs(String id) { 069 return getLastImportLogs(id, -1); 070 } 071 072 @Override 073 public List<CSVImportLog> getImportLogs(String id, CSVImportLog.Status... status) { 074 return getLastImportLogs(id, -1, status); 075 } 076 077 @Override 078 public List<CSVImportLog> getLastImportLogs(String id, int max) { 079 WorkManager workManager = Framework.getLocalService(WorkManager.class); 080 Work work = workManager.find(id, null); 081 if (work == null) { 082 work = workManager.find(id, State.COMPLETED); 083 if (work == null) { 084 return Collections.emptyList(); 085 } 086 } 087 List<CSVImportLog> importLogs = ((CSVImporterWork) work).getImportLogs(); 088 max = (max == -1 || max > importLogs.size()) ? importLogs.size() : max; 089 return importLogs.subList(importLogs.size() - max, importLogs.size()); 090 } 091 092 @Override 093 public List<CSVImportLog> getLastImportLogs(String id, int max, CSVImportLog.Status... status) { 094 List<CSVImportLog> importLogs = getLastImportLogs(id, max); 095 return status.length == 0 ? importLogs : filterImportLogs(importLogs, status); 096 } 097 098 protected List<CSVImportLog> filterImportLogs(List<CSVImportLog> importLogs, CSVImportLog.Status... status) { 099 List<CSVImportLog.Status> statusList = Arrays.asList(status); 100 List<CSVImportLog> filteredLogs = new ArrayList<CSVImportLog>(); 101 for (CSVImportLog log : importLogs) { 102 if (statusList.contains(log.getStatus())) { 103 filteredLogs.add(log); 104 } 105 } 106 return filteredLogs; 107 } 108 109 @Override 110 public CSVImportResult getImportResult(String id) { 111 WorkManager workManager = Framework.getLocalService(WorkManager.class); 112 Work work = workManager.find(id, State.COMPLETED); 113 if (work == null) { 114 return null; 115 } 116 117 List<CSVImportLog> importLogs = ((CSVImporterWork) work).getImportLogs(); 118 return CSVImportResult.fromImportLogs(importLogs); 119 } 120 121}