From c02fe27836b24f29be0fef075ab41dccff34780c Mon Sep 17 00:00:00 2001
From: aozarov <aozarov@gmail.com>
Date: Mon, 24 Aug 2015 17:17:20 -0700
Subject: [PATCH] Support running against gcloud datastore emulator (support
 DATASTORE_HOST)

---
 .../src/main/java/com/google/gcloud/ServiceOptions.java    | 6 +++++-
 .../java/com/google/gcloud/datastore/DatastoreOptions.java | 7 +++++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/gcloud-java-core/src/main/java/com/google/gcloud/ServiceOptions.java b/gcloud-java-core/src/main/java/com/google/gcloud/ServiceOptions.java
index 59c2d57fb3df..206e1fecaa58 100644
--- a/gcloud-java-core/src/main/java/com/google/gcloud/ServiceOptions.java
+++ b/gcloud-java-core/src/main/java/com/google/gcloud/ServiceOptions.java
@@ -154,7 +154,7 @@ public B serviceRpcFactory(ServiceRpcFactory<ServiceRpcT, OptionsT> serviceRpcFa
 
   protected ServiceOptions(Builder<ServiceRpcT, OptionsT, ?> builder) {
     projectId = checkNotNull(builder.projectId != null ? builder.projectId : defaultProject());
-    host = firstNonNull(builder.host, DEFAULT_HOST);
+    host = firstNonNull(builder.host, defaultHost());
     httpTransportFactory =
         firstNonNull(builder.httpTransportFactory, DefaultHttpTransportFactory.INSTANCE);
     authCredentials = firstNonNull(builder.authCredentials, defaultAuthCredentials());
@@ -191,6 +191,10 @@ protected static String appEngineAppId() {
     return System.getProperty("com.google.appengine.application.id");
   }
 
+  protected String defaultHost() {
+    return DEFAULT_HOST;
+  }
+
   protected String defaultProject() {
     String projectId = System.getProperty(PROJECT_ENV_NAME, System.getenv(PROJECT_ENV_NAME));
     if (projectId == null) {
diff --git a/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/DatastoreOptions.java b/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/DatastoreOptions.java
index c670bb5d9a2d..7b0dd3a2a606 100644
--- a/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/DatastoreOptions.java
+++ b/gcloud-java-datastore/src/main/java/com/google/gcloud/datastore/DatastoreOptions.java
@@ -38,6 +38,7 @@ public class DatastoreOptions extends ServiceOptions<DatastoreRpc, DatastoreOpti
 
   private static final long serialVersionUID = -8636602944160689193L;
   private static final String DATASET_ENV_NAME = "DATASTORE_DATASET";
+  private static final String HOST_ENV_NAME = "DATASTORE_HOST";
   private static final String DATASTORE_SCOPE = "https://www.googleapis.com/auth/datastore";
   private static final String USERINFO_SCOPE = "https://www.googleapis.com/auth/userinfo.email";
   private static final Set<String> SCOPES = ImmutableSet.of(DATASTORE_SCOPE, USERINFO_SCOPE);
@@ -122,6 +123,12 @@ private DatastoreOptions normalize() {
     }
   }
 
+  @Override
+  protected String defaultHost() {
+    String host = System.getProperty(HOST_ENV_NAME, System.getenv(HOST_ENV_NAME));
+    return host != null ? host : super.defaultHost();
+  }
+
   @Override
   protected String defaultProject() {
     String projectId = System.getProperty(DATASET_ENV_NAME, System.getenv(DATASET_ENV_NAME));