1 package org.codehaus.xfire.util;
2
3 import javax.xml.stream.XMLStreamException;
4
5 import org.codehaus.xfire.MessageContext;
6 import org.codehaus.xfire.exchange.InMessage;
7 import org.codehaus.xfire.transport.ChannelEndpoint;
8 import org.codehaus.yom.Document;
9 import org.codehaus.yom.Element;
10 import org.codehaus.yom.stax.StaxBuilder;
11
12 public class YOMEndpoint
13 implements ChannelEndpoint
14 {
15 private int count = 0;
16 private Document message;
17
18 public void onReceive(MessageContext context, InMessage msg)
19 {
20 count++;
21 StaxBuilder builder = new StaxBuilder();
22 try
23 {
24 Element root = builder.buildElement(null, msg.getXMLStreamReader());
25 if (root != null)
26 {
27 message = new Document(root);
28 }
29 else
30 {
31 message = null;
32 }
33 }
34 catch (XMLStreamException e)
35 {
36 e.printStackTrace();
37 }
38 }
39
40 public int getCount()
41 {
42 return count;
43 }
44
45 public Document getMessage()
46 {
47 return message;
48 }
49 }