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

   1 package org.collectionspace.services.imports.nuxeo;\r
   2 \r
   3 import java.io.File;\r
   4 import java.util.Calendar;\r
   5 import java.util.HashMap;\r
   6 import java.util.Map;\r
   7 import java.util.TreeSet;\r
   8 import org.apache.commons.logging.Log;\r
   9 import org.apache.commons.logging.LogFactory;\r
  10 import org.collectionspace.services.nuxeo.client.java.NuxeoClientEmbedded;\r
  11 import org.collectionspace.services.nuxeo.client.java.NuxeoConnectorEmbedded;\r
  12 import org.nuxeo.ecm.core.api.DocumentModel;\r
  13 import org.nuxeo.ecm.core.api.DocumentRef;\r
  14 import org.nuxeo.ecm.core.api.repository.RepositoryInstance;\r
  15 import org.nuxeo.ecm.core.io.DocumentPipe;\r
  16 import org.nuxeo.ecm.core.io.DocumentReader;\r
  17 import org.nuxeo.ecm.core.io.DocumentTranslationMap;\r
  18 import org.nuxeo.ecm.core.io.DocumentWriter;\r
  19 import org.nuxeo.ecm.core.io.impl.DocumentPipeImpl;\r
  20 import org.nuxeo.ecm.core.io.impl.plugins.DocumentModelWriter;\r
  21 // we use our own override of this: import org.nuxeo.ecm.core.io.impl.plugins.XMLDirectoryReader;\r
  22 \r
  23 // based loosely on package org.nuxeo.ecm.shell.commands.io.ImportCommand;\r
  24 public class ImportCommand {\r
  25     private static final Log logger = LogFactory.getLog(ImportCommand.class);\r
  26 \r
  27     public String run(String src, String dest, int timeOut) throws Exception {\r
  28         File file = new File(src);\r
  29         ///cspace way of configuring client and auth:\r
  30         NuxeoClientEmbedded client = NuxeoConnectorEmbedded.getInstance().getClient();\r
  31         RepositoryInstance  repoSession = client.openRepository(timeOut);\r
  32         try {\r
  33                 String msg = String.format("Start of import is Local time: %tT", Calendar.getInstance());\r
  34                 logger.debug(msg);\r
  35                 System.out.println(msg);\r
  36             return importTree(repoSession, file, dest);\r
  37         } catch (Exception e) {\r
  38             throw e;\r
  39         } finally {\r
  40                 String msg = String.format("End of import is Local time: %tT", Calendar.getInstance());\r
  41                 logger.debug(msg);\r
  42                 System.out.println(msg);\r
  43             client.releaseRepository(repoSession);\r
  44         }\r
  45     }\r
  46 \r
  47     String importTree(RepositoryInstance repoSession, File file, String toPath) throws Exception {\r
  48         DocumentReader reader = null;\r
  49         DocumentWriter writer = null;\r
  50         DocumentModel docModel = null;\r
  51         DocumentRef keyDocRef, valueDocRef;\r
  52         String docType;\r
  53         StringBuffer dump = new StringBuffer();\r
  54         Map<String,Integer> recordsImportedForDocType = new HashMap<String,Integer>();\r
  55         Integer numRecordsImportedForDocType = new Integer(0);\r
  56         int totalRecordsImported = 0;\r
  57         try {\r
  58             System.out.println("importTree reading file: "+file+(file!=null ? " exists? "+file.exists() : " file param is null"));\r
  59             reader = new LoggedXMLDirectoryReader(file);  //our overload of XMLDirectoryReader.\r
  60             writer = new DocumentModelWriter(repoSession, toPath, 10);\r
  61             DocumentPipe pipe = new DocumentPipeImpl(10);\r
  62             // pipe.addTransformer(transformer);\r
  63             pipe.setReader(reader);\r
  64             pipe.setWriter(writer);\r
  65             DocumentTranslationMap dtm = pipe.run();\r
  66             Map<DocumentRef,DocumentRef> documentRefs = dtm.getDocRefMap();\r
  67             dump.append("<importedRecords>");\r
  68             for (Map.Entry entry: documentRefs.entrySet()) {\r
  69                 keyDocRef = (DocumentRef) entry.getKey();\r
  70                 valueDocRef = (DocumentRef) entry.getValue();\r
  71                 if (keyDocRef == null || valueDocRef == null) {\r
  72                     continue;\r
  73                 }\r
  74                 dump.append("<importedRecord>");\r
  75                 docModel = repoSession.getDocument(valueDocRef);\r
  76                 docType = docModel.getDocumentType().getName();\r
  77                 dump.append("<doctype>"+docType+"</doctype>");\r
  78                 dump.append("<csid>"+keyDocRef.toString()+"</csid>");\r
  79                 dump.append("</importedRecord>");\r
  80                 if (recordsImportedForDocType.containsKey(docType)) {\r
  81                     numRecordsImportedForDocType = (Integer) recordsImportedForDocType.get(docType);\r
  82                     numRecordsImportedForDocType = Integer.valueOf(numRecordsImportedForDocType.intValue() + 1);\r
  83                     recordsImportedForDocType.put(docType, numRecordsImportedForDocType);\r
  84                 } else {\r
  85                     recordsImportedForDocType.put(docType, 1);\r
  86                 }\r
  87                 totalRecordsImported++;\r
  88             }\r
  89             dump.append("</importedRecords>");\r
  90         } catch (Exception e) {\r
  91             throw e;\r
  92         } finally {\r
  93             dump.append("<totalRecordsImported>"+totalRecordsImported+"</totalRecordsImported>");\r
  94             dump.append("<numRecordsImportedByDocType>");\r
  95             TreeSet<String> keys = new TreeSet<String>(recordsImportedForDocType.keySet());\r
  96             for (String key : keys) {\r
  97                 dump.append("<numRecordsImported>");\r
  98                 dump.append("<docType>"+key+"</docType>");\r
  99                 dump.append("<numRecords>"+recordsImportedForDocType.get(key).intValue()+"</numRecords>");\r
 100                 dump.append("</numRecordsImported>");\r
 101             }\r
 102             dump.append("</numRecordsImportedByDocType>");\r
 103             if (reader != null) {\r
 104                 dump.append("<report>"+(((LoggedXMLDirectoryReader)reader).report())+"</report>");\r
 105                 reader.close();\r
 106             }\r
 107             if (writer != null) {\r
 108                 writer.close();\r
 109             }\r
 110         }\r
 111         return dump.toString();\r
 112     }\r
 113 }