/*
* 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.client.config;
import org.apache.axis2.Constants;
import org.apache.axis2.description.Parameter;
import org.apache.axis2.jaxws.ExceptionFactory;
import org.apache.axis2.jaxws.binding.SOAPBinding;
import org.apache.axis2.jaxws.core.MessageContext;
import org.apache.axis2.jaxws.feature.ClientConfigurator;
import org.apache.axis2.jaxws.i18n.Messages;
import org.apache.axis2.jaxws.message.Message;
import org.apache.axis2.jaxws.spi.Binding;
import org.apache.axis2.jaxws.spi.BindingProvider;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import javax.activation.DataHandler;
import javax.activation.DataSource;
import javax.xml.ws.soap.MTOMFeature;
import java.io.InputStream;
import java.util.List;
/**
*
*/
public class MTOMConfigurator implements ClientConfigurator {
private static Log log = LogFactory.getLog(MTOMConfigurator.class);
/*
* (non-Javadoc)
* @see org.apache.axis2.jaxws.feature.util.WebServiceFeatureConfigurator#performConfiguration(org.apache.axis2.jaxws.core.MessageContext, org.apache.axis2.jaxws.spi.BindingProvider)
*/
public void configure(MessageContext messageContext, BindingProvider provider) {
Binding bnd = (Binding) provider.getBinding();
MTOMFeature mtomFeature = (MTOMFeature) bnd.getFeature(MTOMFeature.ID);
Message requestMsg = messageContext.getMessage();
//Disable MTOM.
requestMsg.setMTOMEnabled(false);
if (mtomFeature == null) {
throw ExceptionFactory.
makeWebServiceException(Messages.getMessage("mtomFeatureErr"));
}
//Enable MTOM if specified.
if (mtomFeature.isEnabled()) {
int threshold = mtomFeature.getThreshold();
List<String> attachmentIDs = requestMsg.getAttachmentIDs();
// Enable MTOM
requestMsg.setMTOMEnabled(true);
if (threshold <= 0) {
if (log.isDebugEnabled()) {
log.debug("Enabling MTOM with no threshold.");
}
}else{
if(log.isDebugEnabled()){
log.debug("MTOM Threshold Value ="+threshold);
}
//set MTOM threshold value on message context.
//Once MTOM threshold is set on message context it will be
//read by SOAPMessageFormatter.writeTo() while writing the attachment
//SOAPMessageFormatter will further propogate the threshold value to
//Axiom.OMOutputFormat. JAXBAttachmentUnmarshaller will then make
//decision if the attachment should be inlined or optimized.
messageContext.setProperty(Constants.Configuration.MTOM_THRESHOLD, new Integer(threshold));
}
}
else {
if (log.isDebugEnabled()) {
log.debug("The MTOMFeature was found, but not enabled.");
}
}
}
/*
* (non-Javadoc)
* @see org.apache.axis2.jaxws.feature.ClientConfigurator#supports(org.apache.axis2.jaxws.spi.Binding)
*/
public boolean supports(Binding binding) {
return binding instanceof SOAPBinding;
}
}