com.sun.jersey.json.impl.writer.JsonXmlStreamWriter Maven / Gradle / Ivy
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2010-2012 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* http://glassfish.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package com.sun.jersey.json.impl.writer;
import java.io.IOException;
import java.io.StringWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import com.sun.jersey.api.json.JSONConfiguration;
/**
* Implementation of {@link XMLStreamWriter} for JSON streams in mapped notation.
*
* @author Jakub Podlesak (jakub.podlesak at oracle.com)
* @author Michal Gajdos (michal.gajdos at oracle.com)
*/
public class JsonXmlStreamWriter extends DefaultXmlStreamWriter implements XMLStreamWriter {
private static class WriterAdapter {
Writer writer;
boolean isEmpty = true;
WriterAdapter() {
}
WriterAdapter(Writer w) {
writer = w;
}
void write(String s) throws IOException {
assert null != writer;
writer.write(s);
isEmpty = false;
}
String getContent() {
return null;
}
}
private static final class StringWriterAdapter extends WriterAdapter {
StringWriterAdapter() {
writer = new StringWriter();
}
@Override
String getContent() {
return writer.toString();
}
}
private static final class DummyWriterAdapter extends WriterAdapter {
DummyWriterAdapter() {
}
@Override
void write(String s) throws IOException {
}
@Override
String getContent() {
return null;
}
}
private static final class ProcessingState {
String lastName;
String currentName;
WriterAdapter lastElementWriter;
Boolean lastWasPrimitive;
boolean lastIsArray;
boolean isNotEmpty = false;
WriterAdapter writer;
ProcessingState() {
writer = new StringWriterAdapter();
}
ProcessingState(WriterAdapter w) {
writer = w;
}
@Override
public String toString() {
return String.format("{currentName:%s, writer: \"%s\", lastName:%s, lastWriter: %s}", currentName, ((writer != null) ? writer.getContent() : null), lastName, ((lastElementWriter != null) ? lastElementWriter.getContent() : null));
}
}
Writer mainWriter;
boolean stripRoot;
char nsSeparator;
final List processingStack = new ArrayList();
int depth;
final Collection arrayElementNames = new LinkedList();
final Collection nonStringElementNames = new LinkedList();
final Map xml2JsonNs = new HashMap();
private final String rootName;
private JsonXmlStreamWriter(final Writer writer,
final JSONConfiguration config,
final String rootName) {
this.mainWriter = writer;
this.stripRoot = config.isRootUnwrapping();
this.rootName = rootName;
this.nsSeparator = config.getNsSeparator();
if (null != config.getArrays()) {
this.arrayElementNames.addAll(config.getArrays());
}
if (null != config.getNonStrings()) {
this.nonStringElementNames.addAll(config.getNonStrings());
}
if (null != config.getXml2JsonNs()) {
this.xml2JsonNs.putAll(config.getXml2JsonNs());
}
processingStack.add(createProcessingState());
depth = 0;
}
public static XMLStreamWriter createWriter(final Writer writer,
final JSONConfiguration config,
final String rootName) {
final Collection attrsAsElems = config.getAttributeAsElements();
if ((attrsAsElems != null) && !attrsAsElems.isEmpty()) {
return new A2EXmlStreamWriterProxy(
new JsonXmlStreamWriter(writer, config, rootName), attrsAsElems);
} else {
return new JsonXmlStreamWriter(writer, config, rootName);
}
}
public void close() throws XMLStreamException {
try {
this.mainWriter.close();
} catch (IOException ex) {
throw new XMLStreamException(ex);
}
}
public void flush() throws XMLStreamException {
try {
mainWriter.flush();
} catch (IOException ex) {
throw new XMLStreamException(ex);
}
}
public void writeEndDocument() throws XMLStreamException {
try {
if (null != processingStack.get(depth).lastElementWriter) {
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
}
if ((null == processingStack.get(depth).lastWasPrimitive) || !processingStack.get(depth).lastWasPrimitive) {
processingStack.get(depth).writer.write("}");
}
pollStack();
} catch (IOException ex) {
throw new XMLStreamException(ex);
}
}
public void writeEndElement() throws XMLStreamException {
try {
if (null != processingStack.get(depth).lastElementWriter) {
if (processingStack.get(depth).lastIsArray) {
processingStack.get(depth).writer.write(",");
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
processingStack.get(depth).writer.write("]");
} else {
if (isArrayElement(processingStack.get(depth).lastName)) { // one elem array
processingStack.get(depth).writer.write(processingStack.get(depth).lastIsArray ? "," : "[");
processingStack.get(depth).lastIsArray = true;
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
processingStack.get(depth).writer.write("]");
} else {
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
}
}
}
if (processingStack.get(depth).writer.isEmpty) {
String currentName = processingStack.get(depth).currentName;
currentName = currentName == null ? processingStack.get(depth - 1).currentName : currentName;
if(arrayElementNames.contains(currentName)
|| nonStringElementNames.contains(currentName)
|| rootName.equals(currentName)) {
processingStack.get(depth).writer.write("{}");
} else {
processingStack.get(depth).writer.write("null");
}
} else if ((null == processingStack.get(depth).lastWasPrimitive) || !processingStack.get(depth).lastWasPrimitive) {
processingStack.get(depth).writer.write("}");
}
processingStack.get(depth - 1).lastName = processingStack.get(depth - 1).currentName;
processingStack.get(depth - 1).lastWasPrimitive = false;
processingStack.get(depth - 1).lastElementWriter = processingStack.get(depth).writer;
pollStack();
} catch (IOException ex) {
throw new XMLStreamException(ex);
}
}
private QName getQName(String s) {
final String[] currentName = s.split(Character.toString(nsSeparator));
QName name = new QName(s);
if (currentName.length > 1) {
name = new QName(currentName[0], currentName[1]);
}
return name;
}
public void writeCharacters(char[] text, int start, int length) throws XMLStreamException {
writeCharacters(new String(text, start, length));
}
public void writeCharacters(String text) throws XMLStreamException {
if (processingStack.get(depth).isNotEmpty) {
writeStartElement(null, "$", null);
_writeCharacters(text);
writeEndElement();
} else {
_writeCharacters(text);
}
}
private void _writeCharacters(String text) throws XMLStreamException {
try {
if (isNonString(processingStack.get(depth - 1).currentName)) {
processingStack.get(depth).writer.write(JsonEncoder.encode(text));
} else {
processingStack.get(depth).writer.write("\"" + JsonEncoder.encode(text) + "\"");
}
processingStack.get(depth).lastWasPrimitive = true;
} catch (IOException ex) {
throw new XMLStreamException(ex);
}
}
public void writeEmptyElement(String localName) throws XMLStreamException {
writeEmptyElement(null, localName, null);
}
public void writeStartElement(String localName) throws XMLStreamException {
writeStartElement(null, localName, null);
}
public void writeAttribute(String localName, String value) throws XMLStreamException {
writeAttribute(null, null, localName, value);
}
public void writeEmptyElement(String namespaceURI, String localName) throws XMLStreamException {
writeEmptyElement(null, localName, null);
}
public void writeStartElement(String namespaceURI, String localName) throws XMLStreamException {
writeStartElement(null, localName, namespaceURI);
}
public void writeAttribute(String namespaceURI, String localName, String value) throws XMLStreamException {
writeAttribute(null, namespaceURI, localName, value);
}
public void writeEmptyElement(String prefix, String localName, String namespaceURI) throws XMLStreamException {
writeStartElement(localName);
writeEndElement();
}
private void pollStack() throws IOException {
processingStack.remove(depth--);
}
private void printStack(String localName) {
try {
for (int d = 0; d <= depth; d++) {
mainWriter.write("\n**" + d + ":" + processingStack.get(d));
}
mainWriter.write("\n*** [" + localName + "]");
} catch (IOException ex) {
Logger.getLogger(JsonXmlStreamWriter.class.getName()).log(Level.SEVERE, null, ex);
}
}
private boolean isArrayElement(String name) {
if (null == name) {
return false;
}
return arrayElementNames.contains(name);
}
private boolean isNonString(String name) {
if (null == name) {
return false;
}
return nonStringElementNames.contains(name);
}
private String getEffectiveName(String namespaceURI, String localName) {
if ((namespaceURI != null) && xml2JsonNs.containsKey(namespaceURI)) {
return String.format("%s%c%s", xml2JsonNs.get(namespaceURI), nsSeparator,localName);
} else {
return localName;
}
}
public void writeStartElement(String prefix, String localName, String namespaceURI) throws XMLStreamException {
String effectiveName = getEffectiveName(namespaceURI, localName);
processingStack.get(depth).isNotEmpty = true;
processingStack.get(depth).currentName = effectiveName;
try {
boolean isNextArrayElement = processingStack.get(depth).currentName.equals(processingStack.get(depth).lastName);
if (!isNextArrayElement) {
if (isArrayElement(processingStack.get(depth).lastName)) {
processingStack.get(depth).writer.write(processingStack.get(depth).lastIsArray ? "," : "[");
processingStack.get(depth).lastIsArray = true;
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
} else {
// write previous elements from buffer
if (null != processingStack.get(depth).lastElementWriter) {
if (processingStack.get(depth).lastIsArray) {
processingStack.get(depth).writer.write(",");
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
processingStack.get(depth).writer.write("]");
} else {
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
}
}
processingStack.get(depth).lastIsArray = false;
}
if (null != processingStack.get(depth).lastName) {
if (processingStack.get(depth).lastIsArray) {
processingStack.get(depth).writer.write("]");
processingStack.get(depth).lastIsArray = false;
}
processingStack.get(depth).writer.write(","); // next element at the same level
}
if (null == processingStack.get(depth).lastWasPrimitive) {
processingStack.get(depth).writer.write("{"); // first sub-element
}
processingStack.get(depth).writer.write("\"" + effectiveName + "\":");
} else { // next array element
processingStack.get(depth).writer.write(processingStack.get(depth).lastIsArray ? "," : "["); // next element at the same level
processingStack.get(depth).lastIsArray = true;
processingStack.get(depth).writer.write(processingStack.get(depth).lastElementWriter.getContent());
}
depth++;
processingStack.add(depth, createProcessingState());
} catch (IOException ex) {
throw new XMLStreamException(ex);
}
}
public void writeAttribute(String prefix, String namespaceURI, String localName, String value) throws XMLStreamException {
writeStartElement(prefix, "@" + getEffectiveName(namespaceURI, localName), null);
writeCharacters(value);
writeEndElement();
}
private ProcessingState createProcessingState() {
switch (depth) {
case 0:
return new ProcessingState(
stripRoot ? new DummyWriterAdapter() : new WriterAdapter(mainWriter));
case 1:
return stripRoot ? new ProcessingState(new WriterAdapter(mainWriter)) : new ProcessingState();
default:
return new ProcessingState();
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy