mirror of
https://github.com/google/nomulus.git
synced 2025-05-01 04:27:51 +02:00
d3e9ebad16
Note that even though the nomulus command line tool now supports multiple DNS writers for all subcommands, this still won't work quite yet because the DNS task queue format migration from [] is still in progress. After next week's push that migration will be complete and we can remove the final restriction against only having one DNS writer per TLD. ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=162490399
58 lines
2.2 KiB
Java
58 lines
2.2 KiB
Java
// Copyright 2017 The Nomulus 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 google.registry.dns;
|
|
|
|
import static com.google.common.base.Preconditions.checkState;
|
|
import static com.google.common.collect.Iterables.getOnlyElement;
|
|
import static google.registry.util.FormattingLogger.getLoggerForCallerClass;
|
|
|
|
import com.google.common.collect.ImmutableMap;
|
|
import google.registry.dns.writer.DnsWriter;
|
|
import google.registry.model.registry.Registry;
|
|
import google.registry.util.FormattingLogger;
|
|
import java.util.Map;
|
|
import javax.inject.Inject;
|
|
|
|
/** Proxy for retrieving {@link DnsWriter} implementations. */
|
|
public final class DnsWriterProxy {
|
|
|
|
private static final FormattingLogger logger = getLoggerForCallerClass();
|
|
|
|
private final ImmutableMap<String, DnsWriter> dnsWriters;
|
|
|
|
@Inject
|
|
DnsWriterProxy(Map<String, DnsWriter> dnsWriters) {
|
|
this.dnsWriters = ImmutableMap.copyOf(dnsWriters);
|
|
}
|
|
|
|
/** Returns the {@link DnsWriter} for the given tld. */
|
|
// TODO(b/63385597): Delete this method after DNS writers migration is complete.
|
|
@Deprecated
|
|
public DnsWriter getForTld(String tld) {
|
|
return getByClassNameForTld(getOnlyElement(Registry.get(tld).getDnsWriters()), tld);
|
|
}
|
|
|
|
/** Returns the instance of {@link DnsWriter} by class name. */
|
|
public DnsWriter getByClassNameForTld(String className, String tld) {
|
|
if (!Registry.get(tld).getDnsWriters().contains(className)) {
|
|
logger.warningfmt(
|
|
"Loaded potentially stale DNS writer %s which is no longer active on TLD %s.",
|
|
className, tld);
|
|
}
|
|
DnsWriter dnsWriter = dnsWriters.get(className);
|
|
checkState(dnsWriter != null, "Could not load DnsWriter %s for TLD %s", className, tld);
|
|
return dnsWriter;
|
|
}
|
|
}
|