1 /* 2 * ==================================================================== 3 * Licensed to the Apache Software Foundation (ASF) under one 4 * or more contributor license agreements. See the NOTICE file 5 * distributed with this work for additional information 6 * regarding copyright ownership. The ASF licenses this file 7 * to you under the Apache License, Version 2.0 (the 8 * "License"); you may not use this file except in compliance 9 * with the License. You may obtain a copy of the License at 10 * 11 * http://www.apache.org/licenses/LICENSE-2.0 12 * 13 * Unless required by applicable law or agreed to in writing, 14 * software distributed under the License is distributed on an 15 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 16 * KIND, either express or implied. See the License for the 17 * specific language governing permissions and limitations 18 * under the License. 19 * ==================================================================== 20 * 21 * This software consists of voluntary contributions made by many 22 * individuals on behalf of the Apache Software Foundation. For more 23 * information on the Apache Software Foundation, please see 24 * <http://www.apache.org/>. 25 * 26 */ 27 package org.apache.hc.client5.http.fluent; 28 29 import java.io.IOException; 30 31 import org.apache.hc.client5.http.HttpResponseException; 32 import org.apache.hc.client5.http.impl.classic.AbstractHttpClientResponseHandler; 33 import org.apache.hc.core5.http.ClassicHttpResponse; 34 import org.apache.hc.core5.http.ContentType; 35 import org.apache.hc.core5.http.HttpEntity; 36 import org.apache.hc.core5.http.io.entity.EntityUtils; 37 38 /** 39 * {@link org.apache.hc.core5.http.io.HttpClientResponseHandler} implementation 40 * that converts {@link org.apache.hc.core5.http.HttpResponse} messages 41 * to {@link Content} instances. 42 * 43 * @see Content 44 * @since 4.4 45 */ 46 public class ContentResponseHandler extends AbstractHttpClientResponseHandler<Content> { 47 48 49 /** 50 * The maximum length of the exception message, to avoid excessive memory usage. 51 */ 52 private static final int MAX_MESSAGE_LENGTH = 256; 53 54 @Override 55 public Content handleEntity(final HttpEntity entity) throws IOException { 56 return entity != null ? 57 new Content(EntityUtils.toByteArray(entity), ContentType.parse(entity.getContentType())) : 58 Content.NO_CONTENT; 59 } 60 61 /** 62 * Handles a successful response (2xx status code) and returns the response entity as a {@link Content} object. 63 * If no response entity exists, {@link Content#NO_CONTENT} is returned. 64 * 65 * @param response the HTTP response. 66 * @return a {@link Content} object that encapsulates the response body, or {@link Content#NO_CONTENT} if the 67 * response body is {@code null} or has zero length. 68 * @throws HttpResponseException if the response was unsuccessful (status code greater than 300). 69 * @throws IOException if an I/O error occurs. 70 */ 71 @Override 72 public Content handleResponse(final ClassicHttpResponse response) throws IOException { 73 final int statusCode = response.getCode(); 74 final HttpEntity entity = response.getEntity(); 75 final byte[] contentBytes = (entity != null) ? EntityUtils.toByteArray(entity, MAX_MESSAGE_LENGTH) : new byte[0]; 76 final ContentType contentType = (entity != null && entity.getContentType() != null) ? ContentType.parse(entity.getContentType()) : ContentType.DEFAULT_BINARY; 77 final Content content = new Content(contentBytes, contentType); 78 if (statusCode >= 300) { 79 throw new HttpResponseException(statusCode, response.getReasonPhrase(), contentBytes, contentType); 80 } 81 return content; 82 } 83 }