Java Code Examples for com.google.apphosting.api.ApiProxy#getCurrentEnvironment()

@Test
public void testGetDefaultGcsBucketName() {
    ApiProxy.Environment env = ApiProxy.getCurrentEnvironment();
    String expectedBucketName;
    Property property = property("testGetDefaultGcsBucketName");
    if (property.exists()) {
        expectedBucketName = property.getPropertyValue();
    } else {
        expectedBucketName = (String) env.getAttributes().get("com.google.appengine.runtime.default_version_hostname");
    }

    try {
        String bucketName = appIdentity.getDefaultGcsBucketName();
        Assert.assertEquals(expectedBucketName, bucketName);
    } catch (AppIdentityServiceFailureException aisfe) {
        //TODO: This means that there is no default bucket setup for this project.  Have a better way to verify this.
    }
}
 

/**
 * Runs calls in a background thread so that the results will actually be asynchronous.
 *
 * @see com.google.appengine.tools.cloudstorage.RawGcsService#continueObjectCreationAsync(
 *        com.google.appengine.tools.cloudstorage.RawGcsService.RawGcsCreationToken,
 *        java.nio.ByteBuffer, long)
 */
@Override
public Future<RawGcsCreationToken> continueObjectCreationAsync(final RawGcsCreationToken token,
    final ByteBuffer chunk, long timeoutMillis) {
  try {
    ensureInitialized();
  } catch (IOException e) {
    throw new RuntimeException(e);
  }
  final Environment environment = ApiProxy.getCurrentEnvironment();
  return writePool.schedule(new Callable<RawGcsCreationToken>() {
    @Override
    public RawGcsCreationToken call() throws Exception {
      ApiProxy.setEnvironmentForCurrentThread(environment);
      return append(token, chunk);
    }
  }, 50, TimeUnit.MILLISECONDS);
}
 

@Override
public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
  resp.setContentType("text/plain");
  ApiProxy.Environment env = ApiProxy.getCurrentEnvironment();
  resp.getWriter().print("default_version_hostname: ");
  resp.getWriter()
      .println(env.getAttributes().get("com.google.appengine.runtime.default_version_hostname"));
}
 

@Test
public void testRequestId() {
    ApiProxy.Environment env = ApiProxy.getCurrentEnvironment();
    String requestId = (String) env.getAttributes().get("com.google.appengine.runtime.request_log_id");
    String errMsg = "The request id should not be null";
    Assert.assertNotNull(errMsg, requestId);
}
 

@Test
public void testInstanceId() {
    ApiProxy.Environment env = ApiProxy.getCurrentEnvironment();
    String instanceId = (String) env.getAttributes().get("com.google.appengine.instance.id");
    String errMsg = "The instance id should not be null";
    Assert.assertNotNull(errMsg, instanceId);
}
 

@Test
public void testGetVersionedHostname() {
    String expectedHostname = getExpectedAppHostname("testGetVersionedHostname");
    ApiProxy.Environment env = ApiProxy.getCurrentEnvironment();
    String hostname = (String) env.getAttributes().get("com.google.appengine.runtime.default_version_hostname");
    String errMsg = "The versioned hostname should end with " + expectedHostname + ", but was " + hostname;
    Assert.assertTrue(errMsg, hostname.endsWith(expectedHostname));
}
 

@Test
public void testParseFullAppId() {
    // [(partition)~][(domain):](display-app-id)
    ApiProxy.Environment env = ApiProxy.getCurrentEnvironment();
    String hostname = (String) env.getAttributes().get("com.google.appengine.runtime.default_version_hostname");
    AppIdentityService.ParsedAppId parsed = appIdentity.parseFullAppId(hostname);

    String message = createParsed(parsed);
    Assert.assertEquals(message, property("testParseFullAppId_partition").getPropertyValue(), parsed.getPartition());
    Assert.assertEquals(message, getExpectedAppHostname("testParseFullAppId_domain"), parsed.getDomain());
    Assert.assertEquals(message, getExpectedAppId("testParseFullAppId_id"), parsed.getId());
}
 

@Override
protected void doStart() throws Exception {
    this.helper = createHelper();
    this.environment = ApiProxy.getCurrentEnvironment();
    getServletContext().setAttribute(API_PROXY_LOCAL, ApiProxy.getDelegate());
    getServletContext().setAttribute(APPENGINE_WEB_XML, readAppengineWebXml());
    getServletContext().setAttribute(WEB_XML, readWebXml());
    SystemProperty.environment.set(SystemProperty.Environment.Value.Development);
    configureUserRealmAppengineHelper();
    super.doStart();
}
 

SessionData loadSession(String sessionId) {
  String key = SESSION_PREFIX + sessionId;

  SessionData data = null;
  for (SessionStore sessionStore : sessionStoresInReadOrder) {
    // Keep iterating until we find a store that has the session data we
    // want.
    try {
      data = sessionStore.getSession(key);
      if (data != null) {
        break;
      }
    } catch (RuntimeException e) {
      String msg = "Exception while loading session data";
      logger.log(Level.WARNING, msg, e);
      if (ApiProxy.getCurrentEnvironment() != null) {
        ApiProxy.log(createWarningLogRecord(msg, e));
      }
      break;
    }
  }
  if (data != null) {
    if (System.currentTimeMillis() > data.getExpirationTime()) {
      logger.fine("Session " + sessionId + " expired " +
                  ((System.currentTimeMillis() - data.getExpirationTime()) / 1000) +
                  " seconds ago, ignoring.");
      return null;
    }
  }
  return data;
}
 

