/* | |
* Licensed to the Apache Software Foundation (ASF) under one | |
* or more contributor license agreements. See the NOTICE file | |
* distributed with this work for additional information | |
* regarding copyright ownership. The ASF licenses this file | |
* to you 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 org.apache.axis2.jaxws.sample.addnumbershandler; | |
import org.apache.axis2.jaxws.handler.LogicalMessageContext; | |
import javax.xml.transform.OutputKeys; | |
import javax.xml.transform.Source; | |
import javax.xml.transform.Transformer; | |
import javax.xml.transform.TransformerFactory; | |
import javax.xml.transform.stream.StreamResult; | |
import javax.xml.transform.stream.StreamSource; | |
import javax.xml.ws.LogicalMessage; | |
import javax.xml.ws.handler.MessageContext; | |
import java.io.ByteArrayOutputStream; | |
import java.io.StringBufferInputStream; | |
import java.util.StringTokenizer; | |
/* | |
* You can't actually specify whether a handler is for client or server, | |
* you just have to check in the handleMessage and/or handleFault to make | |
* sure what direction we're going. | |
*/ | |
public class AddNumbersClientLogicalHandler2 implements javax.xml.ws.handler.LogicalHandler<LogicalMessageContext> { | |
HandlerTracker tracker = new HandlerTracker(AddNumbersClientLogicalHandler2.class.getSimpleName()); | |
public void close(MessageContext messagecontext) { | |
tracker.close(); | |
} | |
public boolean handleFault(LogicalMessageContext messagecontext) { | |
Boolean outbound = (Boolean) messagecontext.get(MessageContext.MESSAGE_OUTBOUND_PROPERTY); | |
tracker.handleFault(outbound); | |
return true; | |
} | |
public boolean handleMessage(LogicalMessageContext messagecontext) { | |
Boolean outbound = (Boolean)messagecontext.get(MessageContext.MESSAGE_OUTBOUND_PROPERTY); | |
tracker.handleMessage(outbound); | |
if (outbound) { // outbound request on the client | |
LogicalMessage msg = messagecontext.getMessage(); | |
Source payload = msg.getPayload(); | |
String st = getStringFromSourcePayload(payload); | |
String txt = String.valueOf(Integer.valueOf(getFirstArg(st)) * 2); | |
st = replaceFirstArg(st, txt); | |
msg.setPayload(new StreamSource(new StringBufferInputStream(st))); | |
} | |
return true; | |
} | |
private static String getFirstArg(String payloadString) { | |
StringTokenizer st = new StringTokenizer(payloadString, ">"); | |
st.nextToken(); // skip first token. | |
st.nextToken(); // skip second | |
String tempString = st.nextToken(); | |
String returnString = new StringTokenizer(tempString, "<").nextToken(); | |
return returnString; | |
} | |
private static String replaceFirstArg(String payloadString, String newArg) { | |
String firstArg = getFirstArg(payloadString); | |
payloadString = payloadString.replaceFirst(firstArg, newArg); | |
return payloadString; | |
} | |
private static String getStringFromSourcePayload(Source payload) { | |
try { | |
TransformerFactory factory = TransformerFactory.newInstance(); | |
Transformer trans = factory.newTransformer(); | |
ByteArrayOutputStream baos = new ByteArrayOutputStream(); | |
StreamResult result = new StreamResult(baos); | |
trans.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes"); | |
trans.transform(payload, result); | |
return new String(baos.toByteArray()); | |
} catch (Exception e) { | |
throw new RuntimeException(e); | |
} | |
} | |
} |