/*
* 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.dataretrieval;
import org.apache.axiom.om.OMElement;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.description.AxisService;
import org.apache.axis2.description.AxisService2WSDL11;
import org.apache.axis2.AxisFault;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
* Axis 2 Data Locator responsibles for retrieving WSDL metadata.
*/
public class WSDLDataLocator extends BaseAxisDataLocator implements AxisDataLocator {
private static final Log log = LogFactory.getLog(WSDLDataLocator.class);
String serviceURL = null;
AxisService theService = null;
String request_Identifier = null;
protected WSDLDataLocator() {
}
/**
* Constructor
*
* @param data an array of ServiceData instance defined in the
* ServiceData.xml for the WSDL dialect.
*/
protected WSDLDataLocator(ServiceData[] data) {
dataList = data;
}
/**
* getData API
* Implement data retrieval logic for WSDL dialect
*/
public Data[] getData(DataRetrievalRequest request,
MessageContext msgContext) throws DataRetrievalException {
log.trace("Default WSDL DataLocator getData starts");
request_Identifier = request.getIdentifier();
serviceURL = msgContext.getTo().getAddress();
OutputForm outputform = request.getOutputForm();
if (outputform == null) { // not defined, defualt to inline
outputform = OutputForm.INLINE_FORM;
}
Data[] output;
String outputFormString = outputform.getType();
if (outputform == OutputForm.INLINE_FORM) {
output = outputInlineForm(msgContext, dataList);
} else if (outputform == OutputForm.LOCATION_FORM) {
output = outputLocationForm(dataList);
} else if (outputform == OutputForm.REFERENCE_FORM) {
output = outputReferenceForm(msgContext, dataList);
} else {
output = outputInlineForm(msgContext, dataList);
}
if (output == null) {
if (log.isTraceEnabled()) {
log.trace(
"Null data return! Data Locator does not know how to handle request for dialect= " +
request.getDialect() + " in the form of " + outputFormString);
}
}
log.trace("Default WSDL DataLocator getData ends");
return output;
}
/*
* (non-Javadoc)
* @see org.apache.axis2.dataretrieval.BaseAxisDataLocator#outputInlineForm(org.apache.axis2.context.MessageContext, org.apache.axis2.dataretrieval.ServiceData[])
*/
protected Data[] outputInlineForm(MessageContext msgContext, ServiceData[] dataList)
throws DataRetrievalException {
Data[] result = super.outputInlineForm(msgContext, dataList);
// Do not generate WSDL if Identifier was specified in the request as
// (1) this is to support ?wsdl request;
// (2) Data for specified Identifier must be available to satisfy the GetMetadata request.
if (result.length == 0) {
log.trace("Default WSDL DataLocator attempt to generates WSDL.");
if (msgContext != null) {
theService = msgContext.getAxisService();
serviceURL = msgContext.getTo().getAddress();
} else {
throw new DataRetrievalException("MessageContext was not set!");
}
if (request_Identifier == null || request_Identifier.equals(theService.getTargetNamespace())) {
AxisService2WSDL11 axisService2WOM;
OMElement wsdlElement;
try {
axisService2WOM = new AxisService2WSDL11(theService);
wsdlElement = axisService2WOM.generateOM();
} catch (Exception e) {
log.debug(e);
throw new DataRetrievalException(e);
}
if (wsdlElement != null) {
log.trace("Default WSDL DataLocator successfully generated WSDL.");
result = new Data[1];
result[0] = new Data(wsdlElement, null);
}
}
}
return result;
}
/*
*
*/
protected Data[] outputLocationForm(ServiceData[] serviceData) throws DataRetrievalException {
Data[] result = super.outputLocationForm(serviceData);
// Do not generate URL if Identifier was specified in the request as
// (1) Axis2 ?wsdl URL request is not supporting Identifier;
// (2) URL data for specified Identifier must be available to satisfy
// the GetMetadata request.
if (result.length == 0 && request_Identifier == null) {
result = new Data[1];
result[0] = new Data(serviceURL + "?wsdl", null);
}
return result;
}
}