RequestScope.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.jooby.internal;

import java.util.Map;

import com.google.inject.Key;
import com.google.inject.OutOfScopeException;
import com.google.inject.Provider;
import com.google.inject.Scope;
import com.google.inject.Scopes;

public class RequestScope implements Scope {

  private final ThreadLocal<Map<Object, Object>> scope = new ThreadLocal<>();

  public void enter(final Map<Object, Object> locals) {
    scope.set(locals);
  }

  public void exit() {
    scope.remove();
  }

  @SuppressWarnings({"unchecked", "rawtypes" })
  @Override
  public <T> Provider<T> scope(final Key<T> key, final Provider<T> unscoped) {
    return () -> {
      Map<Object, Object> scopedObjects = getScopedObjectMap(key);

      T current = (T) scopedObjects.get(key);
      if (current == null && !scopedObjects.containsKey(key)) {
        current = unscoped.get();

        // don't remember proxies; these exist only to serve circular dependencies
        if (Scopes.isCircularProxy(current)) {
          return current;
        }

        scopedObjects.put(key, current);
      }
      if (current instanceof javax.inject.Provider) {
        if (!javax.inject.Provider.class.isAssignableFrom(key.getTypeLiteral().getRawType())) {
          return (T) ((javax.inject.Provider) current).get();
        }
      }
      return current;
    };
  }

  private <T> Map<Object, Object> getScopedObjectMap(final Key<T> key) {
    Map<Object, Object> scopedObjects = scope.get();
    if (scopedObjects == null) {
      throw new OutOfScopeException("Cannot access " + key + " outside of a scoping block");
    }
    return scopedObjects;
  }

}