1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.kuali.ole.documenthandler;
17
18 import org.apache.solr.common.SolrDocument;
19 import org.kuali.ole.docstore.model.enums.DocCategory;
20 import org.kuali.ole.docstore.model.enums.DocFormat;
21 import org.kuali.ole.docstore.model.enums.DocType;
22 import org.kuali.ole.docstore.model.xmlpojo.ingest.AdditionalAttributes;
23 import org.kuali.ole.docstore.model.xmlpojo.ingest.Content;
24 import org.kuali.ole.docstore.model.xmlpojo.ingest.RequestDocument;
25 import org.kuali.ole.docstore.model.xmlpojo.work.instance.oleml.*;
26 import org.kuali.ole.docstore.model.xstream.work.instance.oleml.WorkInstanceOlemlRecordProcessor;
27 import org.kuali.ole.docstore.process.ProcessParameters;
28 import org.kuali.ole.docstore.service.DocumentIngester;
29 import org.kuali.ole.docstore.service.ServiceLocator;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32
33 import java.util.ArrayList;
34 import java.util.Collection;
35 import java.util.List;
36
37
38
39
40
41
42
43 public class InstanceRequestDocumentResolver {
44 private static Logger logger = LoggerFactory.getLogger(InstanceRequestDocumentResolver.class);
45
46 private WorkInstanceOlemlRecordProcessor olemlProcessor = new WorkInstanceOlemlRecordProcessor();
47
48
49
50
51
52
53
54
55 public List<RequestDocument> getParsedHoldingsNItemDocuments(RequestDocument instanceDoc,
56 List<String> linkedBibIds) {
57 List<RequestDocument> parsedItemNHoldingsDocuments = new ArrayList<RequestDocument>();
58 if (instanceDoc != null && DocCategory.WORK.isEqualTo(instanceDoc.getCategory())
59 && DocType.INSTANCE.isEqualTo(instanceDoc.getType()) && DocFormat.OLEML
60 .isEqualTo(instanceDoc.getFormat())) {
61 String docContent = instanceDoc.getContent().getContent();
62 InstanceCollection instanceCollection = olemlProcessor.fromXML(docContent);
63 instanceDoc.getContent().setContentObject(instanceCollection);
64
65 if (instanceCollection.getInstance() != null && instanceCollection.getInstance().size() > 0) {
66 Instance instance = instanceCollection.getInstance().get(0);
67 resolveLinkingWithBib(instance);
68
69 if (instance.getResourceIdentifier().size() == 1) {
70 if (instance.getResourceIdentifier().get(0) == null || ""
71 .equals(instance.getResourceIdentifier().get(0))) {
72 instance.getResourceIdentifier().remove(0);
73 }
74 }
75 if (linkedBibIds != null && linkedBibIds.size() > 0) {
76 for (String likedBibId : linkedBibIds) {
77 instance.getResourceIdentifier().add(likedBibId);
78 }
79 }
80 parsedItemNHoldingsDocuments.add(generateInstanceDocument(instance));
81
82 OleHoldings oleHolding = instance.getOleHoldings();
83 RequestDocument rdHol = (RequestDocument) instanceDoc.clone();
84 Content content = new Content();
85 content.setContent(olemlProcessor.toXML(oleHolding));
86 content.setContentObject(oleHolding);
87 rdHol.setContent(content);
88 if (oleHolding.getExtension() != null) {
89 rdHol.setAdditionalAttributes(getFirstAdditionalAttributes(oleHolding.getExtension()));
90 }
91 parsedItemNHoldingsDocuments.add(rdHol);
92
93 SourceHoldings sourceHoldings = instance.getSourceHoldings();
94 RequestDocument rdSrcHol = (RequestDocument) instanceDoc.clone();
95 Content sourceHolContent = new Content();
96 sourceHolContent.setContent(olemlProcessor.toXML(sourceHoldings));
97 sourceHolContent.setContentObject(sourceHoldings);
98 rdSrcHol.setContent(sourceHolContent);
99 if (sourceHoldings != null && sourceHoldings.getExtension() != null) {
100 rdSrcHol.setAdditionalAttributes(getFirstAdditionalAttributes(sourceHoldings.getExtension()));
101 }
102 parsedItemNHoldingsDocuments.add(rdSrcHol);
103
104 for (Item oleItem : instance.getItems().getItem()) {
105 RequestDocument rdItm = (RequestDocument) instanceDoc.clone();
106 Content itemContent = new Content();
107 itemContent.setContent(olemlProcessor.toXML(oleItem));
108 itemContent.setContentObject(oleItem);
109 rdItm.setContent(itemContent);
110 if (oleItem != null && oleItem.getExtension() != null) {
111 rdItm.setAdditionalAttributes(getFirstAdditionalAttributes(oleItem.getExtension()));
112 }
113 parsedItemNHoldingsDocuments.add(rdItm);
114 }
115 }
116 }
117 return parsedItemNHoldingsDocuments;
118 }
119
120 private void resolveLinkingWithBib(Instance instance) {
121 instance.getResourceIdentifier().clear();
122 if (ProcessParameters.BULK_INGEST_IS_LINKING_ENABLED) {
123 for (FormerIdentifier frids : instance.getFormerResourceIdentifier()) {
124 Identifier identifier = frids.getIdentifier();
125 try {
126 if (identifier.getIdentifierValue() != null
127 && identifier.getIdentifierValue().trim().length() != 0) {
128 List<SolrDocument> solrDocs = ServiceLocator.getIndexerService()
129 .getSolrDocument("SystemControlNumber",
130 "\"" + identifier
131 .getIdentifierValue()
132 + "\"");
133 if (solrDocs != null && solrDocs.size() > 0) {
134 for (SolrDocument solrDoc : solrDocs) {
135 if (checkApplicability(identifier.getIdentifierValue(),
136 solrDoc.getFieldValue("SystemControlNumber"))) {
137 instance.getResourceIdentifier().add(solrDoc.getFieldValue("id").toString());
138 }
139 }
140 }
141 }
142 }
143 catch (Exception e) {
144 logger.error("Exception (ignored) while linking instance with bib: ", e);
145 }
146 }
147 }
148 }
149
150 private boolean checkApplicability(Object value, Object fieldValue) {
151 if (fieldValue instanceof Collection) {
152 for (Object object : (Collection) fieldValue) {
153 if (object.equals(value)) {
154 return true;
155 }
156 }
157 return false;
158 }
159 else {
160 return value.equals(fieldValue);
161 }
162 }
163
164 private RequestDocument generateInstanceDocument(Instance instance) {
165 InstanceCollection instanceCollection = new InstanceCollection();
166 List<Instance> instances = new ArrayList<Instance>();
167 instanceCollection.setInstance(instances);
168 Instance inst = new Instance();
169 instances.add(inst);
170 inst.setInstanceIdentifier(instance.getInstanceIdentifier());
171 inst.setResourceIdentifier(instance.getResourceIdentifier());
172 inst.setFormerResourceIdentifier(instance.getFormerResourceIdentifier());
173 inst.setExtension(instance.getExtension());
174 String cont = olemlProcessor.toXML(instanceCollection);
175 RequestDocument requestDocument = new RequestDocument();
176 requestDocument.setCategory(DocCategory.WORK.getCode());
177 requestDocument.setType(DocType.INSTANCE.getCode());
178 requestDocument.setFormat(DocFormat.OLEML.getCode());
179 requestDocument.setContent(new Content());
180 requestDocument.getContent().setContent(cont);
181 requestDocument.getContent().setContentObject(inst);
182 return requestDocument;
183 }
184
185
186
187
188
189
190
191 private AdditionalAttributes getFirstAdditionalAttributes(Extension extension) {
192 if (extension != null && extension.getContent() != null) {
193 for (Object obj : extension.getContent()) {
194 if (obj instanceof AdditionalAttributes) {
195 return (AdditionalAttributes) obj;
196 }
197 }
198 }
199 return null;
200 }
201 }