1/*
2 * Copyright (c) 1997, 2013, 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
26package com.sun.xml.internal.ws.message;
27
28import com.sun.xml.internal.ws.api.message.AttachmentSet;
29import com.sun.xml.internal.ws.api.message.Attachment;
30import com.sun.xml.internal.ws.encoding.MimeMultipartParser;
31import com.sun.xml.internal.ws.resources.EncodingMessages;
32import com.sun.istack.internal.Nullable;
33
34import javax.xml.ws.WebServiceException;
35import java.util.Iterator;
36import java.util.Map;
37import java.util.HashMap;
38import java.io.IOException;
39
40/**
41 * {@link AttachmentSet} backed by {@link com.sun.xml.internal.ws.encoding.MimeMultipartParser}
42 *
43 * @author Vivek Pandey
44 */
45public final class MimeAttachmentSet implements AttachmentSet {
46    private final MimeMultipartParser mpp;
47    private Map<String, Attachment> atts = new HashMap<String, Attachment>();
48
49
50    public MimeAttachmentSet(MimeMultipartParser mpp) {
51        this.mpp = mpp;
52    }
53
54    @Nullable
55    public Attachment get(String contentId) {
56        Attachment att;
57        /**
58         * First try to get the Attachment from internal map, maybe this attachment
59         * is added by the user.
60         */
61        att = atts.get(contentId);
62        if(att != null)
63            return att;
64        try {
65            /**
66             * Attachment is not found in the internal map, now do look in
67             * the mpp, if found add to the internal Attachment map.
68             */
69            att = mpp.getAttachmentPart(contentId);
70            if(att != null){
71                atts.put(contentId, att);
72            }
73        } catch (IOException e) {
74            throw new WebServiceException(EncodingMessages.NO_SUCH_CONTENT_ID(contentId), e);
75        }
76        return att;
77    }
78
79    /**
80     * This is expensive operation, its going to to read all the underlying
81     * attachments in {@link MimeMultipartParser}.
82     */
83    public boolean isEmpty() {
84        return atts.size() <= 0 && mpp.getAttachmentParts().isEmpty();
85    }
86
87    public void add(Attachment att) {
88        atts.put(att.getContentId(), att);
89    }
90
91    /**
92     * Expensive operation.
93     */
94    public Iterator<Attachment> iterator() {
95        /**
96         * Browse thru all the attachments in the mpp, add them to #atts,
97         * then return whether its empty.
98         */
99        Map<String, Attachment> attachments = mpp.getAttachmentParts();
100        for(Map.Entry<String, Attachment> att : attachments.entrySet()) {
101            if(atts.get(att.getKey()) == null){
102                atts.put(att.getKey(), att.getValue());
103            }
104        }
105
106        return atts.values().iterator();
107    }
108}
109