com.palantir.atlasdb.http.TimelockUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of atlasdb-jepsen-tests Show documentation
Palantir open source project
There is a newer version: 0.1152.0
/*
 * (c) Copyright 2018 Palantir Technologies 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.palantir.atlasdb.http;

import com.palantir.atlasdb.config.ImmutableServerListConfig;
import com.palantir.atlasdb.factory.AtlasDbDialogueServiceProvider;
import com.palantir.atlasdb.util.MetricsManager;
import com.palantir.common.concurrent.PTExecutors;
import com.palantir.conjure.java.api.config.service.ServicesConfigBlock;
import com.palantir.conjure.java.api.config.service.UserAgent;
import com.palantir.conjure.java.api.config.service.UserAgent.Agent;
import com.palantir.conjure.java.api.config.ssl.SslConfiguration;
import com.palantir.dialogue.clients.DialogueClients;
import com.palantir.refreshable.Refreshable;
import java.nio.file.Paths;
import java.util.List;
import java.util.stream.Collectors;

public final class TimelockUtils {
    private static final int PORT = 8080;
    static final String NAMESPACE = "test";

    private static final SslConfiguration SSL_CONFIGURATION = SslConfiguration.of(
            Paths.get("var", "security", "trustStore.jks"), Paths.get("var", "security", "keyStore.jks"), "keystore");

    private TimelockUtils() {}

    public static AtlasDbDialogueServiceProvider createServiceProvider(
            MetricsManager metricsManager, List hosts) {
        return AtlasDbDialogueServiceProvider.create(
                Refreshable.only(ImmutableServerListConfig.builder()
                        .addAllServers(hostnamesToEndpointUris(hosts))
                        .sslConfiguration(SSL_CONFIGURATION)
                        .build()),
                newMinimalDialogueFactory(),
                UserAgent.of(Agent.of("jepsen", "1.2.3")),
                metricsManager.getTaggedRegistry());
    }

    private static DialogueClients.ReloadingFactory newMinimalDialogueFactory() {
        return DialogueClients.create(
                        Refreshable.only(ServicesConfigBlock.builder().build()))
                .withBlockingExecutor(PTExecutors.newCachedThreadPool("atlas-dialogue-blocking"));
    }

    private static List hostnamesToEndpointUris(List hosts) {
        return hosts.stream()
                .map(host -> String.format("https://%s:%d", host, PORT))
                .collect(Collectors.toList());
    }
}