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  
25  /**
26   * HTTP context that shares certain attributes among requests to optimize the communication with the server.
27   *
28   * @see <a href="http://hc.apache.org/httpcomponents-client-ga/tutorial/html/advanced.html#stateful_conn">Stateful HTTP
29   *      connections</a>
30   */
31  final class SharingHttpContext extends HttpClientContext implements Closeable {
32  
33      private final LocalState state;
34  
35      private final SharingAuthCache authCache;
36  
37      SharingHttpContext(LocalState state) {
38          this.state = state;
39          authCache = new SharingAuthCache(state);
40          super.setAttribute(HttpClientContext.AUTH_CACHE, authCache);
41      }
42  
43      @Override
44      public Object getAttribute(String id) {
45          if (HttpClientContext.USER_TOKEN.equals(id)) {
46              return state.getUserToken();
47          }
48          return super.getAttribute(id);
49      }
50  
51      @Override
52      public void setAttribute(String id, Object obj) {
53          if (HttpClientContext.USER_TOKEN.equals(id)) {
54              state.setUserToken(obj);
55          } else {
56              super.setAttribute(id, obj);
57          }
58      }
59  
60      @Override
61      public Object removeAttribute(String id) {
62          if (HttpClientContext.USER_TOKEN.equals(id)) {
63              state.setUserToken(null);
64              return null;
65          }
66          return super.removeAttribute(id);
67      }
68  
69      @Override
70      public void close() {
71          authCache.clear();
72      }
73  }