/**
 * Returns the thread local environment if it is a VmApiProxyEnvironment.
 *
 * @return The ThreadLocal environment or null if no VmApiProxyEnvironment is set.
 */
private VmApiProxyEnvironment getThreadLocalEnvironment() {
  Environment env = ApiProxy.getCurrentEnvironment();
  if (env instanceof VmApiProxyEnvironment) {
    return (VmApiProxyEnvironment) env;
  }
  return null;
}
 

/**
 * Returns the thread local environment if it is a VmApiProxyEnvironment.
 *
 * @return The ThreadLocal environment or null if no VmApiProxyEnvironment is set on this thread.
 */
private VmApiProxyEnvironment getThreadLocalEnvironment() {
  Environment env = ApiProxy.getCurrentEnvironment();
  if (env instanceof VmApiProxyEnvironment) {
    return (VmApiProxyEnvironment) env;
  }
  return null;
}
 

private String fullVersionId() {
  ApiProxy.Environment environment = ApiProxy.getCurrentEnvironment();
  String actualVersionId = environment.getVersionId();
  if (!(environment.getModuleId() == null
        || environment.getModuleId().isEmpty()
        || environment.getModuleId().equals("default"))) {
    actualVersionId = environment.getModuleId() + ":" + actualVersionId;
  }
  return actualVersionId;
}
 

@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp)
    throws ServletException, IOException {
  resp.setContentType("text/plain");
  ApiProxy.Environment requestEnv = ApiProxy.getCurrentEnvironment();

  for (String key : requestEnv.getAttributes().keySet()) {
    if (!INIT_ENV.getAttributes().containsKey(key) && !REQUEST_ONLY_ATTRIBUTES.contains(key)) {
      resp.getWriter().println("Init environment attributes do not contain " + key);
    }
  }
}
 

public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
  resp.setContentType("text/plain");
  Environment env = ApiProxy.getCurrentEnvironment();

  for (Entry entry : env.getAttributes().entrySet()) {
    resp.getWriter().println(entry.getKey() + " => " + entry.getValue());
  }
}
 

/**
 * Returns the GAE's internal request ID of a request. This is not related to HttpServletRequest.
 *
 * @see <a href="https://cloud.google.com/appengine/docs/standard/java/how-requests-are-handled">https://cloud.google.com/appengine/docs/standard/java/how-requests-are-handled</a>
 */
public static String getRequestId() {
    ApiProxy.Environment serverEnvironment = ApiProxy.getCurrentEnvironment();
    if (serverEnvironment == null) {
        // This will be the case in dev server
        return "dummyrequestid";
    }
    return String.valueOf(ApiProxy.getCurrentEnvironment().getAttributes()
            .get("com.google.appengine.runtime.request_log_id"));
}
 

private static String readAppUrl() {
    ApiProxy.Environment serverEnvironment = ApiProxy.getCurrentEnvironment();
    if (serverEnvironment == null) {
        return null;
    }
    String hostname = (String) serverEnvironment.getAttributes()
            .get("com.google.appengine.runtime.default_version_hostname");
    if (hostname == null) {
        return null;
    }
    return (isDevServer() ? "http://" : "https://") + hostname;
}
 

public static String getInfo() {
  String version = SystemProperty.applicationVersion.get();
  String majorVersion = version.substring(0, version.indexOf('.'));
  Environment env = ApiProxy.getCurrentEnvironment();
  String hostname =
      "" + env.getAttributes().get("com.google.appengine.runtime.default_version_hostname");
  String infostring = "version: " + majorVersion + " and hostname: " + hostname;
  return infostring;
}
 

@Override
public void setUp() throws Exception {
  super.setUp();
  traceBuffer = new StringBuffer();
  helper.setUp();
  apiProxyEnvironment = ApiProxy.getCurrentEnvironment();
  System.setProperty(USE_SIMPLE_GUIDS_FOR_DEBUGGING, "true");
  taskQueue = LocalTaskQueueTestConfig.getLocalTaskQueue();
}
 

public AppEngineEnvironment() {
  isPlaceHolderNeeded = ApiProxy.getCurrentEnvironment() == null;
  // isPlaceHolderNeeded may be true when we are invoked in a test with AppEngineRule.
  if (isPlaceHolderNeeded) {
    ApiProxy.setEnvironmentForCurrentThread(PLACEHOLDER_ENV);
  }
}
 

protected static URL getUrl(String path) throws MalformedURLException {
    ApiProxy.Environment env = ApiProxy.getCurrentEnvironment();
    Object localhost = env.getAttributes().get("com.google.appengine.runtime.default_version_hostname");
    return new URL("http://" + localhost + "/" + path);
}