1 /*
2  * JBoss, Home of Professional Open Source.
3  * Copyright 2014 Red Hat, Inc., and individual contributors
4  * as indicated by the @author tags.
5  *
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *     http://www.apache.org/licenses/LICENSE-2.0
11  *
12  *  Unless required by applicable law or agreed to in writing, software
13  *  distributed under the License is distributed on an "AS IS" BASIS,
14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  *  See the License for the specific language governing permissions and
16  *  limitations under the License.
17  */
18
19 package io.undertow.server.protocol.http;
20
21 import io.undertow.conduits.AbstractFixedLengthStreamSinkConduit;
22 import io.undertow.server.Connectors;
23 import io.undertow.server.HttpServerExchange;
24 import org.xnio.conduits.StreamSinkConduit;
25
26 /**
27  * @author Stuart Douglas
28  */
29 public class ServerFixedLengthStreamSinkConduit extends AbstractFixedLengthStreamSinkConduit {
30
31     private HttpServerExchange exchange;
32
33     /**
34      * Construct a new instance.
35      *
36      * @param next           the next channel
37      * @param configurable   {@code true} if this instance should pass configuration to the next
38      * @param propagateClose {@code true} if this instance should pass close to the next
39      */
40     public ServerFixedLengthStreamSinkConduit(StreamSinkConduit next, boolean configurable, boolean propagateClose) {
41         super(next, 1, configurable, propagateClose);
42     }
43
44     void reset(long contentLength, HttpServerExchange exchange) {
45         this.exchange = exchange;
46         super.reset(contentLength, !exchange.isPersistent());
47     }
48
49     void clearExchange(){
50         channelFinished();
51     }
52
53     @Override
54     protected void channelFinished() {
55         if(exchange != null) {
56             HttpServerExchange exchange = this.exchange;
57             this.exchange = null;
58             Connectors.terminateResponse(exchange);
59         }
60     }
61 }
62