/*
* Copyright (C) 2004, 2005 Joe Walnes.
* Copyright (C) 2006, 2007, 2008 XStream Committers.
* All rights reserved.
*
* The software in this package is published under the terms of the BSD
* style license a copy of which has been included with this distribution in
* the LICENSE.txt file.
*
* Created on 25. March 2004 by Joe Walnes
*/
package com.thoughtworks.xstream.converters.extended;
import com.thoughtworks.xstream.converters.ConversionException;
import com.thoughtworks.xstream.converters.Converter;
import com.thoughtworks.xstream.converters.MarshallingContext;
import com.thoughtworks.xstream.converters.UnmarshallingContext;
import com.thoughtworks.xstream.io.HierarchicalStreamReader;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import com.thoughtworks.xstream.mapper.DynamicProxyMapper;
import com.thoughtworks.xstream.mapper.Mapper;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Proxy;
import java.util.ArrayList;
import java.util.List;
/**
* Converts a dynamic proxy to XML, storing the implemented
* interfaces and handler.
*
* @author Joe Walnes
*/
public class DynamicProxyConverter implements Converter {
private ClassLoader classLoader;
private Mapper mapper;
public DynamicProxyConverter(Mapper mapper) {
this(mapper, DynamicProxyConverter.class.getClassLoader());
}
public DynamicProxyConverter(Mapper mapper, ClassLoader classLoader) {
this.classLoader = classLoader;
this.mapper = mapper;
}
public boolean canConvert(Class type) {
return type.equals(DynamicProxyMapper.DynamicProxy.class) || Proxy.isProxyClass(type);
}
public void marshal(Object source, HierarchicalStreamWriter writer, MarshallingContext context) {
InvocationHandler invocationHandler = Proxy.getInvocationHandler(source);
addInterfacesToXml(source, writer);
writer.startNode("handler");
String attributeName = mapper.aliasForSystemAttribute("class");
if (attributeName != null) {
writer.addAttribute(attributeName, mapper.serializedClass(invocationHandler.getClass()));
}
context.convertAnother(invocationHandler);
writer.endNode();
}
private void addInterfacesToXml(Object source, HierarchicalStreamWriter writer) {
Class[] interfaces = source.getClass().getInterfaces();
for (int i = 0; i < interfaces.length; i++) {
Class currentInterface = interfaces[i];
writer.startNode("interface");
writer.setValue(mapper.serializedClass(currentInterface));
writer.endNode();
}
}
public Object unmarshal(HierarchicalStreamReader reader, UnmarshallingContext context) {
List interfaces = new ArrayList();
InvocationHandler handler = null;
while (reader.hasMoreChildren()) {
reader.moveDown();
String elementName = reader.getNodeName();
if (elementName.equals("interface")) {
interfaces.add(mapper.realClass(reader.getValue()));
} else if (elementName.equals("handler")) {
String attributeName = mapper.aliasForSystemAttribute("class");
if (attributeName != null) {
Class handlerType = mapper.realClass(reader.getAttribute(attributeName));
handler = (InvocationHandler) context.convertAnother(null, handlerType);
}
}
reader.moveUp();
}
if (handler == null) {
throw new ConversionException("No InvocationHandler specified for dynamic proxy");
}
Class[] interfacesAsArray = new Class[interfaces.size()];
interfaces.toArray(interfacesAsArray);
return Proxy.newProxyInstance(classLoader, interfacesAsArray, handler);
}
}