Skip to content

Commit

Permalink
Updating procedure name in response to drivers/po request.
Browse files Browse the repository at this point in the history
  • Loading branch information
Mark Needham committed Aug 12, 2016
1 parent c31f6fb commit 8dd31da
Show file tree
Hide file tree
Showing 6 changed files with 35 additions and 35 deletions.
Expand Up @@ -51,7 +51,7 @@
import org.neo4j.coreedge.messaging.SenderService;
import org.neo4j.coreedge.discovery.procedures.AcquireEndpointsProcedure;
import org.neo4j.coreedge.discovery.procedures.ClusterOverviewProcedure;
import org.neo4j.coreedge.discovery.procedures.DiscoverCoreServersProcedure;
import org.neo4j.coreedge.discovery.procedures.DiscoverEndpointAcquisitionServersProcedure;
import org.neo4j.coreedge.logging.BetterMessageLogger;
import org.neo4j.coreedge.logging.MessageLogger;
import org.neo4j.coreedge.logging.NullMessageLogger;
Expand Down Expand Up @@ -114,7 +114,7 @@ public void registerProcedures( Procedures procedures )
{
try
{
procedures.register( new DiscoverCoreServersProcedure( discoveryService, logProvider ) );
procedures.register( new DiscoverEndpointAcquisitionServersProcedure( discoveryService, logProvider ) );
procedures.register( new AcquireEndpointsProcedure( discoveryService, consensusModule.raftMachine(), logProvider ) );
procedures.register( new ClusterOverviewProcedure( discoveryService, consensusModule.raftMachine(), logProvider ) );
procedures.register( new CoreRoleProcedure( consensusModule.raftMachine()) );
Expand Down
Expand Up @@ -39,13 +39,13 @@
import static java.util.stream.Collectors.toSet;
import static org.neo4j.kernel.api.proc.ProcedureSignature.procedureSignature;

public class DiscoverCoreServersProcedure extends CallableProcedure.BasicProcedure
public class DiscoverEndpointAcquisitionServersProcedure extends CallableProcedure.BasicProcedure
{
public static final String NAME = "discoverCoreServers";
public static final String NAME = "discoverEndpointAcquisitionServers";
private final CoreTopologyService discoveryService;
private final Log log;

public DiscoverCoreServersProcedure( CoreTopologyService discoveryService, LogProvider logProvider )
public DiscoverEndpointAcquisitionServersProcedure( CoreTopologyService discoveryService, LogProvider logProvider )
{
super( procedureSignature( new ProcedureSignature.ProcedureName( new String[]{"dbms", "cluster"}, NAME ) )
.out( "address", Neo4jTypes.NTString ).build() );
Expand Down
Expand Up @@ -53,9 +53,9 @@ public void shouldRecommendTheCoreLeaderForWriteAndEdgeForRead() throws Exceptio

Map<MemberId,CoreAddresses> coreMembers = new HashMap<>();
MemberId theLeader = member( 0 );
coreMembers.put( theLeader, DiscoverCoreServersProcedureTest.coreAddresses( 0 ) );
coreMembers.put( theLeader, DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ) );

final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverCoreServersProcedureTest.addresses( 1 ) );
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverEndpointAcquisitionServersProcedureTest.addresses( 1 ) );
when( topologyService.currentTopology() ).thenReturn( clusterTopology );

LeaderLocator leaderLocator = mock( LeaderLocator.class );
Expand All @@ -69,8 +69,8 @@ public void shouldRecommendTheCoreLeaderForWriteAndEdgeForRead() throws Exceptio

// then
MatcherAssert.assertThat( members, containsInAnyOrder(
new Object[]{DiscoverCoreServersProcedureTest.coreAddresses( 0 ).getRaftServer().toString(), "WRITE"},
new Object[]{DiscoverCoreServersProcedureTest.coreAddresses( 1 ).getRaftServer().toString(), "READ"}
new Object[]{DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ).getRaftServer().toString(), "WRITE"},
new Object[]{DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 1 ).getRaftServer().toString(), "READ"}
) );
}

Expand All @@ -82,9 +82,9 @@ public void shouldOnlyRecommendOneReadServerEvenIfMultipleAreAvailable() throws

Map<MemberId, CoreAddresses> coreMembers = new HashMap<>();
MemberId theLeader = member( 0 );
coreMembers.put( theLeader, DiscoverCoreServersProcedureTest.coreAddresses( 0 ) );
coreMembers.put( theLeader, DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ) );

