git.aero2k.de Git - tmp/jakarta-migration.git/blob

   1 package org.collectionspace.services.imports.nuxeo;
   2
   3 import java.io.File;
   4 import java.util.Calendar;
   5 import java.util.HashMap;
   6 import java.util.Map;
   7 import java.util.TreeSet;
   8
   9 import org.collectionspace.services.nuxeo.client.java.NuxeoClientEmbedded;
  10 import org.collectionspace.services.nuxeo.client.java.NuxeoConnectorEmbedded;
  11 import org.collectionspace.services.nuxeo.client.java.CoreSessionInterface;
  12
  13 import org.nuxeo.ecm.core.api.DocumentModel;
  14 import org.nuxeo.ecm.core.api.DocumentRef;
  15 import org.nuxeo.ecm.core.io.DocumentPipe;
  16 import org.nuxeo.ecm.core.io.DocumentReader;
  17 import org.nuxeo.ecm.core.io.DocumentTranslationMap;
  18 import org.nuxeo.ecm.core.io.DocumentWriter;
  19 import org.nuxeo.ecm.core.io.impl.DocumentPipeImpl;
  20 import org.nuxeo.ecm.core.io.impl.plugins.DocumentModelWriter;
  21 // we use our own override of this: import org.nuxeo.ecm.core.io.impl.plugins.XMLDirectoryReader;
  22 import org.slf4j.Logger;
  23 import org.slf4j.LoggerFactory;
  24
  25 // based loosely on package org.nuxeo.ecm.shell.commands.io.ImportCommand;
  26 public class ImportCommand {
  27
  28     private final Logger logger = LoggerFactory.getLogger(ImportCommand.class);
  29
  30     public String run(String src, String repoName, String workspacesPath, int timeout) throws Exception {
  31         File file = new File(src);
  32         ///cspace way of configuring client and auth:
  33         NuxeoClientEmbedded client = NuxeoConnectorEmbedded.getInstance().getClient();
  34         CoreSessionInterface repoSession = null;
  35         try {
  36             repoSession = client.openRepository(repoName, timeout);
  37             if (logger.isDebugEnabled()) {
  38                 String msg = String.format("Start of import is Local time: %tT", Calendar.getInstance());
  39                 logger.debug(msg);
  40             }
  41             return importTree(repoSession, file, workspacesPath, timeout);
  42         } catch (Exception e) {
  43             throw e;
  44         } finally {
  45             if (logger.isDebugEnabled()) {
  46                 String msg = String.format("End of import is Local time: %tT", Calendar.getInstance());
  47                 logger.debug(msg);
  48             }
  49             client.releaseRepository(repoSession);
  50         }
  51     }
  52
  53     /*
  54      * If the import exceeds the number of seconds in 'timeout', we'll thrown an exception and rollback all import work
  55      */
  56     String importTree(CoreSessionInterface repoSession, File file, String toPath, int timeout) throws Exception {
  57         Exception failed = null;
  58         DocumentReader reader = null;
  59         DocumentWriter writer = null;
  60         DocumentModel docModel = null;
  61         DocumentRef keyDocRef, valueDocRef;
  62         String docType;
  63         StringBuffer dump = new StringBuffer();
  64         Map<String, Integer> recordsImportedForDocType = new HashMap<String, Integer>();
  65         Integer numRecordsImportedForDocType = new Integer(0);
  66         int totalRecordsImported = 0;
  67         try {
  68             if (logger.isInfoEnabled()) {
  69                 logger.info("ImportCommand.importTree() method reading file: " + file + (file != null ? " exists? " + file.exists() : " file param is null"));
  70                 logger.info(String.format("ImportCommand.importTree() will timeout if import does not complete in %d seconds.", timeout));
  71             }
  72             reader = new LoggedXMLDirectoryReader(file, timeout);  //our overload of XMLDirectoryReader.
  73             writer = new DocumentModelWriter(repoSession.getCoreSession(), toPath, 10);
  74             DocumentPipe pipe = new DocumentPipeImpl(10);
  75             // pipe.addTransformer(transformer);
  76             pipe.setReader(reader);
  77             pipe.setWriter(writer);
  78             DocumentTranslationMap dtm = pipe.run();
  79             Map<DocumentRef, DocumentRef> documentRefs = dtm.getDocRefMap(); // FIXME: Should be checking for null here!
  80             if (documentRefs != null && documentRefs.isEmpty()) {
  81                 throw new Exception("No valid records found in import payload. Check XML markup for not-well-formed errors, elements not matching import or document-specific schema, etc.");
  82             }
  83             dump.append("<importedRecords>");
  84             for (Map.Entry<DocumentRef, DocumentRef> entry : documentRefs.entrySet()) {
  85                 keyDocRef = (DocumentRef) entry.getKey();
  86                 valueDocRef = (DocumentRef) entry.getValue();
  87                 if (keyDocRef == null || valueDocRef == null) {
  88                     continue;
  89                 }
  90                 dump.append("<importedRecord>");
  91                 docModel = repoSession.getDocument(valueDocRef);
  92                 docType = docModel.getDocumentType().getName();
  93                 dump.append("<doctype>" + docType + "</doctype>");
  94                 dump.append("<csid>" + keyDocRef.toString() + "</csid>");
  95                 dump.append("</importedRecord>");
  96                 if (recordsImportedForDocType.containsKey(docType)) {
  97                     numRecordsImportedForDocType = (Integer) recordsImportedForDocType.get(docType);
  98                     numRecordsImportedForDocType = Integer.valueOf(numRecordsImportedForDocType.intValue() + 1);
  99                     recordsImportedForDocType.put(docType, numRecordsImportedForDocType);
 100                 } else {
 101                     recordsImportedForDocType.put(docType, 1);
 102                 }
 103                 totalRecordsImported++;
 104             }
 105             dump.append("</importedRecords>");
 106         } catch (Exception e) {
 107             failed = e;
 108             throw failed;
 109         } finally {
 110             String status = failed == null ? "Success" : "Failed";
 111             dump.append("<status>" + status + "</status>");
 112             dump.append("<totalRecordsImported>" + totalRecordsImported + "</totalRecordsImported>");
 113             dump.append("<numRecordsImportedByDocType>");
 114             TreeSet<String> keys = new TreeSet<String>(recordsImportedForDocType.keySet());
 115             for (String key : keys) {
 116                 dump.append("<numRecordsImported>");
 117                 dump.append("<docType>" + key + "</docType>");
 118                 dump.append("<numRecords>" + recordsImportedForDocType.get(key).intValue() + "</numRecords>");
 119                 dump.append("</numRecordsImported>");
 120             }
 121             dump.append("</numRecordsImportedByDocType>");
 122             if (reader != null) {
 123                 dump.append("<report>" + (((LoggedXMLDirectoryReader) reader).report()) + "</report>");
 124                 reader.close();
 125             }
 126             if (writer != null) {
 127                 writer.close();
 128             }
 129
 130             if (failed != null) {
 131                 String msg = "The Import service encountered an exception: " + failed.getLocalizedMessage();
 132                 logger.error(msg, failed);
 133             }
 134         }
 135         return dump.toString();
 136     }
 137 }