com.hp.autonomy.hod.client.token.InMemoryTokenRepository Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-hod-client Show documentation

Java Client for communicating with HP Haven OnDemand

There is a newer version: 0.25.3

/*
 * Copyright 2015-2016 Hewlett-Packard Development Company, L.P.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 */

package com.hp.autonomy.hod.client.token;

import com.hp.autonomy.hod.client.api.authentication.AuthenticationToken;
import com.hp.autonomy.hod.client.api.authentication.EntityType;
import com.hp.autonomy.hod.client.api.authentication.TokenType;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

/**
 * A {@link TokenRepository} which is backed by a {@link java.util.Map}
 *
 * This class is thread safe
 *
 * This class does not expire old tokens
 */
public class InMemoryTokenRepository implements TokenRepository {

    private final ConcurrentMap, AuthenticationToken> map = new ConcurrentHashMap<>();

    @Override
    public  TokenProxy insert(final AuthenticationToken token) {
        checkTokenExpiry(token);

        final TokenProxy key = new TokenProxy<>(token.getEntityType(), token.getTokenType());

        map.put(key, token);

        return key;
    }

    @Override
    public  AuthenticationToken update(final TokenProxy key, final AuthenticationToken newToken) {
        checkTokenExpiry(newToken);

        // we only put matching pairs into the map
        //noinspection unchecked
        return (AuthenticationToken) map.replace(key, newToken);
    }

    @Override
    public  AuthenticationToken get(final TokenProxy key) {
        // we only put matching pairs into the map
        //noinspection unchecked
        return (AuthenticationToken) map.get(key);
    }

    @Override
    public  AuthenticationToken remove(final TokenProxy key) {
        // we only put matching pairs into the map
        //noinspection unchecked
        return (AuthenticationToken) map.remove(key);
    }

    private void checkTokenExpiry(final AuthenticationToken token) {
        if (token.hasExpired()) {
            throw new IllegalArgumentException("Token has already expired");
        }
    }

}