final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverCoreServersProcedureTest.addresses( 1, 2, 3 ) );
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverEndpointAcquisitionServersProcedureTest.addresses( 1, 2, 3 ) );
when( topologyService.currentTopology() ).thenReturn( clusterTopology );

LeaderLocator leaderLocator = mock( LeaderLocator.class );
Expand All @@ -108,8 +108,8 @@ public void shouldReturnCoreMemberAsReadServerIfNoEdgeServersAvailable() throws

Map<MemberId, CoreAddresses> coreMembers = new HashMap<>();
MemberId theLeader = member( 0 );
coreMembers.put( theLeader, DiscoverCoreServersProcedureTest.coreAddresses( 0 ) );
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverCoreServersProcedureTest.addresses() );
coreMembers.put( theLeader, DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ) );
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverEndpointAcquisitionServersProcedureTest.addresses() );

when( topologyService.currentTopology() ).thenReturn( clusterTopology );

Expand All @@ -124,8 +124,8 @@ public void shouldReturnCoreMemberAsReadServerIfNoEdgeServersAvailable() throws

// then
assertThat( members, containsInAnyOrder(
new Object[]{DiscoverCoreServersProcedureTest.coreAddresses( 0 ).getRaftServer().toString(), "WRITE"},
new Object[]{DiscoverCoreServersProcedureTest.coreAddresses( 0 ).getRaftServer().toString(), "READ"}
new Object[]{DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ).getRaftServer().toString(), "WRITE"},
new Object[]{DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ).getRaftServer().toString(), "READ"}
) );
}

Expand All @@ -136,9 +136,9 @@ public void shouldReturnNoWriteEndpointsIfThereIsNoLeader() throws Exception
final CoreTopologyService topologyService = mock( CoreTopologyService.class );

Map<MemberId, CoreAddresses> coreMembers = new HashMap<>();
coreMembers.put( member( 0 ), DiscoverCoreServersProcedureTest.coreAddresses( 0 ) );
coreMembers.put( member( 0 ), DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ) );

final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverCoreServersProcedureTest.addresses() );
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverEndpointAcquisitionServersProcedureTest.addresses() );

when( topologyService.currentTopology() ).thenReturn( clusterTopology );

Expand All @@ -163,9 +163,9 @@ public void shouldReturnNoWriteEndpointsIfThereIsNoAddressForTheLeader() throws
final CoreTopologyService topologyService = mock( CoreTopologyService.class );

Map<MemberId, CoreAddresses> coreMembers = new HashMap<>();
coreMembers.put( member( 0 ), DiscoverCoreServersProcedureTest.coreAddresses( 0 ) );
coreMembers.put( member( 0 ), DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ) );

final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverCoreServersProcedureTest.addresses() );
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, DiscoverEndpointAcquisitionServersProcedureTest.addresses() );

when( topologyService.currentTopology() ).thenReturn( clusterTopology );

Expand Down
Expand Up @@ -55,11 +55,11 @@ public void shouldRecommendTheCoreLeaderForWriteAndEdgeForRead() throws Exceptio
MemberId follower1 = new MemberId( UUID.randomUUID() );
MemberId follower2 = new MemberId( UUID.randomUUID() );

coreMembers.put( theLeader, DiscoverCoreServersProcedureTest.coreAddresses( 0 ) );
coreMembers.put( follower1, DiscoverCoreServersProcedureTest.coreAddresses( 1 ) );
coreMembers.put( follower2, DiscoverCoreServersProcedureTest.coreAddresses( 2 ) );
coreMembers.put( theLeader, DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 0 ) );
coreMembers.put( follower1, DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 1 ) );
coreMembers.put( follower2, DiscoverEndpointAcquisitionServersProcedureTest.coreAddresses( 2 ) );

Set<EdgeAddresses> edges = DiscoverCoreServersProcedureTest.addresses( 4, 5 );
Set<EdgeAddresses> edges = DiscoverEndpointAcquisitionServersProcedureTest.addresses( 4, 5 );

final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, edges );
when( topologyService.currentTopology() ).thenReturn( clusterTopology );
Expand Down
Expand Up @@ -45,7 +45,7 @@
import static org.neo4j.coreedge.identity.RaftTestMember.member;
import static org.neo4j.helpers.collection.Iterators.asList;

