View Javadoc
1   /*
2    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
3    * The software in this package is published under the terms of the CPAL v1.0
4    * license, a copy of which has been included with this distribution in the
5    * LICENSE.txt file.
6    */
7   package org.mule.transport.http.transformers;
8   
9   import org.mule.RequestContext;
10  import org.mule.api.transformer.TransformerException;
11  import org.mule.api.transport.OutputHandler;
12  import org.mule.transformer.AbstractTransformer;
13  import org.mule.transformer.types.DataTypeFactory;
14  import org.mule.transport.http.HttpConstants;
15  import org.mule.transport.http.HttpResponse;
16  import org.mule.transport.http.ResponseWriter;
17  
18  import java.io.IOException;
19  import java.io.OutputStream;
20  import java.io.UnsupportedEncodingException;
21  import java.util.Iterator;
22  
23  import org.apache.commons.httpclient.ChunkedOutputStream;
24  import org.apache.commons.httpclient.Header;
25  import org.apache.commons.io.output.ByteArrayOutputStream;
26  
27  /**
28   * Converts an Http Response object to String. Note that the response headers are
29   * preserved.
30   */
31  public class HttpResponseToString extends AbstractTransformer
32  {
33  
34      public HttpResponseToString()
35      {
36          registerSourceType(DataTypeFactory.create(HttpResponse.class));
37          setReturnDataType(DataTypeFactory.STRING);
38      }
39  
40      /**
41       * Perform the transformation to always return a String object
42       */
43      @Override
44      protected Object doTransform(Object src, String encoding) throws TransformerException
45      {
46          try
47          {
48              HttpResponse response = (HttpResponse)src;
49              ByteArrayOutputStream bos = new ByteArrayOutputStream(8192);
50              OutputStream outstream = bos;
51              ResponseWriter writer = new ResponseWriter(outstream, encoding);
52              writer.println(response.getStatusLine());
53              Iterator item = response.getHeaderIterator();
54              while (item.hasNext())
55              {
56                  Header header = (Header)item.next();
57                  writer.print(header.toExternalForm());
58              }
59              writer.println();
60              writer.flush();
61  
62              if (response.hasBody())
63              {
64                  OutputHandler handler = response.getBody();
65                  Header transferenc = response.getFirstHeader(HttpConstants.HEADER_TRANSFER_ENCODING);
66                  if (transferenc != null)
67                  {
68                      response.removeHeaders(HttpConstants.HEADER_CONTENT_LENGTH);
69                      if (transferenc.getValue().indexOf(HttpConstants.TRANSFER_ENCODING_CHUNKED) != -1)
70                      {
71                          outstream = new ChunkedOutputStream(outstream);
72                      }
73                  }
74  
75                  handler.write(RequestContext.getEvent(), outstream);
76  
77                  if (outstream instanceof ChunkedOutputStream)
78                  {
79                      ((ChunkedOutputStream)outstream).finish();
80                  }
81              }
82  
83              outstream.flush();
84              bos.flush();
85              byte[] result = bos.toByteArray();
86              outstream.close();
87              writer.close();
88              bos.close();
89  
90              String output = null;
91              try
92              {
93                  output = new String(result, encoding);
94              }
95              catch (UnsupportedEncodingException uee)
96              {
97                  // I believe this is never reached since a TransformerExcpetion
98                  // is thrown before at new ResponseWriter(outstream, encoding) if
99                  // encoding is not supported
100                 output = new String(result);
101             }
102 
103             return output;
104         }
105         catch (IOException e)
106         {
107             throw new TransformerException(this, e);
108         }
109     }
110 }