84 lines
2.2 KiB
Java
84 lines
2.2 KiB
Java
package ba.steleks.security.token;
|
|
|
|
import ba.steleks.model.UserRole;
|
|
import ba.steleks.storage.store.KeyValueStore;
|
|
import ba.steleks.util.CalendarUtils;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import java.util.Set;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
/**
|
|
* Created by ensar on 30/05/17.
|
|
*/
|
|
|
|
@Component
|
|
public class BasicInMemoryTokenStore implements TokenStore {
|
|
|
|
// Default one hour ttl
|
|
public static final long DEFAULT_TTL =
|
|
TimeUnit.MILLISECONDS.convert(1, TimeUnit.HOURS);
|
|
|
|
private static KeyValueStore<String, TokenInfo> tokenStore;
|
|
private long ttl = DEFAULT_TTL;
|
|
|
|
@Autowired
|
|
public BasicInMemoryTokenStore(KeyValueStore<String, TokenInfo> tokenStore) {
|
|
if(BasicInMemoryTokenStore.tokenStore == null) {
|
|
BasicInMemoryTokenStore.tokenStore = tokenStore;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean isValidToken(String token) {
|
|
if (tokenStore.contains(token)) {
|
|
// Find token in store
|
|
TokenInfo basicToken = tokenStore.get(token);
|
|
|
|
// Token is invalid, it has expired
|
|
if(basicToken.saveTime + ttl < CalendarUtils.getUTCCalendar().getTimeInMillis()) {
|
|
tokenStore.remove(token);
|
|
return false;
|
|
}
|
|
|
|
// Token valid!
|
|
return true;
|
|
} else {
|
|
// No token in store
|
|
return false;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Long getTokenInfo(String token) {
|
|
if (isValidToken(token)) {
|
|
// Find token in store
|
|
TokenInfo basicToken = tokenStore.get(token);
|
|
|
|
return basicToken.userId;
|
|
} else {
|
|
// No token in store
|
|
return null;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void saveToken(Long id, String token) {
|
|
TokenInfo tokenInfo = new TokenInfo();
|
|
tokenInfo.userId = id;
|
|
tokenInfo.saveTime = CalendarUtils.getUTCCalendar().getTimeInMillis();
|
|
tokenStore.save(token, tokenInfo);
|
|
}
|
|
|
|
@Override
|
|
public void removeToken(String token) {
|
|
tokenStore.remove(token);
|
|
}
|
|
|
|
private static class TokenInfo {
|
|
Long userId;
|
|
Long saveTime;
|
|
}
|
|
}
|