use of org.codice.ddf.spatial.ogc.csw.catalog.common.transaction.InsertAction in project ddf by codice.
the class TransactionRequestConverter method marshal.
@Override
public void marshal(Object o, HierarchicalStreamWriter writer, MarshallingContext marshallingContext) {
if (o == null || !CswTransactionRequest.class.isAssignableFrom(o.getClass())) {
return;
}
CswTransactionRequest request = (CswTransactionRequest) o;
writer.addAttribute(CswConstants.SERVICE, request.getService());
writer.addAttribute(CswConstants.VERSION, request.getVersion());
writer.addAttribute(CswConstants.VERBOSE_RESPONSE, String.valueOf(request.isVerbose()));
writer.addAttribute(CswConstants.XMLNS + CswConstants.NAMESPACE_DELIMITER + CswConstants.CSW_NAMESPACE_PREFIX, CswConstants.CSW_OUTPUT_SCHEMA);
writer.addAttribute(CswConstants.XMLNS + CswConstants.NAMESPACE_DELIMITER + CswConstants.OGC_NAMESPACE_PREFIX, CswConstants.OGC_SCHEMA);
for (InsertAction insertAction : request.getInsertActions()) {
writer.startNode(CswConstants.CSW_TRANSACTION_INSERT_NODE);
writer.addAttribute(CswConstants.TYPE_NAME_PARAMETER, insertAction.getTypeName());
marshallingContext.put(CswConstants.TRANSFORMER_LOOKUP_KEY, TransformerManager.ID);
marshallingContext.put(CswConstants.TRANSFORMER_LOOKUP_VALUE, insertAction.getTypeName());
for (Metacard metacard : insertAction.getRecords()) {
marshallingContext.convertAnother(metacard, delegatingTransformer);
}
writer.endNode();
}
for (UpdateAction updateAction : request.getUpdateActions()) {
writer.startNode(CswConstants.CSW_TRANSACTION_UPDATE_NODE);
writer.addAttribute(CswConstants.TYPE_NAME_PARAMETER, updateAction.getTypeName());
marshallingContext.put(CswConstants.TRANSFORMER_LOOKUP_KEY, TransformerManager.ID);
marshallingContext.put(CswConstants.TRANSFORMER_LOOKUP_VALUE, updateAction.getTypeName());
marshallingContext.convertAnother(updateAction.getMetacard(), delegatingTransformer);
writer.endNode();
}
for (DeleteAction deleteAction : request.getDeleteActions()) {
writer.startNode(CswConstants.CSW_TRANSACTION_DELETE_NODE);
writer.addAttribute(CswConstants.TYPE_NAME_PARAMETER, deleteAction.getTypeName());
writer.startNode(CswConstants.CSW_CONSTRAINT);
writer.addAttribute(CswConstants.VERSION, CswConstants.CONSTRAINT_VERSION);
writer.startNode(CswConstants.CSW_CQL_TEXT);
writer.setValue(deleteAction.getConstraint().getCqlText());
writer.endNode();
writer.endNode();
writer.endNode();
}
}
use of org.codice.ddf.spatial.ogc.csw.catalog.common.transaction.InsertAction in project ddf by codice.
the class TestCswEndpoint method testIngestVerboseTransaction.
@Test
public void testIngestVerboseTransaction() throws CswException, SourceUnavailableException, FederationException, IngestException {
CswTransactionRequest request = new CswTransactionRequest();
request.getInsertActions().add(new InsertAction(CswConstants.CSW_TYPE, null, Arrays.asList(new MetacardImpl())));
request.setVerbose(true);
TransactionResponseType response = csw.transaction(request);
assertThat(response, notNullValue());
assertThat(response.getInsertResult().size(), is(1));
assertThat(response.getTransactionSummary(), notNullValue());
TransactionSummaryType summary = response.getTransactionSummary();
assertThat(summary.getTotalDeleted().intValue(), is(0));
assertThat(summary.getTotalUpdated().intValue(), is(0));
assertThat(summary.getTotalInserted().intValue(), is(1));
String contextPath = StringUtils.join(new String[] { CswConstants.OGC_CSW_PACKAGE, CswConstants.OGC_FILTER_PACKAGE, CswConstants.OGC_GML_PACKAGE, CswConstants.OGC_OWS_PACKAGE }, ":");
verifyMarshalResponse(response, contextPath, new QName(CswConstants.CSW_OUTPUT_SCHEMA, CswConstants.TRANSACTION));
}
use of org.codice.ddf.spatial.ogc.csw.catalog.common.transaction.InsertAction in project ddf by codice.
the class TransactionRequestConverter method unmarshal.
@Override
public Object unmarshal(HierarchicalStreamReader reader, UnmarshallingContext context) {
CswTransactionRequest cswTransactionRequest = new CswTransactionRequest();
cswTransactionRequest.setVersion(reader.getAttribute(CswConstants.VERSION));
cswTransactionRequest.setService(reader.getAttribute(CswConstants.SERVICE));
cswTransactionRequest.setVerbose(Boolean.valueOf(reader.getAttribute(CswConstants.VERBOSE_RESPONSE)));
XStreamAttributeCopier.copyXmlNamespaceDeclarationsIntoContext(reader, context);
while (reader.hasMoreChildren()) {
reader.moveDown();
if (reader.getNodeName().contains("Insert")) {
String typeName = StringUtils.defaultIfEmpty(reader.getAttribute(CswConstants.TYPE_NAME_PARAMETER), CswConstants.CSW_RECORD);
String handle = StringUtils.defaultIfEmpty(reader.getAttribute(CswConstants.HANDLE_PARAMETER), "");
context.put(CswConstants.TRANSFORMER_LOOKUP_KEY, TransformerManager.ID);
context.put(CswConstants.TRANSFORMER_LOOKUP_VALUE, typeName);
List<Metacard> metacards = new ArrayList<>();
// Loop through the individual records to be inserted, converting each into a Metacard
while (reader.hasMoreChildren()) {
// move down to the record's tag
reader.moveDown();
Metacard metacard = (Metacard) context.convertAnother(null, MetacardImpl.class, delegatingTransformer);
if (metacard != null) {
metacards.add(metacard);
}
// move back up to the <SearchResults> parent of the <csw:Record> tags
reader.moveUp();
}
cswTransactionRequest.getInsertActions().add(new InsertAction(typeName, handle, metacards));
} else if (reader.getNodeName().contains("Delete")) {
XStreamAttributeCopier.copyXmlNamespaceDeclarationsIntoContext(reader, context);
Map<String, String> xmlnsAttributeToUriMappings = getXmlnsAttributeToUriMappingsFromContext(context);
Map<String, String> prefixToUriMappings = getPrefixToUriMappingsFromXmlnsAttributes(xmlnsAttributeToUriMappings);
StringWriter writer = new StringWriter();
XStreamAttributeCopier.copyXml(reader, writer, xmlnsAttributeToUriMappings);
DeleteType deleteType = getElementFromXml(writer.toString(), DeleteType.class);
cswTransactionRequest.getDeleteActions().add(new DeleteAction(deleteType, prefixToUriMappings));
} else if (reader.getNodeName().contains("Update")) {
XStreamAttributeCopier.copyXmlNamespaceDeclarationsIntoContext(reader, context);
UpdateAction updateAction = parseUpdateAction(reader, context);
cswTransactionRequest.getUpdateActions().add(updateAction);
}
reader.moveUp();
}
return cswTransactionRequest;
}
use of org.codice.ddf.spatial.ogc.csw.catalog.common.transaction.InsertAction in project ddf by codice.
the class TestTransactionRequestConverter method testValidInsertMarshal.
@Test
public void testValidInsertMarshal() throws SAXException, IOException, XpathException {
CswTransactionRequest transactionRequest = new CswTransactionRequest();
MetacardImpl metacard = new MetacardImpl();
metacard.setId(METACARD_ID);
InsertAction insertAction = new InsertAction(CswConstants.CSW_METACARD_TYPE_NAME, null, Arrays.asList(metacard));
transactionRequest.getInsertActions().add(insertAction);
transactionRequest.setService(CswConstants.CSW);
transactionRequest.setVerbose(true);
transactionRequest.setVersion(CswConstants.VERSION_2_0_2);
String xml = xStream.toXML(transactionRequest);
System.out.println("xml = " + xml);
System.out.println("EXPECTED_INSERT_XML = " + EXPECTED_INSERT_XML);
Diff diff = XMLUnit.compareXML(xml, EXPECTED_INSERT_XML);
assertThat(diff.similar(), is(true));
}
use of org.codice.ddf.spatial.ogc.csw.catalog.common.transaction.InsertAction in project ddf by codice.
the class TestTransactionMessageBodyReader method testReadInsertAndDeleteFrom.
@Test
public void testReadInsertAndDeleteFrom() throws IOException {
Converter mockConverter = mock(Converter.class);
when(mockConverter.canConvert(any(Metacard.class.getClass()))).thenReturn(true);
when(mockConverter.unmarshal(any(HierarchicalStreamReader.class), any(UnmarshallingContext.class))).thenReturn(mock(Metacard.class));
TransactionMessageBodyReader reader = new TransactionMessageBodyReader(mockConverter, CswQueryFactoryTest.getCswMetacardType(), registry);
CswTransactionRequest request = reader.readFrom(CswTransactionRequest.class, null, null, null, null, IOUtils.toInputStream(INSERT_AND_DELETE_REQUEST_XML));
assertThat(request, notNullValue());
assertThat(request.getDeleteActions().size(), is(1));
assertThat(request.getInsertActions().size(), is(1));
assertThat(request.getUpdateActions().size(), is(0));
DeleteAction deleteAction = request.getDeleteActions().get(0);
assertThat(deleteAction, notNullValue());
assertThat(deleteAction.getTypeName(), is(CswConstants.CSW_RECORD));
assertThat(deleteAction.getHandle(), is("something"));
assertThat(deleteAction.getConstraint(), notNullValue());
assertThat(deleteAction.getConstraint().getCqlText().trim(), is("title = 'foo'"));
InsertAction insertAction = request.getInsertActions().get(0);
assertThat(insertAction, notNullValue());
assertThat(insertAction.getRecords().size(), is(1));
assertThat(request.getService(), is(CswConstants.CSW));
assertThat(request.getVersion(), is(CswConstants.VERSION_2_0_2));
assertThat(request.isVerbose(), is(true));
}
Aggregations