LiRuochen_WorkFlow/MaxKey
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v1.2.1 GA

Browse Source
This commit is contained in:
shimingxy
2020-03-25 01:14:02 +08:00
parent 4163acf289
commit 7e1fb358da
3 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
maxkey-protocols/maxkey-protocol-cas/src/main/java/org/maxkey/authz/cas/endpoint/ticket/service/InMemoryTicketServices.java
View File

@@ -1,22 +1,30 @@
package org.maxkey.authz.cas.endpoint.ticket.service;
import java.util.concurrent.ConcurrentHashMap;
import java.time.Duration;
import org.ehcache.UserManagedCache;
import org.ehcache.config.builders.ExpiryPolicyBuilder;
import org.ehcache.config.builders.UserManagedCacheBuilder;
import org.maxkey.authz.cas.endpoint.ticket.Ticket;
public class InMemoryTicketServices extends RandomServiceTicketServices {
protected final static ConcurrentHashMap<String, Ticket> authorizationTicketStore = new ConcurrentHashMap<String, Ticket>();
protected final static UserManagedCache<String, Ticket> casTicketStore =
UserManagedCacheBuilder.newUserManagedCacheBuilder(String.class, Ticket.class)
.withExpiry(ExpiryPolicyBuilder.timeToLiveExpiration(Duration.ofSeconds(60)))
.build(true);
@Override
protected void store(String ticketId, Ticket ticket) {
this.authorizationTicketStore.put(ticketId, ticket);
casTicketStore.put(ticketId, ticket);
}
@Override
public Ticket remove(String ticketId) {
Ticket ticket = this.authorizationTicketStore.remove(ticketId);
Ticket ticket=casTicketStore.get(ticketId);
casTicketStore.remove(ticketId);
return ticket;
}

17
maxkey-protocols/maxkey-protocol-oauth-2.0/src/main/java/org/maxkey/authz/oauth2/provider/code/InMemoryAuthorizationCodeServices.java
View File

@@ -1,7 +1,10 @@
package org.maxkey.authz.oauth2.provider.code;
import java.util.concurrent.ConcurrentHashMap;
import java.time.Duration;
import org.ehcache.UserManagedCache;
import org.ehcache.config.builders.ExpiryPolicyBuilder;
import org.ehcache.config.builders.UserManagedCacheBuilder;
import org.maxkey.authz.oauth2.provider.OAuth2Authentication;
/**
@@ -11,17 +14,19 @@ import org.maxkey.authz.oauth2.provider.OAuth2Authentication;
* @author Dave Syer
*/
public class InMemoryAuthorizationCodeServices extends RandomValueAuthorizationCodeServices {
protected final ConcurrentHashMap<String, OAuth2Authentication> authorizationCodeStore = new ConcurrentHashMap<String, OAuth2Authentication>();
protected final static UserManagedCache<String, OAuth2Authentication> authorizationCodeStore =
UserManagedCacheBuilder.newUserManagedCacheBuilder(String.class, OAuth2Authentication.class)
.withExpiry(ExpiryPolicyBuilder.timeToLiveExpiration(Duration.ofSeconds(60)))
.build(true);
@Override
protected void store(String code, OAuth2Authentication authentication) {
this.authorizationCodeStore.put(code, authentication);
authorizationCodeStore.put(code, authentication);
}
@Override
public OAuth2Authentication remove(String code) {
OAuth2Authentication auth = this.authorizationCodeStore.remove(code);
OAuth2Authentication auth = authorizationCodeStore.get(code);
authorizationCodeStore.remove(code);
return auth;
}