public class DiscoverCoreServersProcedureTest
public class DiscoverEndpointAcquisitionServersProcedureTest
{
@Test
public void shouldOnlyReturnCoreMembers() throws Exception
Expand All @@ -61,8 +61,8 @@ public void shouldOnlyReturnCoreMembers() throws Exception
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, addresses( 3, 4, 5 ) );
when( coreTopologyService.currentTopology() ).thenReturn( clusterTopology );

final DiscoverCoreServersProcedure proc =
new DiscoverCoreServersProcedure( coreTopologyService, NullLogProvider.getInstance() );
final DiscoverEndpointAcquisitionServersProcedure proc =
new DiscoverEndpointAcquisitionServersProcedure( coreTopologyService, NullLogProvider.getInstance() );

// when
final List<Object[]> members = asList( proc.apply( null, new Object[0] ) );
Expand All @@ -85,8 +85,8 @@ public void shouldReturnSelfIfOnlyMemberOfTheCluster() throws Exception

final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, addresses( 3, 4, 5 ) );
when( coreTopologyService.currentTopology() ).thenReturn( clusterTopology );
final DiscoverCoreServersProcedure proc =
new DiscoverCoreServersProcedure( coreTopologyService, NullLogProvider.getInstance() );
final DiscoverEndpointAcquisitionServersProcedure proc =
new DiscoverEndpointAcquisitionServersProcedure( coreTopologyService, NullLogProvider.getInstance() );

// when
final List<Object[]> members = asList( proc.apply( null, new Object[0] ) );
Expand All @@ -109,8 +109,8 @@ public void shouldReturnLimitedNumberOfAddresses() throws Exception
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, addresses( 3, 4, 5) );
when( coreTopologyService.currentTopology() ).thenReturn( clusterTopology );

final DiscoverCoreServersProcedure proc =
new DiscoverCoreServersProcedure( coreTopologyService, NullLogProvider.getInstance() );
final DiscoverEndpointAcquisitionServersProcedure proc =
new DiscoverEndpointAcquisitionServersProcedure( coreTopologyService, NullLogProvider.getInstance() );

// when
final List<Object[]> members = asList( proc.apply( null, new Object[]{1} ) );
Expand All @@ -133,8 +133,8 @@ public void shouldReturnAllAddressesWhenLimitInNotNumeric() throws Exception
final ClusterTopology clusterTopology = new ClusterTopology( false, coreMembers, addresses( 3, 4, 5 ) );
when( coreTopologyService.currentTopology() ).thenReturn( clusterTopology );

final DiscoverCoreServersProcedure proc =
new DiscoverCoreServersProcedure( coreTopologyService, NullLogProvider.getInstance() );
final DiscoverEndpointAcquisitionServersProcedure proc =
new DiscoverEndpointAcquisitionServersProcedure( coreTopologyService, NullLogProvider.getInstance() );

// when
final List<Object[]> members = asList( proc.apply( null, new Object[]{"not numeric"} ) );
Expand All @@ -146,7 +146,7 @@ public void shouldReturnAllAddressesWhenLimitInNotNumeric() throws Exception
static Set<EdgeAddresses> addresses( int... ids )
{
return Arrays.stream( ids )
.mapToObj( DiscoverCoreServersProcedureTest::edgeAddresses )
.mapToObj( DiscoverEndpointAcquisitionServersProcedureTest::edgeAddresses )
.collect( Collectors.toSet() );
}

Expand Down
Expand Up @@ -26,7 +26,7 @@

import org.neo4j.coreedge.discovery.Cluster;
import org.neo4j.coreedge.discovery.procedures.AcquireEndpointsProcedure;
import org.neo4j.coreedge.discovery.procedures.DiscoverCoreServersProcedure;
import org.neo4j.coreedge.discovery.procedures.DiscoverEndpointAcquisitionServersProcedure;
import org.neo4j.kernel.api.KernelAPI;
import org.neo4j.kernel.api.KernelTransaction;
import org.neo4j.kernel.api.KernelTransaction.Type;
Expand Down Expand Up @@ -116,7 +116,7 @@ private List<Object[]> discoverClusterMembers( GraphDatabaseFacade db ) throws T

// when
return asList( statement.readOperations().procedureCallRead(
procedureName( "dbms", "cluster", DiscoverCoreServersProcedure.NAME ),
procedureName( "dbms", "cluster", DiscoverEndpointAcquisitionServersProcedure.NAME ),
new Object[0] ) );
}

Expand Down

0 comments on commit 8dd31da

Please sign in to comment.