Java Code Examples for com.google.apphosting.api.ApiProxy#Environment

@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.
    }
}
 

@Override
public Future<byte[]> makeAsyncCall(
    ApiProxy.Environment environment,
    String packageName,
    String methodName,
    byte[] request,
    ApiProxy.ApiConfig apiConfig) {
  if ("remote_socket".equals(packageName)) {
    final byte[] response = makeResponse(methodName, request);
    return new Future<byte[]>() {
      @Override
      public boolean cancel(boolean mayInterruptIfRunning) {
        return false;
      }

      @Override
      public boolean isCancelled() {
        return false;
      }

      @Override
      public boolean isDone() {
        return true;
      }

      @Override
      public byte[] get() throws InterruptedException, ExecutionException {
        return response;
      }

      @Override
      public byte[] get(long timeout, TimeUnit unit)
          throws InterruptedException, ExecutionException, TimeoutException {
        return response;
      }
    };
  }
  throw new UnsupportedOperationException();
}
 

@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);
    }
  }
}
 

@Override
public Future<byte[]> makeAsyncCall(
    ApiProxy.Environment environment,
    String packageName,
    String methodName,
    byte[] request,
    ApiProxy.ApiConfig apiConfig) {
  if ("remote_socket".equals(packageName) && "Resolve".equals(methodName)) {
    return new Future<byte[]>() {
      @Override
      public boolean cancel(boolean mayInterruptIfRunning) {
        return false;
      }

      @Override
      public boolean isCancelled() {
        return false;
      }

      @Override
      public boolean isDone() {
        return true;
      }

      @Override
      public byte[] get() throws InterruptedException, ExecutionException {
        return RESOLVER_RESPONSE;
      }

      @Override
      public byte[] get(long timeout, TimeUnit unit)
          throws InterruptedException, ExecutionException, TimeoutException {
        return RESOLVER_RESPONSE;
      }
    };
  }
  throw new UnsupportedOperationException();
}
 

@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));
}
 

@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"));
}
 

@Override
public byte[] makeSyncCall(
    ApiProxy.Environment environment, String packageName, String methodName, byte[] request) {
  if (!"remote_socket".equals(packageName)) {
    throw new UnsupportedOperationException();
  }
  return makeResponse(methodName, request);
}
 

public Future<byte[]> makeAsyncCall(
    ApiProxy.Environment environment,
    String packageName,
    String methodName,
    byte[] request,
    ApiProxy.ApiConfig apiConfig) {
  if (packageName.equals("datastore_v3") && timeoutCount > 0) {
    timeoutCount--;
    throw new DatastoreTimeoutException("Timeout");
  }
  return delegate.makeAsyncCall(environment, packageName, methodName, request, apiConfig);
}
 

@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);
}
 

@Override
public byte[] makeSyncCall(
    ApiProxy.Environment environment, String packageName, String methodName, byte[] request) {
  if (!"remote_socket".equals(packageName)) {
    throw new UnsupportedOperationException();
  }
  return makeResponse(methodName, request);
}
 

/**
 * 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;
}
 

protected ApiProxy.Environment getBackgroundEnvironment() {
  return backgroundEnvironment;
}
 

@Override
public void log(ApiProxy.Environment environment, ApiProxy.LogRecord record) {}
 

@Override
public void flushLogs(ApiProxy.Environment environment) {}
 

@Override
public List<Thread> getRequestThreads(ApiProxy.Environment environment) {
  return null;
}
 

@Override
public void log(ApiProxy.Environment environment, ApiProxy.LogRecord record) {}
 

@Override
public ApiProxy.Environment newEnvironment() {
    defaultEnvironment.setThreadLocalAttributes();
    return defaultEnvironment;
}
 

@Override
public List<Thread> getRequestThreads(ApiProxy.Environment environment) {
  return null;
}
 

@SuppressWarnings("unchecked")
public void log(ApiProxy.Environment environment, LogRecord record) {
  delegate.log(environment, record);
}