org.mule.security.oauth.DefaultRefreshTokenManager.java Source code

Introduction

Here is the source code for org.mule.security.oauth.DefaultRefreshTokenManager.java

Source

/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */

package org.mule.security.oauth;

import static org.mule.api.store.ObjectStoreManager.UNBOUNDED;

import org.mule.api.MuleContext;
import org.mule.api.context.MuleContextAware;
import org.mule.api.store.ObjectStore;
import org.mule.api.store.ObjectStoreManager;

import java.util.concurrent.locks.Lock;

import org.apache.commons.lang.StringUtils;

/**
 * Implementation of {@link RefreshTokenManager} that guarantees that no refresh
 * token is used more than once. If two threads try to refresh the same token
 * concurrently, only one will succeed and the other one will rely on the result of
 * the first one
 */
public class DefaultRefreshTokenManager implements MuleContextAware, RefreshTokenManager {

    private ObjectStore<Boolean> refreshedTokens;
    private MuleContext muleContext;
    private int minRefreshInterval = DEFAULT_MIN_REFRESH_INTERVAL;

    /**
     * {@inheritDoc} This implementation uses a lock to guarantee that no refresh
     * token is consumed more than once
     *
     * @see org.mule.security.oauth.RefreshTokenManager#refreshToken(org.mule.security.oauth.OAuth2Adapter,
     * java.lang.String)
     */
    @Override
    public void refreshToken(OAuth2Adapter adapter, String accessTokenId) throws Exception {
        if (StringUtils.isEmpty(accessTokenId)) {
            throw new IllegalArgumentException("Cannot refresh a blank accessTokenId");
        }

        String id = String.format("%s:%s:%s", this.getClass().getCanonicalName(), adapter.getName(), accessTokenId);
        Lock lock = this.muleContext.getLockFactory().createLock(id);
        lock.lock();
        try {
            if (!this.getRefreshedTokens().contains(id)) {
                adapter.refreshAccessToken(accessTokenId);
                this.getRefreshedTokens().store(id, true);
            }
        } finally {
            lock.unlock();
        }

    }

    private synchronized ObjectStore<Boolean> getRefreshedTokens() {
        if (this.refreshedTokens == null) {
            ObjectStoreManager osManager = this.muleContext.getObjectStoreManager();
            this.refreshedTokens = osManager.getObjectStore("RefreshTokenStore", false, UNBOUNDED,
                    this.minRefreshInterval, this.minRefreshInterval);
        }
        return this.refreshedTokens;
    }

    @Override
    public void setMuleContext(MuleContext context) {
        this.muleContext = context;
    }

    @Override
    public void setMinRefreshIntervalInMillis(int minRefreshIntervalInMillis) {
        this.minRefreshInterval = minRefreshIntervalInMillis;
    }
}