01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * published by the Free Software Foundation; either version 2.1 of
10: * the License, or (at your option) any later version.
11: *
12: * This software is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this software; if not, write to the Free
19: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21: */
22: package org.jboss.invocation.unified.marshall;
23:
24: import org.jboss.remoting.marshal.serializable.SerializableUnMarshaller;
25: import org.jboss.remoting.marshal.UnMarshaller;
26: import org.jboss.remoting.marshal.UnMarshallerDecorator;
27: import org.jboss.remoting.InvocationRequest;
28: import org.jboss.invocation.MarshalledInvocation;
29: import org.jboss.tm.TransactionPropagationContextUtil;
30: import org.jboss.tm.TransactionPropagationContextImporter;
31:
32: import java.io.InputStream;
33: import java.io.IOException;
34: import java.util.Map;
35:
36: /**
37: * This is a hollow implementation in that it only over rides the DATATYPE
38: * value. All behavior is that of SerializableUnMarshaller.
39: *
40: * @author <a href="mailto:tom@jboss.org">Tom Elrod</a>
41: */
42: public class InvocationUnMarshaller extends SerializableUnMarshaller
43: implements UnMarshallerDecorator {
44: /** @since 4.2.0 */
45: static final long serialVersionUID = -7323800106085276090L;
46:
47: public final static String DATATYPE = "invocation";
48:
49: public Object read(InputStream inputStream, Map metadata)
50: throws IOException, ClassNotFoundException {
51: Object ret = super .read(inputStream, metadata);
52:
53: return removeDecoration(ret);
54: }
55:
56: public Object read(InputStream inputStream, Map metadata,
57: int version) throws IOException, ClassNotFoundException {
58: Object ret = super .read(inputStream, metadata, version);
59:
60: return removeDecoration(ret);
61: }
62:
63: public UnMarshaller cloneUnMarshaller()
64: throws CloneNotSupportedException {
65: InvocationUnMarshaller unmarshaller = new InvocationUnMarshaller();
66: unmarshaller.setClassLoader(this .customClassLoader);
67: return unmarshaller;
68: }
69:
70: public Object removeDecoration(Object obj) throws IOException {
71: if (obj instanceof InvocationRequest) {
72: InvocationRequest remoteInv = (InvocationRequest) obj;
73: Object param = remoteInv.getParameter();
74:
75: if (param instanceof MarshalledInvocation) {
76: MarshalledInvocation mi = (MarshalledInvocation) param;
77: Object txCxt = mi.getTransactionPropagationContext();
78: if (txCxt != null) {
79: TransactionPropagationContextImporter tpcImporter = TransactionPropagationContextUtil
80: .getTPCImporter();
81: mi
82: .setTransaction(tpcImporter
83: .importTransactionPropagationContext(txCxt));
84: }
85: }
86: }
87: return obj;
88: }
89:
90: }
|