1   /*
2    * $HeadURL: https://svn.apache.org/repos/asf/httpcomponents/oac.hc3x/trunk/src/test/org/apache/commons/httpclient/server/SimpleConnManager.java $
3    * $Revision: 1425331 $
4    * $Date: 2012-12-22 19:29:41 +0100 (Sat, 22 Dec 2012) $
5    *
6    * ====================================================================
7    *
8    *  Licensed to the Apache Software Foundation (ASF) under one or more
9    *  contributor license agreements.  See the NOTICE file distributed with
10   *  this work for additional information regarding copyright ownership.
11   *  The ASF licenses this file to You under the Apache License, Version 2.0
12   *  (the "License"); you may not use this file except in compliance with
13   *  the License.  You may obtain a copy of the License at
14   *
15   *      http://www.apache.org/licenses/LICENSE-2.0
16   *
17   *  Unless required by applicable law or agreed to in writing, software
18   *  distributed under the License is distributed on an "AS IS" BASIS,
19   *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
20   *  See the License for the specific language governing permissions and
21   *  limitations under the License.
22   * ====================================================================
23   *
24   * This software consists of voluntary contributions made by many
25   * individuals on behalf of the Apache Software Foundation.  For more
26   * information on the Apache Software Foundation, please see
27   * <http://www.apache.org/>.
28   *
29   */
30  
31  package org.apache.commons.httpclient.server;
32  
33  import java.io.IOException;
34  import java.net.Socket;
35  import java.util.HashMap;
36  import java.util.Iterator;
37  import java.util.Map;
38  
39  /***
40   * A REALLY simple connection manager.
41   * 
42   * @author Oleg Kalnichevski
43   */
44  public class SimpleConnManager {
45      
46      private Map connsets = new HashMap();
47      
48      public SimpleConnManager() {
49          super();
50      }
51  
52      public synchronized SimpleHttpServerConnection openConnection(final SimpleHost host)
53          throws IOException {
54          if (host == null) {
55              throw new IllegalArgumentException("Host may not be null");
56          }
57          SimpleHttpServerConnection conn = null;
58          SimpleConnList connlist = (SimpleConnList)this.connsets.get(host);
59          if (connlist != null) {
60              conn = connlist.removeFirst();
61              if (conn != null && !conn.isOpen()) {
62                  conn = null;
63              }
64          }
65          if (conn == null) {
66              Socket socket = new Socket(host.getHostName(), host.getPort());
67              conn = new SimpleHttpServerConnection(socket);
68          }
69          return conn;
70      }
71      
72      public synchronized void releaseConnection(final SimpleHost host, 
73              final SimpleHttpServerConnection conn) throws IOException {
74          if (host == null) {
75              throw new IllegalArgumentException("Host may not be null");
76          }
77          if (conn == null) {
78              return;
79          }
80          if (!conn.isKeepAlive()) {
81              conn.close();
82          }
83          if (conn.isOpen()) {
84              SimpleConnList connlist = (SimpleConnList)this.connsets.get(host);
85              if (connlist == null) {
86                  connlist = new SimpleConnList();
87                  this.connsets.put(host, connlist);
88              }
89              connlist.addConnection(conn);
90          }
91      }
92  
93      public synchronized void shutdown() {
94          for (Iterator i = this.connsets.values().iterator(); i.hasNext();) {
95              SimpleConnList connlist = (SimpleConnList) i.next();
96              connlist.shutdown();
97          }
98          this.connsets.clear();
99      }
100 
101 }