mirror of
https://github.com/google/nomulus.git
synced 2025-05-02 04:57:51 +02:00
077600971f
This is a temporary change used for b/71607306 only, and has TODOs to revert at every change once the bug is done. We want to check how Cloud DNS handles large (1M+ domain) zones, especially during resigning. However, due to a separate bug (b/70980350, and maybe another one) we can't currently create such a large zone in nomulus within the required 4 day timeframe. The most we managed is 300k domains. We could wait until the bug is fixed, but if there's a problem with Cloud DNS - we want to find out as fast as possible. Hence, this CL that allows us to register 1M domains by creating "just" 100k domains in nomulus. The CL creates a new "MultiplyingCloudDnsWriter" writer, that when publishing a domain, pretends that we are publishing 10 domains (with 9 additional "fictional" domains, that get their Datastore data from the actual domain). ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=180962149
105 lines
3.3 KiB
Java
105 lines
3.3 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.writer.clouddns;
|
|
|
|
import com.google.api.client.http.HttpRequestInitializer;
|
|
import com.google.api.client.http.HttpTransport;
|
|
import com.google.api.client.json.JsonFactory;
|
|
import com.google.api.services.dns.Dns;
|
|
import com.google.api.services.dns.DnsScopes;
|
|
import com.google.common.util.concurrent.RateLimiter;
|
|
import dagger.Module;
|
|
import dagger.Provides;
|
|
import dagger.multibindings.IntoMap;
|
|
import dagger.multibindings.IntoSet;
|
|
import dagger.multibindings.StringKey;
|
|
import google.registry.config.RegistryConfig.Config;
|
|
import google.registry.dns.writer.DnsWriter;
|
|
import java.util.Optional;
|
|
import java.util.Set;
|
|
import java.util.function.Function;
|
|
import javax.inject.Named;
|
|
|
|
/** Dagger module for Google Cloud DNS service connection objects. */
|
|
@Module
|
|
public final class CloudDnsWriterModule {
|
|
|
|
@Provides
|
|
static Dns provideDns(
|
|
HttpTransport transport,
|
|
JsonFactory jsonFactory,
|
|
Function<Set<String>, ? extends HttpRequestInitializer> credential,
|
|
@Config("projectId") String projectId,
|
|
@Config("cloudDnsRootUrl") Optional<String> rootUrl,
|
|
@Config("cloudDnsServicePath") Optional<String> servicePath) {
|
|
Dns.Builder builder =
|
|
new Dns.Builder(transport, jsonFactory, credential.apply(DnsScopes.all()))
|
|
.setApplicationName(projectId);
|
|
|
|
rootUrl.ifPresent(builder::setRootUrl);
|
|
servicePath.ifPresent(builder::setServicePath);
|
|
|
|
return builder.build();
|
|
}
|
|
|
|
@Provides
|
|
@IntoMap
|
|
@StringKey(CloudDnsWriter.NAME)
|
|
static DnsWriter provideWriter(CloudDnsWriter writer) {
|
|
return writer;
|
|
}
|
|
|
|
// TODO(b/71607306): Remove once large zone resigning is tested
|
|
@Provides
|
|
@IntoMap
|
|
@StringKey(MultiplyingCloudDnsWriter.NAME)
|
|
static DnsWriter provideMultiplyingWriter(MultiplyingCloudDnsWriter writer) {
|
|
return writer;
|
|
}
|
|
|
|
@Provides
|
|
@IntoSet
|
|
@Named("dnsWriterNames")
|
|
static String provideWriterName() {
|
|
return CloudDnsWriter.NAME;
|
|
}
|
|
|
|
// TODO(b/71607306): Remove once large zone resigning is tested
|
|
@Provides
|
|
@IntoSet
|
|
@Named("dnsWriterNames")
|
|
static String provideMultiplyingWriterName() {
|
|
return MultiplyingCloudDnsWriter.NAME;
|
|
}
|
|
|
|
@Provides
|
|
@Named("cloudDns")
|
|
static RateLimiter provideRateLimiter() {
|
|
// This is the default max QPS for Cloud DNS. It can be increased by contacting the team
|
|
// via the Quotas page on the Cloud Console.
|
|
int cloudDnsMaxQps = 20;
|
|
return RateLimiter.create(cloudDnsMaxQps);
|
|
}
|
|
|
|
@Provides
|
|
@Named("cloudDnsNumThreads")
|
|
static int provideNumThreads() {
|
|
// TODO(b/70217860): find the "best" number of threads, taking into account running time, App
|
|
// Engine constraints, and any Cloud DNS comsiderations etc.
|
|
//
|
|
// NOTE: any number below 2 will not use threading at all.
|
|
return 10;
|
|
}
|
|
}
|