001 /** 002 * Licensed to the Apache Software Foundation (ASF) under one or more 003 * contributor license agreements. See the NOTICE file distributed with 004 * this work for additional information regarding copyright ownership. 005 * The ASF licenses this file to You under the Apache License, Version 2.0 006 * (the "License"); you may not use this file except in compliance with 007 * the License. You may obtain a copy of the License at 008 * 009 * http://www.apache.org/licenses/LICENSE-2.0 010 * 011 * Unless required by applicable law or agreed to in writing, software 012 * distributed under the License is distributed on an "AS IS" BASIS, 013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 014 * See the License for the specific language governing permissions and 015 * limitations under the License. 016 */ 017 package org.apache.camel.impl; 018 019 import java.util.HashMap; 020 import java.util.Map; 021 import java.util.Set; 022 023 import javax.activation.DataHandler; 024 025 import org.apache.camel.Message; 026 027 /** 028 * The default implementation of {@link Message} 029 * 030 * @version $Revision: 37863 $ 031 */ 032 public class DefaultMessage extends MessageSupport { 033 private Map<String, Object> headers; 034 private Map<String, DataHandler> attachments; 035 036 @Override 037 public String toString() { 038 return "Message: " + getBody(); 039 } 040 041 public Object getHeader(String name) { 042 return getHeaders().get(name); 043 } 044 045 public <T> T getHeader(String name, Class<T> type) { 046 Object value = getHeader(name); 047 return getExchange().getContext().getTypeConverter().convertTo(type, value); 048 } 049 050 public void setHeader(String name, Object value) { 051 if (headers == null) { 052 headers = createHeaders(); 053 } 054 headers.put(name, value); 055 } 056 057 public Object removeHeader(String name) { 058 if (headers != null) { 059 return headers.remove(name); 060 } else { 061 return null; 062 } 063 } 064 065 public Map<String, Object> getHeaders() { 066 if (headers == null) { 067 headers = createHeaders(); 068 } 069 return headers; 070 } 071 072 public void setHeaders(Map<String, Object> headers) { 073 this.headers = headers; 074 } 075 076 public DefaultMessage newInstance() { 077 return new DefaultMessage(); 078 } 079 080 /** 081 * A factory method to lazily create the headers to make it easy to create 082 * efficient Message implementations which only construct and populate the 083 * Map on demand 084 * 085 * @return return a newly constructed Map possibly containing headers from 086 * the underlying inbound transport 087 */ 088 protected Map<String, Object> createHeaders() { 089 HashMap<String, Object> map = new HashMap<String, Object>(); 090 populateInitialHeaders(map); 091 return map; 092 } 093 094 /** 095 * A strategy method populate the initial set of headers on an inbound 096 * message from an underlying binding 097 * 098 * @param map is the empty header map to populate 099 */ 100 protected void populateInitialHeaders(Map<String, Object> map) { 101 } 102 103 /** 104 * A factory method to lazily create the attachments to make it easy to 105 * create efficient Message implementations which only construct and 106 * populate the Map on demand 107 * 108 * @return return a newly constructed Map 109 */ 110 protected Map<String, DataHandler> createAttachments() { 111 HashMap<String, DataHandler> map = new HashMap<String, DataHandler>(); 112 populateInitialAttachments(map); 113 return map; 114 } 115 116 /** 117 * A strategy method populate the initial set of attachments on an inbound 118 * message from an underlying binding 119 * 120 * @param map is the empty attachment map to populate 121 */ 122 protected void populateInitialAttachments(Map<String, DataHandler> map) { 123 } 124 125 /* (non-Javadoc) 126 * @see org.apache.camel.Message#addAttachment(java.lang.String, javax.activation.DataHandler) 127 */ 128 public void addAttachment(String id, DataHandler content) { 129 if (attachments == null) { 130 attachments = createAttachments(); 131 } 132 attachments.put(id, content); 133 } 134 135 /* (non-Javadoc) 136 * @see org.apache.camel.Message#getAttachment(java.lang.String) 137 */ 138 public DataHandler getAttachment(String id) { 139 return getAttachments().get(id); 140 } 141 142 /* (non-Javadoc) 143 * @see org.apache.camel.Message#getAttachmentNames() 144 */ 145 public Set<String> getAttachmentNames() { 146 if (attachments == null) { 147 attachments = createAttachments(); 148 } 149 return attachments.keySet(); 150 } 151 152 /* (non-Javadoc) 153 * @see org.apache.camel.Message#removeAttachment(java.lang.String) 154 */ 155 public void removeAttachment(String id) { 156 if (attachments != null && attachments.containsKey(id)) { 157 attachments.remove(id); 158 } 159 } 160 161 /* (non-Javadoc) 162 * @see org.apache.camel.Message#getAttachments() 163 */ 164 public Map<String, DataHandler> getAttachments() { 165 if (attachments == null) { 166 attachments = createAttachments(); 167 } 168 return attachments; 169 } 170 171 /* (non-Javadoc) 172 * @see org.apache.camel.Message#setAttachments(java.util.Map) 173 */ 174 public void setAttachments(Map<String, DataHandler> attachments) { 175 this.attachments = attachments; 176 } 177 }