/******************************************************************************* | |
* Copyright (c) 1998, 2013 Oracle and/or its affiliates. All rights reserved. | |
* This program and the accompanying materials are made available under the | |
* terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 | |
* which accompanies this distribution. | |
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html | |
* and the Eclipse Distribution License is available at | |
* http://www.eclipse.org/org/documents/edl-v10.php. | |
* | |
* Contributors: | |
* Oracle - initial API and implementation from Oracle TopLink | |
******************************************************************************/ | |
package org.eclipse.persistence.jaxb; | |
import java.lang.reflect.InvocationTargetException; | |
import java.security.AccessController; | |
import java.security.PrivilegedActionException; | |
import java.util.Map; | |
import javax.xml.bind.Marshaller; | |
import org.eclipse.persistence.exceptions.XMLMarshalException; | |
import org.eclipse.persistence.internal.oxm.Root; | |
import org.eclipse.persistence.internal.security.PrivilegedAccessHelper; | |
import org.eclipse.persistence.internal.security.PrivilegedMethodInvoker; | |
import org.eclipse.persistence.jaxb.compiler.MarshalCallback; | |
import org.eclipse.persistence.oxm.XMLMarshalListener; | |
/** | |
* INTERNAL: | |
* <p><b>Purpose:</b>Provide a wrapper for a JAXB 2.0 Marshal Listener that implements | |
* XMLMarshalListener | |
* <p><b>Responsibilities:</b><ul> | |
* <li>Implement the XMLMarshalListener API</li> | |
* <li>Delegate event callbacks to the wrapped Listener instance</li> | |
* <li>Perform JAXB 2.0 Class-Based marshal event callbacks</li> | |
* </ul> | |
* | |
* @see javax.xml.bind.Marshaller.Listener | |
* @see org.eclipse.persistence.oxm.XMLMarshalListener | |
*/ | |
public class JAXBMarshalListener implements XMLMarshalListener { | |
private Marshaller.Listener listener; | |
private Map classBasedMarshalEvents; | |
private javax.xml.bind.Marshaller marshaller; | |
private JAXBContext jaxbContext; | |
public JAXBMarshalListener(JAXBContext context, javax.xml.bind.Marshaller marshaller) { | |
jaxbContext = context; | |
this.marshaller = marshaller; | |
} | |
public void setListener(Marshaller.Listener jaxbListener) { | |
this.listener = jaxbListener; | |
} | |
public Marshaller.Listener getListener() { | |
return listener; | |
} | |
public void beforeMarshal(Object obj) { | |
if(classBasedMarshalEvents != null) { | |
MarshalCallback callback = (MarshalCallback)classBasedMarshalEvents.get(obj.getClass().getName()); | |
if(callback != null && callback.getBeforeMarshalCallback() != null) { | |
try { | |
if (PrivilegedAccessHelper.shouldUsePrivilegedAccess()){ | |
try{ | |
AccessController.doPrivileged(new PrivilegedMethodInvoker(callback.getBeforeMarshalCallback(), obj, new Object[]{marshaller})); | |
}catch (PrivilegedActionException ex){ | |
if (ex.getCause() instanceof IllegalAccessException){ | |
throw (IllegalAccessException) ex.getCause(); | |
} | |
if (ex.getCause() instanceof InvocationTargetException){ | |
throw (InvocationTargetException) ex.getCause(); | |
} | |
throw (RuntimeException)ex.getCause(); | |
} | |
}else{ | |
PrivilegedAccessHelper.invokeMethod(callback.getBeforeMarshalCallback(), obj, new Object[]{marshaller}); | |
} | |
} catch(Exception ex) { | |
throw XMLMarshalException.marshalException(ex); | |
} | |
} | |
} | |
if (listener != null) { | |
if(obj instanceof Root){ | |
obj = jaxbContext.createJAXBElementFromXMLRoot((Root) obj, ((Root) obj).getDeclaredType()); | |
} | |
listener.beforeMarshal(obj); | |
} | |
} | |
public void afterMarshal(Object obj) { | |
if(classBasedMarshalEvents != null) { | |
MarshalCallback callback = (MarshalCallback)classBasedMarshalEvents.get(obj.getClass().getName()); | |
if (callback != null && callback.getAfterMarshalCallback() != null) { | |
try { | |
if (PrivilegedAccessHelper.shouldUsePrivilegedAccess()){ | |
try{ | |
AccessController.doPrivileged(new PrivilegedMethodInvoker(callback.getAfterMarshalCallback(), obj, new Object[]{marshaller})); | |
}catch (PrivilegedActionException ex){ | |
if (ex.getCause() instanceof IllegalAccessException){ | |
throw (IllegalAccessException) ex.getCause(); | |
} | |
if (ex.getCause() instanceof InvocationTargetException){ | |
throw (InvocationTargetException) ex.getCause(); | |
} | |
throw (RuntimeException)ex.getCause(); | |
} | |
}else{ | |
PrivilegedAccessHelper.invokeMethod(callback.getAfterMarshalCallback(), obj, new Object[]{marshaller}); | |
} | |
} catch(Exception ex) { | |
throw XMLMarshalException.marshalException(ex); | |
} | |
} | |
} | |
if (listener != null) { | |
if(obj instanceof Root){ | |
obj = jaxbContext.createJAXBElementFromXMLRoot((Root) obj, ((Root) obj).getDeclaredType()); | |
} | |
listener.afterMarshal(obj); | |
} | |
} | |
public void setClassBasedMarshalEvents(Map events) { | |
this.classBasedMarshalEvents = events; | |
} | |
} |