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