1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
package org.mule.providers.http.transformers; |
12 | |
|
13 | |
import org.mule.MuleManager; |
14 | |
import org.mule.config.MuleProperties; |
15 | |
import org.mule.providers.NullPayload; |
16 | |
import org.mule.providers.http.HttpConnector; |
17 | |
import org.mule.providers.http.HttpConstants; |
18 | |
import org.mule.providers.http.i18n.HttpMessages; |
19 | |
import org.mule.transformers.AbstractEventAwareTransformer; |
20 | |
import org.mule.transformers.simple.SerializableToByteArray; |
21 | |
import org.mule.umo.UMOEventContext; |
22 | |
import org.mule.umo.UMOMessage; |
23 | |
import org.mule.umo.transformer.TransformerException; |
24 | |
import org.mule.util.StringUtils; |
25 | |
|
26 | |
import java.io.InputStream; |
27 | |
import java.net.URI; |
28 | |
import java.util.Iterator; |
29 | |
|
30 | |
import org.apache.commons.httpclient.HttpMethod; |
31 | |
import org.apache.commons.httpclient.HttpVersion; |
32 | |
import org.apache.commons.httpclient.methods.ByteArrayRequestEntity; |
33 | |
import org.apache.commons.httpclient.methods.GetMethod; |
34 | |
import org.apache.commons.httpclient.methods.InputStreamRequestEntity; |
35 | |
import org.apache.commons.httpclient.methods.PostMethod; |
36 | |
import org.apache.commons.httpclient.methods.StringRequestEntity; |
37 | |
import org.apache.commons.httpclient.params.HttpMethodParams; |
38 | |
|
39 | |
|
40 | |
|
41 | |
|
42 | |
|
43 | |
|
44 | |
public class ObjectToHttpClientMethodRequest extends AbstractEventAwareTransformer |
45 | |
{ |
46 | |
private final SerializableToByteArray serializableToByteArray; |
47 | |
|
48 | |
public ObjectToHttpClientMethodRequest() |
49 | 0 | { |
50 | 0 | setReturnClass(HttpMethod.class); |
51 | 0 | serializableToByteArray = new SerializableToByteArray(); |
52 | 0 | } |
53 | |
|
54 | |
private int addParameters(String queryString, PostMethod postMethod) |
55 | |
{ |
56 | |
|
57 | |
|
58 | |
|
59 | 0 | if (StringUtils.isEmpty(queryString)) |
60 | |
{ |
61 | 0 | return 0; |
62 | |
} |
63 | |
|
64 | |
String currentParam; |
65 | |
int equals; |
66 | 0 | equals = queryString.indexOf("&"); |
67 | 0 | if (equals > -1) |
68 | |
{ |
69 | 0 | currentParam = queryString.substring(0, equals); |
70 | 0 | queryString = queryString.substring(equals + 1); |
71 | |
} |
72 | |
else |
73 | |
{ |
74 | 0 | currentParam = queryString; |
75 | 0 | queryString = StringUtils.EMPTY; |
76 | |
} |
77 | 0 | int parameterIndex = -1; |
78 | 0 | while (StringUtils.isNotBlank(currentParam)) |
79 | |
{ |
80 | |
String paramName, paramValue; |
81 | 0 | equals = currentParam.indexOf("="); |
82 | 0 | if (equals > -1) |
83 | |
{ |
84 | 0 | paramName = currentParam.substring(0, equals); |
85 | 0 | paramValue = currentParam.substring(equals + 1); |
86 | 0 | parameterIndex++; |
87 | 0 | postMethod.addParameter(paramName, paramValue); |
88 | |
} |
89 | 0 | equals = queryString.indexOf("&"); |
90 | 0 | if (equals > -1) |
91 | |
{ |
92 | 0 | currentParam = queryString.substring(0, equals); |
93 | 0 | queryString = queryString.substring(equals + 1); |
94 | |
} |
95 | |
else |
96 | |
{ |
97 | 0 | currentParam = queryString; |
98 | 0 | queryString = StringUtils.EMPTY; |
99 | |
} |
100 | |
} |
101 | 0 | return parameterIndex + 1; |
102 | |
} |
103 | |
|
104 | |
public Object transform(Object src, String encoding, UMOEventContext context) throws TransformerException |
105 | |
{ |
106 | 0 | UMOMessage msg = context.getMessage(); |
107 | |
|
108 | 0 | String endpoint = msg.getStringProperty(MuleProperties.MULE_ENDPOINT_PROPERTY, null); |
109 | 0 | if (endpoint == null) |
110 | |
{ |
111 | 0 | throw new TransformerException( |
112 | |
HttpMessages.eventPropertyNotSetCannotProcessRequest( |
113 | |
MuleProperties.MULE_ENDPOINT_PROPERTY), this); |
114 | |
} |
115 | |
|
116 | 0 | String method = msg.getStringProperty(HttpConnector.HTTP_METHOD_PROPERTY, "POST"); |
117 | |
try |
118 | |
{ |
119 | 0 | URI uri = new URI(endpoint); |
120 | 0 | HttpMethod httpMethod = null; |
121 | |
|
122 | 0 | if (HttpConstants.METHOD_GET.equals(method)) |
123 | |
{ |
124 | 0 | httpMethod = new GetMethod(uri.toString()); |
125 | 0 | setHeaders(httpMethod, context); |
126 | 0 | String paramName = msg.getStringProperty(HttpConnector.HTTP_GET_BODY_PARAM_PROPERTY, |
127 | |
HttpConnector.DEFAULT_HTTP_GET_BODY_PARAM_PROPERTY); |
128 | 0 | String query = uri.getQuery(); |
129 | 0 | if (!(src instanceof NullPayload) && !StringUtils.EMPTY.equals(src)) |
130 | |
{ |
131 | 0 | if (query == null) |
132 | |
{ |
133 | 0 | query = paramName + "=" + src.toString(); |
134 | |
} |
135 | |
else |
136 | |
{ |
137 | 0 | query += "&" + paramName + "=" + src.toString(); |
138 | |
} |
139 | |
} |
140 | 0 | httpMethod.setQueryString(query); |
141 | |
|
142 | |
} |
143 | |
else |
144 | |
{ |
145 | 0 | PostMethod postMethod = new PostMethod(uri.toString()); |
146 | 0 | setHeaders(postMethod, context); |
147 | 0 | String paramName = msg.getStringProperty(HttpConnector.HTTP_POST_BODY_PARAM_PROPERTY, null); |
148 | |
|
149 | 0 | if (paramName == null) |
150 | |
{ |
151 | |
|
152 | 0 | addParameters(uri.getQuery(), postMethod); |
153 | |
|
154 | |
|
155 | |
|
156 | 0 | if (!(context.getMessage().getPayload() instanceof NullPayload)) |
157 | |
{ |
158 | |
|
159 | 0 | String mimeType = msg.getStringProperty(HttpConstants.HEADER_CONTENT_TYPE, null); |
160 | |
|
161 | 0 | if (src instanceof String) |
162 | |
{ |
163 | |
|
164 | |
|
165 | 0 | if (mimeType != null) |
166 | |
{ |
167 | 0 | int parameterIndex = mimeType.indexOf(";"); |
168 | 0 | if (parameterIndex > 0) |
169 | |
{ |
170 | 0 | mimeType = mimeType.substring(0, parameterIndex); |
171 | |
} |
172 | |
} |
173 | 0 | if (mimeType == null) mimeType = HttpConstants.DEFAULT_CONTENT_TYPE; |
174 | 0 | if (encoding == null) encoding = MuleManager.getConfiguration().getEncoding(); |
175 | 0 | postMethod.setRequestEntity(new StringRequestEntity(src.toString(), mimeType, |
176 | |
encoding)); |
177 | |
} |
178 | 0 | else if (src instanceof InputStream) |
179 | |
{ |
180 | |
|
181 | |
|
182 | 0 | if (mimeType == null) mimeType = HttpConstants.DEFAULT_CONTENT_TYPE; |
183 | 0 | postMethod.setRequestEntity(new InputStreamRequestEntity((InputStream)src, |
184 | |
mimeType)); |
185 | |
} |
186 | |
else |
187 | |
{ |
188 | |
|
189 | |
|
190 | 0 | if (mimeType == null) mimeType = HttpConstants.DEFAULT_CONTENT_TYPE; |
191 | 0 | byte[] buffer = (byte[])serializableToByteArray.doTransform(src, encoding); |
192 | 0 | postMethod.setRequestEntity(new ByteArrayRequestEntity(buffer, mimeType)); |
193 | |
} |
194 | |
} |
195 | |
} |
196 | |
else |
197 | |
{ |
198 | 0 | postMethod.addParameter(paramName, src.toString()); |
199 | |
} |
200 | |
|
201 | 0 | httpMethod = postMethod; |
202 | |
} |
203 | |
|
204 | |
|
205 | 0 | HttpMethodParams params = (HttpMethodParams)msg.removeProperty(HttpConnector.HTTP_PARAMS_PROPERTY); |
206 | 0 | if(params!=null) |
207 | |
{ |
208 | 0 | httpMethod.setParams(params); |
209 | |
} |
210 | |
else |
211 | |
{ |
212 | |
|
213 | 0 | String httpVersion = msg.getStringProperty(HttpConnector.HTTP_VERSION_PROPERTY, HttpConstants.HTTP11); |
214 | 0 | if(HttpConstants.HTTP10.equals(httpVersion)) |
215 | |
{ |
216 | 0 | httpMethod.getParams().setVersion(HttpVersion.HTTP_1_0); |
217 | |
} |
218 | |
else |
219 | |
{ |
220 | 0 | httpMethod.getParams().setVersion(HttpVersion.HTTP_1_1); |
221 | |
} |
222 | |
} |
223 | 0 | return httpMethod; |
224 | |
} |
225 | 0 | catch (Exception e) |
226 | |
{ |
227 | 0 | throw new TransformerException(this, e); |
228 | |
} |
229 | |
} |
230 | |
|
231 | |
protected void setHeaders(HttpMethod httpMethod, UMOEventContext context) |
232 | |
{ |
233 | |
|
234 | |
String headerValue; |
235 | |
String headerName; |
236 | 0 | UMOMessage msg = context.getMessage(); |
237 | 0 | for (Iterator iterator = msg.getPropertyNames().iterator(); iterator.hasNext();) |
238 | |
{ |
239 | 0 | headerName = (String)iterator.next(); |
240 | 0 | headerValue = msg.getStringProperty(headerName, null); |
241 | 0 | if (HttpConstants.REQUEST_HEADER_NAMES.get(headerName) == null) |
242 | |
{ |
243 | 0 | if (headerName.startsWith(MuleProperties.PROPERTY_PREFIX)) |
244 | |
{ |
245 | 0 | headerName = new StringBuffer(30).append("X-").append(headerName).toString(); |
246 | |
} |
247 | |
|
248 | |
|
249 | |
|
250 | |
|
251 | 0 | if (headerName.startsWith(HttpConstants.HEADER_CONTENT_LENGTH)) |
252 | |
{ |
253 | 0 | if (httpMethod.getResponseHeader(HttpConstants.HEADER_CONTENT_LENGTH) == null) |
254 | |
{ |
255 | 0 | httpMethod.addRequestHeader(headerName, headerValue); |
256 | |
} |
257 | |
} |
258 | |
else |
259 | |
{ |
260 | 0 | httpMethod.addRequestHeader(headerName, headerValue); |
261 | |
} |
262 | |
} |
263 | |
} |
264 | |
|
265 | 0 | if (context.getMessage().getPayload() instanceof InputStream) |
266 | |
{ |
267 | |
|
268 | 0 | httpMethod.addRequestHeader(HttpConstants.HEADER_CONTENT_TYPE, "multipart/related"); |
269 | |
} |
270 | 0 | } |
271 | |
} |