Import code from internal repository to git

java/com/google/domain/registry/dns/PublishDnsUpdatesAction.java

@@ -0,0 +1,98 @@
+// 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.dns;
+
+import static com.google.domain.registry.model.server.Lock.executeWithLocks;
+import static com.google.domain.registry.request.Action.Method.POST;
+import static com.google.domain.registry.util.CollectionUtils.nullToEmpty;
+
+import com.google.common.net.InternetDomainName;
+import com.google.domain.registry.config.ConfigModule.Config;
+import com.google.domain.registry.dns.writer.api.DnsWriter;
+import com.google.domain.registry.request.Action;
+import com.google.domain.registry.request.HttpException.ServiceUnavailableException;
+import com.google.domain.registry.request.Parameter;
+import com.google.domain.registry.request.RequestParameters;
+import com.google.domain.registry.util.DomainNameUtils;
+import com.google.domain.registry.util.FormattingLogger;
+
+import org.joda.time.Duration;
+
+import java.util.Set;
+import java.util.concurrent.Callable;
+
+import javax.inject.Inject;
+import javax.inject.Provider;
+
+/** Task that sends domain and host updates to the DNS server. */
+@Action(path = PublishDnsUpdatesAction.PATH, method = POST, automaticallyPrintOk = true)
+public final class PublishDnsUpdatesAction implements Runnable, Callable<Void> {
+
+  public static final String PATH = "/_dr/task/publishDnsUpdates";
+  public static final String DOMAINS_PARAM = "domains";
+  public static final String HOSTS_PARAM = "hosts";
+
+  private static final FormattingLogger logger = FormattingLogger.getLoggerForCallerClass();
+
+  @Inject DnsQueue dnsQueue;
+  @Inject Provider<DnsWriter> writerProvider;
+  @Inject @Config("dnsWriteLockTimeout") Duration timeout;
+  @Inject @Parameter(RequestParameters.PARAM_TLD) String tld;
+  @Inject @Parameter(DOMAINS_PARAM) Set<String> domains;
+  @Inject @Parameter(HOSTS_PARAM) Set<String> hosts;
+  @Inject PublishDnsUpdatesAction() {}
+
+  /** Runs the task. */
+  @Override
+  public void run() {
+    String lockName = String.format("DNS zone %s", tld);
+    // If executeWithLocks fails to get the lock, it does not throw an exception, simply returns
+    // false. We need to make sure to take note of this error; otherwise, a failed lock might result
+    // in the update task being dequeued and dropped. A message will already have been logged
+    // to indicate the problem.
+    if (!executeWithLocks(this, getClass(), tld, timeout, lockName)) {
+      throw new ServiceUnavailableException("Lock failure");
+    }
+  }
+
+  /** Runs the task, with the lock. */
+  @Override
+  public Void call() {
+    processBatch();
+    return null;
+  }
+
+  /** Steps through the domain and host refreshes contained in the parameters and processes them. */
+  private void processBatch() {
+    try (DnsWriter writer = writerProvider.get()) {
+      for (String domain : nullToEmpty(domains)) {
+        if (!DomainNameUtils.isUnder(
+            InternetDomainName.from(domain), InternetDomainName.from(tld))) {
+          logger.severefmt("%s: skipping domain %s not under tld", tld, domain);
+        } else {
+          writer.publishDomain(domain);
+        }
+      }
+      for (String host : nullToEmpty(hosts)) {
+        if (!DomainNameUtils.isUnder(
+            InternetDomainName.from(host), InternetDomainName.from(tld))) {
+          logger.severefmt("%s: skipping host %s not under tld", tld, host);
+        } else {
+          writer.publishHost(host);
+        }
+      }
+    }
+  }
+}