001/* 002 * ==================================================================== 003 * Licensed to the Apache Software Foundation (ASF) under one 004 * or more contributor license agreements. See the NOTICE file 005 * distributed with this work for additional information 006 * regarding copyright ownership. The ASF licenses this file 007 * to you under the Apache License, Version 2.0 (the 008 * "License"); you may not use this file except in compliance 009 * with the License. You may obtain a copy of the License at 010 * 011 * http://www.apache.org/licenses/LICENSE-2.0 012 * 013 * Unless required by applicable law or agreed to in writing, 014 * software distributed under the License is distributed on an 015 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 016 * KIND, either express or implied. See the License for the 017 * specific language governing permissions and limitations 018 * under the License. 019 * ==================================================================== 020 * 021 * This software consists of voluntary contributions made by many 022 * individuals on behalf of the Apache Software Foundation. For more 023 * information on the Apache Software Foundation, please see 024 * <http://www.apache.org/>. 025 * 026 */ 027 028package org.apache.http.nio.protocol; 029 030import java.io.IOException; 031 032import org.apache.http.HttpEntity; 033import org.apache.http.HttpResponse; 034import org.apache.http.nio.ContentEncoder; 035import org.apache.http.nio.IOControl; 036import org.apache.http.nio.entity.EntityAsyncContentProducer; 037import org.apache.http.nio.entity.HttpAsyncContentProducer; 038import org.apache.http.protocol.HTTP; 039import org.apache.http.protocol.HttpContext; 040 041/** 042 * This class changed from package private to public in version 4.4.8. 043 * 044 * @since 4.4.8 045 */ 046public class ErrorResponseProducer implements HttpAsyncResponseProducer { 047 048 private final HttpResponse response; 049 private final HttpEntity entity; 050 private final HttpAsyncContentProducer contentProducer; 051 private final boolean keepAlive; 052 053 public ErrorResponseProducer( 054 final HttpResponse response, 055 final HttpEntity entity, 056 final boolean keepAlive) { 057 super(); 058 this.response = response; 059 this.entity = entity; 060 if (entity instanceof HttpAsyncContentProducer) { 061 this.contentProducer = (HttpAsyncContentProducer) entity; 062 } else { 063 this.contentProducer = new EntityAsyncContentProducer(entity); 064 } 065 this.keepAlive = keepAlive; 066 } 067 068 @Override 069 public HttpResponse generateResponse() { 070 if (this.keepAlive) { 071 response.addHeader(HTTP.CONN_DIRECTIVE, HTTP.CONN_KEEP_ALIVE); 072 } else { 073 response.addHeader(HTTP.CONN_DIRECTIVE, HTTP.CONN_CLOSE); 074 } 075 response.setEntity(this.entity); 076 return response; 077 } 078 079 @Override 080 public void produceContent( 081 final ContentEncoder encoder, final IOControl ioctrl) throws IOException { 082 this.contentProducer.produceContent(encoder, ioctrl); 083 } 084 085 @Override 086 public void responseCompleted(final HttpContext context) { 087 } 088 089 @Override 090 public void failed(final Exception ex) { 091 } 092 093 @Override 094 public void close() throws IOException { 095 this.contentProducer.close(); 096 } 097 098}