View Javadoc
1   /*
2    * Copyright 2008 Web Cohesion
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *   https://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  
17  package org.springframework.security.oauth.provider.token;
18  
19  import java.util.concurrent.ConcurrentHashMap;
20  
21  /**
22   * Implementation of TokenServices that stores tokens in memory.
23   *
24   * @author Ryan Heaton
25   */
26  public class InMemoryProviderTokenServices extends RandomValueProviderTokenServices {
27  
28    protected final ConcurrentHashMap<String, OAuthProviderTokenImpl> tokenStore = new ConcurrentHashMap<String, OAuthProviderTokenImpl>();
29  
30    protected OAuthProviderTokenImpl readToken(String token) {
31      return tokenStore.get(token);
32    }
33  
34    protected void storeToken(String tokenValue, OAuthProviderTokenImpl token) {
35      tokenStore.put(tokenValue, token);
36    }
37  
38    protected OAuthProviderTokenImpl removeToken(String tokenValue) {
39      return tokenStore.remove(tokenValue);
40    }
41  
42  }