mv com/google/domain/registry google/registry

This change renames directories in preparation for the great package
rename. The repository is now in a broken state because the code
itself hasn't been updated. However this should ensure that git
correctly preserves history for each file.

java/google/registry/rde/RdeCounter.java

@@ -0,0 +1,97 @@
+// Copyright 2016 The Domain Registry Authors. 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.rde;
+
+import static com.google.common.base.Predicates.equalTo;
+import static com.google.common.base.Predicates.not;
+import static com.google.common.collect.Iterables.filter;
+
+import com.google.domain.registry.model.rde.RdeMode;
+import com.google.domain.registry.xjc.rde.XjcRdeDepositTypeType;
+import com.google.domain.registry.xjc.rdeheader.XjcRdeHeader;
+import com.google.domain.registry.xjc.rdeheader.XjcRdeHeaderCount;
+import com.google.domain.registry.xjc.rdeheader.XjcRdeHeaderElement;
+import com.google.domain.registry.xjc.rdereport.XjcRdeReport;
+
+import org.joda.time.DateTime;
+
+import java.util.EnumMap;
+import java.util.EnumSet;
+import java.util.concurrent.atomic.AtomicLong;
+
+import javax.annotation.concurrent.NotThreadSafe;
+import javax.inject.Inject;
+
+/** Utility class for generating a single {@link XjcRdeHeader} while marshalling a deposit. */
+@NotThreadSafe
+public final class RdeCounter {
+
+  private static final String URI_ESCROW = "draft-arias-noguchi-registry-data-escrow-06";
+  private static final String URI_MAPPING = "draft-arias-noguchi-dnrd-objects-mapping-05";
+  private static final int ICANN_REPORT_SPEC_VERSION = 1;
+
+  private final EnumMap<RdeResourceType, AtomicLong> counts = new EnumMap<>(RdeResourceType.class);
+
+  @Inject
+  public RdeCounter() {
+    for (RdeResourceType resourceType : getResourceTypesExcludingHeader()) {
+      counts.put(resourceType, new AtomicLong());
+    }
+  }
+
+  /** Increment the count on a given resource. */
+  public void increment(RdeResourceType type) {
+    counts.get(type).incrementAndGet();
+  }
+
+  /** Constructs a header containing the sum of {@link #increment(RdeResourceType)} calls. */
+  public XjcRdeHeader makeHeader(String tld, RdeMode mode) {
+    XjcRdeHeader header = new XjcRdeHeader();
+    header.setTld(tld);
+    for (RdeResourceType resourceType : getResourceTypesExcludingHeader()) {
+      if (resourceType.getModes().contains(mode)) {
+        header.getCounts().add(makeCount(resourceType.getUri(), counts.get(resourceType).get()));
+      }
+    }
+    return header;
+  }
+
+  /** Returns an ICANN notification report as a JAXB object. */
+  public XjcRdeReport
+      makeReport(String id, DateTime watermark, XjcRdeHeader header, int revision) {
+    XjcRdeReport report = new XjcRdeReport();
+    report.setId(id);
+    report.setKind(XjcRdeDepositTypeType.FULL);
+    report.setCrDate(watermark);
+    report.setWatermark(watermark);
+    report.setVersion(ICANN_REPORT_SPEC_VERSION);
+    report.setRydeSpecEscrow(URI_ESCROW);
+    report.setRydeSpecMapping(URI_MAPPING);
+    report.setResend(revision);
+    report.setHeader(new XjcRdeHeaderElement(header));
+    return report;
+  }
+
+  private Iterable<RdeResourceType> getResourceTypesExcludingHeader() {
+    return filter(EnumSet.allOf(RdeResourceType.class), not(equalTo(RdeResourceType.HEADER)));
+  }
+
+  private static XjcRdeHeaderCount makeCount(String uri, long count) {
+    XjcRdeHeaderCount bean = new XjcRdeHeaderCount();
+    bean.setUri(uri);
+    bean.setValue(count);
+    return bean;
+  }
+}