/******************************************************************************* * Copyright 2013 SAP AG * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. ******************************************************************************/ package com.sap.core.odata.core.ep.producer; import java.io.Writer; import java.util.List; import javax.xml.stream.FactoryConfigurationError; import javax.xml.stream.XMLOutputFactory; import javax.xml.stream.XMLStreamException; import javax.xml.stream.XMLStreamWriter; import com.sap.core.odata.api.edm.Edm; import com.sap.core.odata.api.edm.EdmEntitySetInfo; import com.sap.core.odata.api.edm.EdmServiceMetadata; import com.sap.core.odata.api.ep.EntityProviderException; import com.sap.core.odata.api.exception.ODataException; import com.sap.core.odata.core.commons.ContentType; import com.sap.core.odata.core.ep.util.FormatXml; /** * Writes the OData service document in XML. * @author SAP AG */ public class AtomServiceDocumentProducer { private static final String DEFAULT_CHARSET = ContentType.CHARSET_UTF_8; private static final String XML_VERSION = "1.0"; private final Edm edm; private final String serviceRoot; public AtomServiceDocumentProducer(final Edm edm, final String serviceRoot) { this.edm = edm; this.serviceRoot = serviceRoot; } public void writeServiceDocument(final Writer writer) throws EntityProviderException { EdmServiceMetadata serviceMetadata = edm.getServiceMetadata(); try { XMLStreamWriter xmlStreamWriter = XMLOutputFactory.newInstance().createXMLStreamWriter(writer); xmlStreamWriter.writeStartDocument(DEFAULT_CHARSET, XML_VERSION); xmlStreamWriter.setPrefix(Edm.PREFIX_XML, Edm.NAMESPACE_XML_1998); xmlStreamWriter.setPrefix(Edm.PREFIX_ATOM, Edm.NAMESPACE_ATOM_2005); xmlStreamWriter.setDefaultNamespace(Edm.NAMESPACE_APP_2007); xmlStreamWriter.writeStartElement(FormatXml.APP_SERVICE); xmlStreamWriter.writeAttribute(Edm.PREFIX_XML, Edm.NAMESPACE_XML_1998, FormatXml.XML_BASE, serviceRoot); xmlStreamWriter.writeDefaultNamespace(Edm.NAMESPACE_APP_2007); xmlStreamWriter.writeNamespace(Edm.PREFIX_ATOM, Edm.NAMESPACE_ATOM_2005); xmlStreamWriter.writeStartElement(FormatXml.APP_WORKSPACE); xmlStreamWriter.writeStartElement(Edm.NAMESPACE_ATOM_2005, FormatXml.ATOM_TITLE); xmlStreamWriter.writeCharacters(FormatXml.ATOM_TITLE_DEFAULT); xmlStreamWriter.writeEndElement(); List<EdmEntitySetInfo> entitySetInfos = serviceMetadata.getEntitySetInfos(); for (EdmEntitySetInfo info : entitySetInfos) { xmlStreamWriter.writeStartElement(FormatXml.APP_COLLECTION); xmlStreamWriter.writeAttribute(FormatXml.ATOM_HREF, info.getEntitySetUri().toASCIIString()); xmlStreamWriter.writeStartElement(Edm.NAMESPACE_ATOM_2005, FormatXml.ATOM_TITLE); xmlStreamWriter.writeCharacters(info.getEntitySetName()); xmlStreamWriter.writeEndElement(); xmlStreamWriter.writeEndElement(); } // Collection<Schema> schemas = edmProvider.getSchemas(); // if (schemas != null) { // for (Schema schema : schemas) { // Collection<EntityContainer> entityContainers = schema.getEntityContainers(); // if (entityContainers != null) { // for (EntityContainer entityContainer : entityContainers) { // Collection<EntitySet> entitySets = entityContainer.getEntitySets(); // for (EntitySet entitySet : entitySets) { // xmlStreamWriter.writeStartElement(FormatXml.APP_COLLECTION); // if (entityContainer.isDefaultEntityContainer()) { // xmlStreamWriter.writeAttribute(FormatXml.ATOM_HREF, entitySet.getName()); // } else { // xmlStreamWriter.writeAttribute(FormatXml.ATOM_HREF, entityContainer.getName() + Edm.DELIMITER + entitySet.getName()); // } // xmlStreamWriter.writeStartElement(Edm.NAMESPACE_ATOM_2005, FormatXml.ATOM_TITLE); // xmlStreamWriter.writeCharacters(entitySet.getName()); // xmlStreamWriter.writeEndElement(); // xmlStreamWriter.writeEndElement(); // } // } // } // } // } xmlStreamWriter.writeEndElement(); xmlStreamWriter.writeEndElement(); xmlStreamWriter.writeEndDocument(); xmlStreamWriter.flush(); } catch (FactoryConfigurationError e) { throw new EntityProviderException(EntityProviderException.COMMON, e); } catch (XMLStreamException e) { throw new EntityProviderException(EntityProviderException.COMMON, e); } catch (ODataException e) { throw new EntityProviderException(EntityProviderException.COMMON, e); } } }