|
1 /* |
|
2 * $Revision $ |
|
3 * ==================================================================== |
|
4 * |
|
5 * Licensed to the Apache Software Foundation (ASF) under one or more |
|
6 * contributor license agreements. See the NOTICE file distributed with |
|
7 * this work for additional information regarding copyright ownership. |
|
8 * The ASF licenses this file to You under the Apache License, Version 2.0 |
|
9 * (the "License"); you may not use this file except in compliance with |
|
10 * the License. You may obtain a copy of the License at |
|
11 * |
|
12 * http://www.apache.org/licenses/LICENSE-2.0 |
|
13 * |
|
14 * Unless required by applicable law or agreed to in writing, software |
|
15 * distributed under the License is distributed on an "AS IS" BASIS, |
|
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
17 * See the License for the specific language governing permissions and |
|
18 * limitations 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 |
|
28 package ch.boye.httpclientandroidlib.conn; |
|
29 |
|
30 import java.io.InputStream; |
|
31 import java.io.IOException; |
|
32 |
|
33 import ch.boye.httpclientandroidlib.annotation.NotThreadSafe; |
|
34 |
|
35 /** |
|
36 * Basic implementation of {@link EofSensorWatcher}. The underlying connection |
|
37 * is released on close or EOF. |
|
38 * |
|
39 * @since 4.0 |
|
40 */ |
|
41 @NotThreadSafe |
|
42 public class BasicEofSensorWatcher implements EofSensorWatcher { |
|
43 |
|
44 /** The connection to auto-release. */ |
|
45 protected final ManagedClientConnection managedConn; |
|
46 |
|
47 /** Whether to keep the connection alive. */ |
|
48 protected final boolean attemptReuse; |
|
49 |
|
50 /** |
|
51 * Creates a new watcher for auto-releasing a connection. |
|
52 * |
|
53 * @param conn the connection to auto-release |
|
54 * @param reuse whether the connection should be re-used |
|
55 */ |
|
56 public BasicEofSensorWatcher(ManagedClientConnection conn, |
|
57 boolean reuse) { |
|
58 if (conn == null) |
|
59 throw new IllegalArgumentException |
|
60 ("Connection may not be null."); |
|
61 |
|
62 managedConn = conn; |
|
63 attemptReuse = reuse; |
|
64 } |
|
65 |
|
66 public boolean eofDetected(InputStream wrapped) |
|
67 throws IOException { |
|
68 |
|
69 try { |
|
70 if (attemptReuse) { |
|
71 // there may be some cleanup required, such as |
|
72 // reading trailers after the response body: |
|
73 wrapped.close(); |
|
74 managedConn.markReusable(); |
|
75 } |
|
76 } finally { |
|
77 managedConn.releaseConnection(); |
|
78 } |
|
79 return false; |
|
80 } |
|
81 |
|
82 public boolean streamClosed(InputStream wrapped) |
|
83 throws IOException { |
|
84 |
|
85 try { |
|
86 if (attemptReuse) { |
|
87 // this assumes that closing the stream will |
|
88 // consume the remainder of the response body: |
|
89 wrapped.close(); |
|
90 managedConn.markReusable(); |
|
91 } |
|
92 } finally { |
|
93 managedConn.releaseConnection(); |
|
94 } |
|
95 return false; |
|
96 } |
|
97 |
|
98 public boolean streamAbort(InputStream wrapped) |
|
99 throws IOException { |
|
100 |
|
101 managedConn.abortConnection(); |
|
102 return false; |
|
103 } |
|
104 |
|
105 } |