com.github.zhengframework.rest.RequestScopeModule Maven / Gradle / Ivy

Go to download
package com.github.zhengframework.rest;

import com.google.inject.AbstractModule;
import com.google.inject.Provider;
import com.google.inject.servlet.RequestScoped;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.Request;
import javax.ws.rs.core.SecurityContext;
import javax.ws.rs.core.UriInfo;
import org.jboss.resteasy.core.ResteasyContext;

public class RequestScopeModule extends AbstractModule {

  @Override
  protected void configure() {
    bind(Request.class).toProvider(new ResteasyContextProvider<>(Request.class))
        .in(RequestScoped.class);
    bind(HttpHeaders.class).toProvider(new ResteasyContextProvider<>(HttpHeaders.class))
        .in(RequestScoped.class);
    bind(UriInfo.class).toProvider(new ResteasyContextProvider<>(UriInfo.class))
        .in(RequestScoped.class);
    bind(SecurityContext.class)
        .toProvider(new ResteasyContextProvider<>(SecurityContext.class))
        .in(RequestScoped.class);
  }

  private static class ResteasyContextProvider implements Provider {

    private final Class instanceClass;

    ResteasyContextProvider(final Class instanceClass) {
      this.instanceClass = instanceClass;
    }

    @Override
    public T get() {
      return ResteasyContext.getContextData(instanceClass);
    }
  }
}