1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/parser/MexEntityResolver.java Wed Apr 27 01:27:09 2016 +0800
1.3 @@ -0,0 +1,80 @@
1.4 +/*
1.5 + * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
1.7 + *
1.8 + * This code is free software; you can redistribute it and/or modify it
1.9 + * under the terms of the GNU General Public License version 2 only, as
1.10 + * published by the Free Software Foundation. Oracle designates this
1.11 + * particular file as subject to the "Classpath" exception as provided
1.12 + * by Oracle in the LICENSE file that accompanied this code.
1.13 + *
1.14 + * This code is distributed in the hope that it will be useful, but WITHOUT
1.15 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
1.16 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
1.17 + * version 2 for more details (a copy is included in the LICENSE file that
1.18 + * accompanied this code).
1.19 + *
1.20 + * You should have received a copy of the GNU General Public License version
1.21 + * 2 along with this work; if not, write to the Free Software Foundation,
1.22 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
1.23 + *
1.24 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
1.25 + * or visit www.oracle.com if you need additional information or have any
1.26 + * questions.
1.27 + */
1.28 +
1.29 +package com.sun.xml.internal.ws.wsdl.parser;
1.30 +
1.31 +import com.sun.xml.internal.stream.buffer.XMLStreamBufferResult;
1.32 +import com.sun.xml.internal.ws.api.server.SDDocumentSource;
1.33 +import com.sun.xml.internal.ws.api.wsdl.parser.XMLEntityResolver;
1.34 +import com.sun.xml.internal.ws.util.JAXWSUtils;
1.35 +import com.sun.xml.internal.ws.util.xml.XmlUtil;
1.36 +import org.xml.sax.SAXException;
1.37 +
1.38 +import javax.xml.stream.XMLStreamException;
1.39 +import javax.xml.transform.Source;
1.40 +import javax.xml.transform.Transformer;
1.41 +import javax.xml.transform.TransformerException;
1.42 +import javax.xml.ws.WebServiceException;
1.43 +import java.io.IOException;
1.44 +import java.util.HashMap;
1.45 +import java.util.List;
1.46 +import java.util.Map;
1.47 +
1.48 +/**
1.49 + * Entity resolver that works on MEX Metadata
1.50 + *
1.51 + * @author Vivek Pandey
1.52 + */
1.53 +public final class MexEntityResolver implements XMLEntityResolver {
1.54 + private final Map<String, SDDocumentSource> wsdls = new HashMap<String, SDDocumentSource>();
1.55 +
1.56 + public MexEntityResolver(List<? extends Source> wsdls) throws IOException {
1.57 + Transformer transformer = XmlUtil.newTransformer();
1.58 + for (Source source : wsdls) {
1.59 + XMLStreamBufferResult xsbr = new XMLStreamBufferResult();
1.60 + try {
1.61 + transformer.transform(source, xsbr);
1.62 + } catch (TransformerException e) {
1.63 + throw new WebServiceException(e);
1.64 + }
1.65 + String systemId = source.getSystemId();
1.66 +
1.67 + //TODO: can we do anything if the given mex Source has no systemId?
1.68 + if(systemId != null){
1.69 + SDDocumentSource doc = SDDocumentSource.create(JAXWSUtils.getFileOrURL(systemId), xsbr.getXMLStreamBuffer());
1.70 + this.wsdls.put(systemId, doc);
1.71 + }
1.72 + }
1.73 + }
1.74 +
1.75 + public Parser resolveEntity(String publicId, String systemId) throws SAXException, IOException, XMLStreamException {
1.76 + if (systemId != null) {
1.77 + SDDocumentSource src = wsdls.get(systemId);
1.78 + if (src != null)
1.79 + return new Parser(src);
1.80 + }
1.81 + return null;
1.82 + }
1.83 +}
