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 StaxBuilder builder = new StaxBuilder(); 21 try 22 { 23 Element root = builder.buildElement(null, msg.getXMLStreamReader()); 24 if (root != null) 25 { 26 message = new Document(root); 27 } 28 else 29 { 30 message = null; 31 } 32 } 33 catch (XMLStreamException e) 34 { 35 e.printStackTrace(); 36 } 37 count++; 38 } 39 40 public int getCount() 41 { 42 return count; 43 } 44 45 public Document getMessage() 46 { 47 return message; 48 } 49 }