ClusterCompositionUtil.java

/*
 * Copyright (c) 2002-2017 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This file is part of Neo4j.
 *
 * 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 org.neo4j.driver.internal.cluster;

import java.util.ArrayList;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;

import org.neo4j.driver.internal.net.BoltServerAddress;

import static java.util.Arrays.asList;

public final class ClusterCompositionUtil
{
    private ClusterCompositionUtil() {}

    public static final long NEVER_EXPIRE = System.currentTimeMillis() + TimeUnit.HOURS.toMillis( 1 );

    public static final BoltServerAddress A = new BoltServerAddress( "1111:11" );
    public static final BoltServerAddress B = new BoltServerAddress( "2222:22" );
    public static final BoltServerAddress C = new BoltServerAddress( "3333:33" );
    public static final BoltServerAddress D = new BoltServerAddress( "4444:44" );
    public static final BoltServerAddress E = new BoltServerAddress( "5555:55" );
    public static final BoltServerAddress F = new BoltServerAddress( "6666:66" );

    public static final List<BoltServerAddress> EMPTY = new ArrayList<>();

    public static final ClusterComposition VALID_CLUSTER_COMPOSITION =
            createClusterComposition( asList( A, B ), asList( C ), asList( D, E ) );

    public static final ClusterComposition INVALID_CLUSTER_COMPOSITION =
            createClusterComposition( asList( A, B ), EMPTY, asList( D, E ) );

    @SafeVarargs
    public static ClusterComposition createClusterComposition( List<BoltServerAddress>... servers )
    {
        return createClusterComposition( NEVER_EXPIRE, servers );
    }

    @SafeVarargs
    public static ClusterComposition createClusterComposition( long expirationTimestamp, List<BoltServerAddress>...
            servers )
    {
        Set<BoltServerAddress> routers = new LinkedHashSet<>();
        Set<BoltServerAddress> writers = new LinkedHashSet<>();
        Set<BoltServerAddress> readers = new LinkedHashSet<>();

        switch( servers.length )
        {
        case 3:
            readers.addAll( servers[2] );
            // no break on purpose
        case 2:
            writers.addAll( servers[1] );
            // no break on purpose
        case 1:
            routers.addAll( servers[0] );
        }
        return new ClusterComposition( expirationTimestamp, readers, writers, routers );
    }
}