1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
|
/*
* Copyright 2001-2004 The Apache Software Foundation.
*
* 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 samples.attachments;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import javax.activation.DataHandler;
/**
* @author Rick Rineholt
*/
/**
* An example of
* This class has a main method that beside the standard arguments
* allows you to specify an attachment that will be sent to a
* service which will then send it back.
*
*/
public class EchoAttachmentsService {
/**
* This method implements a web service that sends back
* any attachment it receives.
*/
public DataHandler echo( DataHandler dh) {
System.err.println("In echo");
//Attachments are sent by default back as a MIME stream if no attachments were
// received. If attachments are received the same format that was received will
// be the default stream type for any attachments sent.
//The following two commented lines would force any attachments sent back.
// to be in DIME format.
//Message rspmsg=AxisEngine.getCurrentMessageContext().getResponseMessage();
//rspmsg.getAttachmentsImpl().setSendType(org.apache.axis.attachments.Attachments.SEND_TYPE_DIME);
if (dh == null ) System.err.println("dh is null");
else System.err.println("Received \""+dh.getClass().getName()+"\".");
return dh;
}
/**
* This method implements a web service that sends back
* an array of attachment it receives.
*/
public DataHandler[] echoDir( DataHandler[] attachments) {
System.err.println("In echoDir");
//Attachments are sent by default back as a MIME stream if no attachments were
// received. If attachments are received the same format that was received will
// be the default stream type for any attachments sent.
//The following two commented lines would force any attachments sent back.
// to be in DIME format.
//Message rspmsg=AxisEngine.getCurrentMessageContext().getResponseMessage();
//rspmsg.getAttachmentsImpl().setSendType(org.apache.axis.attachments.Attachments.SEND_TYPE_DIME);
if (attachments == null ) System.err.println("attachments is null!");
else System.err.println("Got " + attachments.length + " attachments!");
return attachments;
}
public Document attachments( Document xml)
throws org.apache.axis.AxisFault,java.io.IOException, org.xml.sax.SAXException,
java.awt.datatransfer.UnsupportedFlavorException,javax.xml.parsers.ParserConfigurationException,
java.lang.ClassNotFoundException,javax.xml.soap.SOAPException {
System.err.println("In message handling attachments directly.");
org.apache.axis.MessageContext msgContext= org.apache.axis.MessageContext.getCurrentContext();
org.apache.axis.Message reqMsg= msgContext.getRequestMessage();
org.apache.axis.attachments.Attachments attachments=reqMsg.getAttachmentsImpl();
if(null == attachments){
throw new org.apache.axis.AxisFault("No support for attachments" );
}
Element rootEl= xml.getDocumentElement();
Element caEl= getNextFirstChildElement(rootEl);
StringBuffer fullmsg= new StringBuffer();
java.util.Vector reply= new java.util.Vector();
for(int count=1 ;caEl != null; caEl= getNextSiblingElement(caEl), ++count){
String href= caEl.getAttribute("href");
org.apache.axis.Part p= attachments.getAttachmentByReference(href);
if(null == p)
throw new org.apache.axis.AxisFault("Attachment for ref='"+href+"' not found." );
String ordinalStr =getOrdinalHeaders(p);
if( null == ordinalStr || ordinalStr.trim().length()==0)
throw new org.apache.axis.AxisFault("Ordinal for attachment ref='"+href+"' not found." );
int ordinal= Integer.parseInt(ordinalStr);
if(count != ordinal)
throw new org.apache.axis.AxisFault("Ordinal for attachment ref='"+href+"' excpected" + count + " got " + ordinal +"." );
//check content type.
if(!"text/plain".equals(p.getContentType()))
throw new org.apache.axis.AxisFault("Attachment ref='"+href+"' bad content-type:'"+p.getContentType()+"'." );
//now get at the data...
DataHandler dh= ((org.apache.axis.attachments.AttachmentPart)p).getDataHandler();
String pmsg=(String )dh.getContent();
fullmsg.append(pmsg);
reply.add(pmsg);
}
if(!(samples.attachments.TestRef .TheKey.equals(fullmsg.toString())))
throw new org.apache.axis.AxisFault("Fullmsg not correct'"+fullmsg +"'." );
System.out.println(fullmsg.toString());
//Now lets Java serialize the reply...
java.io.ByteArrayOutputStream byteStream = new java.io.ByteArrayOutputStream();
java.io.ObjectOutputStream oos = new java.io.ObjectOutputStream(byteStream);
oos.writeObject(reply);
oos.close();
byte[] replyJavaSerialized= byteStream.toByteArray();
byteStream=null; oos= null;
org.apache.axis.attachments.AttachmentPart replyPart= new
org.apache.axis.attachments.AttachmentPart(
new DataHandler( new MemoryOnlyDataSource(replyJavaSerialized,
java.awt.datatransfer.DataFlavor.javaSerializedObjectMimeType+"; class=\""
+ reply.getClass().getName()+"\"")));
//Now lets add the attachment to the response message.
org.apache.axis.Message rspMsg= msgContext.getResponseMessage();
rspMsg.addAttachmentPart(replyPart);
//Iterate over the attachments... not by reference.
String ordinalPattern="";
for(java.util.Iterator ai=reqMsg.getAttachments(); ai.hasNext();){
org.apache.axis.Part p= (org.apache.axis.Part) ai.next();
ordinalPattern += getOrdinalHeaders(p);
}
//Now build the return document in a string buffer...
StringBuffer msgBody = new StringBuffer("\n<attachments xmlns=\"");
msgBody.append(rootEl.getNamespaceURI())
.append("\">\n")
.append("\t<attachment href=\"")
.append(replyPart.getContentIdRef())
.append("\" ordinalPattern=\"")
.append(ordinalPattern)
.append("\"/>\n")
.append("</attachments>\n");
//Convert the string buffer to an XML document and return it.
return
org.apache.axis.utils.XMLUtils.newDocument(
new org.xml.sax.InputSource(new java.io.ByteArrayInputStream(
msgBody.toString().getBytes())));
}
Element getNextFirstChildElement(Node n) {
if(n== null) return null;
n= n.getFirstChild();
for(; n!= null && !(n instanceof Element); n= n.getNextSibling());
return (Element)n;
}
Element getNextSiblingElement(Node n) {
if(n== null) return null;
n= n.getNextSibling();
for(; n!= null && !(n instanceof Element); n= n.getNextSibling());
return (Element)n;
}
String getOrdinalHeaders( org.apache.axis.Part p){
StringBuffer ret= new StringBuffer();
for(java.util.Iterator i= p.getMatchingMimeHeaders( new String[]{samples.attachments.TestRef.positionHTTPHeader});
i.hasNext();){
javax.xml.soap.MimeHeader mh= (javax.xml.soap.MimeHeader) i.next();
String v= mh.getValue();
if(v != null) ret.append(v.trim());
}
return ret.toString();
}
/**This class should store all attachment data in memory */
static class MemoryOnlyDataSource extends org.apache.axis.attachments.ManagedMemoryDataSource{
MemoryOnlyDataSource( byte [] in, String contentType) throws java.io.IOException{
super( new java.io.ByteArrayInputStream( in) , Integer.MAX_VALUE -2, contentType, true);
}
MemoryOnlyDataSource( String in, String contentType)throws java.io.IOException{
this( in.getBytes() , contentType);
}
}
}
|
