be.objectify.deadbolt.java.cache.DefaultSubjectCache Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012-2015 Steve Chaloner
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package be.objectify.deadbolt.java.cache;

import be.objectify.deadbolt.core.models.Subject;
import be.objectify.deadbolt.java.ConfigKeys;
import be.objectify.deadbolt.java.DeadboltHandler;
import play.Configuration;
import play.libs.F;
import play.mvc.Http;

import javax.inject.Inject;
import javax.inject.Singleton;
import java.util.Optional;
import java.util.concurrent.TimeUnit;

/**
 * @author Steve Chaloner ([email protected])
 */
@Singleton
public class DefaultSubjectCache implements SubjectCache
{
    private final boolean cacheUserPerRequestEnabled;

    private final boolean blocking;

    private final long blockingTimeout;

    @Inject
    public DefaultSubjectCache(final Configuration configuration)
    {
        this.cacheUserPerRequestEnabled = configuration.getBoolean(ConfigKeys.CACHE_DEADBOLT_USER_DEFAULT._1,
                                                                   ConfigKeys.CACHE_DEADBOLT_USER_DEFAULT._2);
        this.blocking = configuration.getBoolean(ConfigKeys.BLOCKING_DEFAULT._1,
                                                 ConfigKeys.BLOCKING_DEFAULT._2);

        this.blockingTimeout = configuration.getLong(ConfigKeys.DEFAULT_BLOCKING_TIMEOUT_DEFAULT._1,
                                                     ConfigKeys.DEFAULT_BLOCKING_TIMEOUT_DEFAULT._2);
    }

    @Override
    public F.Promise> apply(final DeadboltHandler deadboltHandler,
                                              final Http.Context context)
    {
        F.Promise> promise;
        if (cacheUserPerRequestEnabled)
        {
            final Optional cachedUser = Optional.ofNullable((Subject) context.args.get(ConfigKeys.CACHE_DEADBOLT_USER_DEFAULT._1));
            if (cachedUser.isPresent())
            {
                promise = F.Promise.pure(cachedUser);
            }
            else
            {
                promise = deadboltHandler.getSubject(context)
                                   .map(subjectOption -> {
                                       subjectOption.ifPresent(subject -> context.args.put(ConfigKeys.CACHE_DEADBOLT_USER_DEFAULT._1,
                                                                                           subject));
                                       return subjectOption;
                                   });
            }
        }
        else
        {
            promise = deadboltHandler.getSubject(context);
        }
        
        if (this.blocking)
        {
            promise = F.Promise.pure(promise.get(this.blockingTimeout,
                                                 TimeUnit.MILLISECONDS));
        }
        
        return promise;
    }
}