src/share/jaxws_classes/com/sun/xml/internal/ws/message/DataHandlerAttachment.java

Fri, 14 Feb 2014 11:13:45 +0100

author
mkos
date
Fri, 14 Feb 2014 11:13:45 +0100
changeset 515
6cd506508147
parent 368
0989ad8c0860
child 637
9c07ef4934dd
permissions
-rw-r--r--

8026188: Enhance envelope factory
Summary: Avoiding caching data initialized via TCCL in static context; fix also reviewed by Alexander Fomin
Reviewed-by: ahgross, mgrebac, skoivu

ohair@286 1 /*
alanb@368 2 * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
ohair@286 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
ohair@286 4 *
ohair@286 5 * This code is free software; you can redistribute it and/or modify it
ohair@286 6 * under the terms of the GNU General Public License version 2 only, as
ohair@286 7 * published by the Free Software Foundation. Oracle designates this
ohair@286 8 * particular file as subject to the "Classpath" exception as provided
ohair@286 9 * by Oracle in the LICENSE file that accompanied this code.
ohair@286 10 *
ohair@286 11 * This code is distributed in the hope that it will be useful, but WITHOUT
ohair@286 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
ohair@286 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
ohair@286 14 * version 2 for more details (a copy is included in the LICENSE file that
ohair@286 15 * accompanied this code).
ohair@286 16 *
ohair@286 17 * You should have received a copy of the GNU General Public License version
ohair@286 18 * 2 along with this work; if not, write to the Free Software Foundation,
ohair@286 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
ohair@286 20 *
ohair@286 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@286 22 * or visit www.oracle.com if you need additional information or have any
ohair@286 23 * questions.
ohair@286 24 */
ohair@286 25
ohair@286 26 package com.sun.xml.internal.ws.message;
ohair@286 27
ohair@286 28 import com.sun.istack.internal.NotNull;
ohair@286 29 import com.sun.xml.internal.ws.api.message.Attachment;
ohair@286 30 import com.sun.xml.internal.ws.util.ByteArrayBuffer;
ohair@286 31
ohair@286 32 import javax.activation.DataHandler;
ohair@286 33 import javax.xml.soap.AttachmentPart;
ohair@286 34 import javax.xml.soap.SOAPException;
ohair@286 35 import javax.xml.soap.SOAPMessage;
ohair@286 36 import javax.xml.transform.Source;
ohair@286 37 import javax.xml.transform.stream.StreamSource;
ohair@286 38 import javax.xml.ws.WebServiceException;
ohair@286 39 import java.io.IOException;
ohair@286 40 import java.io.InputStream;
ohair@286 41 import java.io.OutputStream;
ohair@286 42 import java.io.ByteArrayOutputStream;
ohair@286 43
ohair@286 44 /**
ohair@286 45 * @author Jitendra Kotamraju
ohair@286 46 */
ohair@286 47 public final class DataHandlerAttachment implements Attachment {
ohair@286 48
ohair@286 49 private final DataHandler dh;
ohair@286 50 private final String contentId;
alanb@368 51 String contentIdNoAngleBracket;
ohair@286 52
ohair@286 53 /**
alanb@368 54 * This will be constructed by {@link com.sun.xml.internal.ws.message.jaxb.AttachmentMarshallerImpl}
ohair@286 55 */
ohair@286 56 public DataHandlerAttachment(@NotNull String contentId, @NotNull DataHandler dh) {
ohair@286 57 this.dh = dh;
ohair@286 58 this.contentId = contentId;
ohair@286 59 }
ohair@286 60
ohair@286 61 public String getContentId() {
alanb@368 62 // return contentId;
alanb@368 63 if (contentIdNoAngleBracket == null) {
alanb@368 64 contentIdNoAngleBracket = contentId;
alanb@368 65 if (contentIdNoAngleBracket != null && contentIdNoAngleBracket.charAt(0) == '<')
alanb@368 66 contentIdNoAngleBracket = contentIdNoAngleBracket.substring(1, contentIdNoAngleBracket.length()-1);
alanb@368 67 }
alanb@368 68 return contentIdNoAngleBracket;
ohair@286 69 }
ohair@286 70
ohair@286 71 public String getContentType() {
ohair@286 72 return dh.getContentType();
ohair@286 73 }
ohair@286 74
ohair@286 75 public byte[] asByteArray() {
ohair@286 76 try {
ohair@286 77 ByteArrayOutputStream os = new ByteArrayOutputStream();
ohair@286 78 dh.writeTo(os);
ohair@286 79 return os.toByteArray();
ohair@286 80 } catch (IOException e) {
ohair@286 81 throw new WebServiceException(e);
ohair@286 82 }
ohair@286 83 }
ohair@286 84
ohair@286 85 public DataHandler asDataHandler() {
ohair@286 86 return dh;
ohair@286 87 }
ohair@286 88
ohair@286 89 public Source asSource() {
ohair@286 90 try {
ohair@286 91 return new StreamSource(dh.getInputStream());
ohair@286 92 } catch (IOException e) {
ohair@286 93 throw new WebServiceException(e);
ohair@286 94 }
ohair@286 95 }
ohair@286 96
ohair@286 97 public InputStream asInputStream() {
ohair@286 98 try {
ohair@286 99 return dh.getInputStream();
ohair@286 100 } catch (IOException e) {
ohair@286 101 throw new WebServiceException(e);
ohair@286 102 }
ohair@286 103 }
ohair@286 104
ohair@286 105 public void writeTo(OutputStream os) throws IOException {
ohair@286 106 dh.writeTo(os);
ohair@286 107 }
ohair@286 108
ohair@286 109 public void writeTo(SOAPMessage saaj) throws SOAPException {
ohair@286 110 AttachmentPart part = saaj.createAttachmentPart();
ohair@286 111 part.setDataHandler(dh);
ohair@286 112 part.setContentId(contentId);
ohair@286 113 saaj.addAttachmentPart(part);
ohair@286 114 }
ohair@286 115 }

mercurial