Import code from internal repository to git

2025-06-03 02:57:33 +02:00 · 2016-03-01 17:18:14 -05:00 · 2016-03-01 17:18:14 -05:00 · 0ef0c933d2
commit 0ef0c933d2
2490 changed files with 281594 additions and 0 deletions
--- a/java/com/google/domain/registry/flows/async/DnsRefreshForHostRenameAction.java
+++ b/java/com/google/domain/registry/flows/async/DnsRefreshForHostRenameAction.java
@ -0,0 +1,110 @@
+// Copyright 2016 Google Inc. All Rights Reserved.
+//
+// Licensed 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 com.google.domain.registry.flows.async;
+
+import static com.google.domain.registry.model.EppResourceUtils.isActive;
+import static com.google.domain.registry.model.ofy.ObjectifyService.ofy;
+import static com.google.domain.registry.util.PipelineUtils.createJobPath;
+import static com.google.domain.registry.util.PreconditionsUtils.checkArgumentNotNull;
+
+import com.google.appengine.tools.mapreduce.Mapper;
+import com.google.common.collect.ImmutableList;
+import com.google.domain.registry.dns.DnsQueue;
+import com.google.domain.registry.mapreduce.EppResourceInputs;
+import com.google.domain.registry.mapreduce.MapreduceAction;
+import com.google.domain.registry.mapreduce.MapreduceRunner;
+import com.google.domain.registry.model.domain.DomainResource;
+import com.google.domain.registry.model.domain.ReferenceUnion;
+import com.google.domain.registry.model.host.HostResource;
+import com.google.domain.registry.request.Action;
+import com.google.domain.registry.request.HttpException.BadRequestException;
+import com.google.domain.registry.request.Parameter;
+import com.google.domain.registry.request.Response;
+import com.google.domain.registry.util.FormattingLogger;
+import com.google.domain.registry.util.NonFinalForTesting;
+
+import com.googlecode.objectify.Key;
+import com.googlecode.objectify.Ref;
+
+import org.joda.time.DateTime;
+
+import javax.inject.Inject;
+
+/**
+ * Enqueues DNS refreshes for applicable domains following a host rename.
+ */
+@Action(path = "/_dr/task/dnsRefreshForHostRename")
+public class DnsRefreshForHostRenameAction implements MapreduceAction {
+
+  /** The HTTP parameter name used to specify the websafe key of the host to rename. */
+  public static final String PARAM_HOST_KEY = "hostKey";
+
+  private static final FormattingLogger logger = FormattingLogger.getLoggerForCallerClass();
+
+  @NonFinalForTesting
+  static DnsQueue dnsQueue = DnsQueue.create();
+
+  @Inject @Parameter(PARAM_HOST_KEY) String hostKeyString;
+  @Inject MapreduceRunner mrRunner;
+  @Inject Response response;
+  @Inject DnsRefreshForHostRenameAction() {}
+
+  @Override
+  public void run() {
+    Key<HostResource> resourceKey = null;
+    HostResource host;
+    try {
+      resourceKey = Key.create(hostKeyString);
+      host = checkArgumentNotNull(ofy().load().key(resourceKey).now());
+    } catch (IllegalArgumentException e) {
+      throw new BadRequestException(resourceKey == null
+          ? "Could not parse key string: " + hostKeyString
+          : "Could not load resource for key: " + resourceKey);
+    }
+    response.sendJavaScriptRedirect(createJobPath(mrRunner
+        .setJobName("Enqueue DNS refreshes for domains following a host rename")
+        .setModuleName("backend")
+        .runMapOnly(
+            new DnsRefreshForHostRenameMapper(host),
+            ImmutableList.of(EppResourceInputs.createEntityInput(DomainResource.class)))));
+    }
+
+  /** Map over domains and refresh the dns of those that referenced this host. */
+  public static class DnsRefreshForHostRenameMapper extends Mapper<DomainResource, Void, Void> {
+
+    private static final long serialVersionUID = -4707015136971008447L;
+
+    private final DateTime hostUpdateTime;
+    private final Key<HostResource> targetHostKey;
+
+    DnsRefreshForHostRenameMapper(HostResource host) {
+      this.targetHostKey = Key.create(host);
+      this.hostUpdateTime = host.getUpdateAutoTimestamp().getTimestamp();
+    }
+
+    @Override
+    public final void map(DomainResource domain) {
+      if (isActive(domain, hostUpdateTime)
+          && domain.getNameservers().contains(ReferenceUnion.create(Ref.create(targetHostKey)))) {
+        try {
+          dnsQueue.addDomainRefreshTask(domain.getFullyQualifiedDomainName());
+          logger.infofmt("Enqueued refresh for domain %s", domain.getFullyQualifiedDomainName());
+        } catch (Throwable t) {
+          logger.severefmt(t, "Error while refreshing DNS for host rename %s", targetHostKey);
+        }
+      }
+    }
+  }
+}