diff -r 88b85470e72c -r f50545b5e2f1 src/share/jaxws_classes/com/sun/xml/internal/ws/message/ByteArrayAttachment.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/message/ByteArrayAttachment.java Tue Mar 06 16:09:35 2012 -0800 @@ -0,0 +1,110 @@ +/* + * Copyright (c) 1997, 2010, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ + +package com.sun.xml.internal.ws.message; + +import com.sun.istack.internal.NotNull; +import com.sun.xml.internal.ws.api.message.Attachment; +import com.sun.xml.internal.ws.util.ByteArrayDataSource; +import com.sun.xml.internal.ws.encoding.DataSourceStreamingDataHandler; + +import java.io.ByteArrayInputStream; + +import javax.activation.DataHandler; +import javax.xml.soap.AttachmentPart; +import javax.xml.soap.SOAPException; +import javax.xml.soap.SOAPMessage; +import javax.xml.transform.Source; +import javax.xml.transform.stream.StreamSource; +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; + +/** + * @author Jitendra Kotamraju + */ +public final class ByteArrayAttachment implements Attachment { + + private final String contentId; + private byte[] data; + private int start; + private final int len; + private final String mimeType; + + public ByteArrayAttachment(@NotNull String contentId, byte[] data, int start, int len, String mimeType) { + this.contentId = contentId; + this.data = data; + this.start = start; + this.len = len; + this.mimeType = mimeType; + } + + public ByteArrayAttachment(@NotNull String contentId, byte[] data, String mimeType) { + this(contentId, data, 0, data.length, mimeType); + } + + public String getContentId() { + return contentId; + } + + public String getContentType() { + return mimeType; + } + + public byte[] asByteArray() { + if(start!=0 || len!=data.length) { + // if our buffer isn't exact, switch to the exact one + byte[] exact = new byte[len]; + System.arraycopy(data,start,exact,0,len); + start = 0; + data = exact; + } + return data; + } + + public DataHandler asDataHandler() { + return new DataSourceStreamingDataHandler(new ByteArrayDataSource(data,start,len,getContentType())); + } + + public Source asSource() { + return new StreamSource(asInputStream()); + } + + public InputStream asInputStream() { + return new ByteArrayInputStream(data,start,len); + } + + public void writeTo(OutputStream os) throws IOException { + os.write(asByteArray()); + } + + public void writeTo(SOAPMessage saaj) throws SOAPException { + AttachmentPart part = saaj.createAttachmentPart(); + part.setDataHandler(asDataHandler()); + part.setContentId(contentId); + saaj.addAttachmentPart(part); + } + +}