View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  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,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.eclipse.aether.transport.http;
20  
21  import java.io.Closeable;
22  
23  import org.apache.http.client.protocol.HttpClientContext;
24  import org.apache.http.protocol.BasicHttpContext;
25  
26  /**
27   * HTTP context that shares certain attributes among requests to optimize the communication with the server.
28   *
29   * @see <a href="http://hc.apache.org/httpcomponents-client-ga/tutorial/html/advanced.html#stateful_conn">Stateful HTTP
30   *      connections</a>
31   */
32  final class SharingHttpContext extends BasicHttpContext implements Closeable {
33  
34      private final LocalState state;
35  
36      private final SharingAuthCache authCache;
37  
38      SharingHttpContext(LocalState state) {
39          this.state = state;
40          authCache = new SharingAuthCache(state);
41          super.setAttribute(HttpClientContext.AUTH_CACHE, authCache);
42      }
43  
44      @Override
45      public Object getAttribute(String id) {
46          if (HttpClientContext.USER_TOKEN.equals(id)) {
47              return state.getUserToken();
48          }
49          return super.getAttribute(id);
50      }
51  
52      @Override
53      public void setAttribute(String id, Object obj) {
54          if (HttpClientContext.USER_TOKEN.equals(id)) {
55              state.setUserToken(obj);
56          } else {
57              super.setAttribute(id, obj);
58          }
59      }
60  
61      @Override
62      public Object removeAttribute(String id) {
63          if (HttpClientContext.USER_TOKEN.equals(id)) {
64              state.setUserToken(null);
65              return null;
66          }
67          return super.removeAttribute(id);
68      }
69  
70      @Override
71      public void close() {
72          authCache.clear();
73      }
74  }