cockroach: github.com/abhinavdahiya/cockroach/testutils/testcluster Index | Files

package testcluster

import "github.com/abhinavdahiya/cockroach/testutils/testcluster"

Index

Package Files

testcluster.go

Variables

var TestClusterFactory serverutils.TestClusterFactory = testClusterFactoryImpl{}

TestClusterFactory can be passed to serverutils.InitTestClusterFactory

type ReplicationTarget Uses

type ReplicationTarget struct {
    NodeID  roachpb.NodeID
    StoreID roachpb.StoreID
}

ReplicationTarget identifies a node/store pair.

type TestCluster Uses

type TestCluster struct {
    Servers []*server.TestServer
    Conns   []*gosql.DB
}

TestCluster represents a set of TestServers. The hope is that it can be used analoguous to TestServer, but with control over range replication.

func StartTestCluster Uses

func StartTestCluster(t testing.TB, nodes int, args base.TestClusterArgs) *TestCluster

StartTestCluster starts up a TestCluster made up of `nodes` in-memory testing servers. The cluster should be stopped using cluster.Stopper().Stop().

func (*TestCluster) AddReplicas Uses

func (tc *TestCluster) AddReplicas(
    startKey roachpb.Key, targets ...ReplicationTarget,
) (*roachpb.RangeDescriptor, error)

AddReplicas adds replicas for a range on a set of stores. It's illegal to have multiple replicas of the same range on stores of a single node. The method blocks until a snapshot of the range has been copied to all the new replicas and the new replicas become part of the Raft group.

func (*TestCluster) FindRangeLeaseHolder Uses

func (tc *TestCluster) FindRangeLeaseHolder(
    rangeDesc *roachpb.RangeDescriptor,
    hint *ReplicationTarget,
) (ReplicationTarget, error)

FindRangeLeaseHolder returns the current lease holder for the given range. If there is no lease at the time of the call, a replica is gets one as a side-effect of calling this; if hint is not nil, that replica will be the one.

One of the Stores in the cluster is used as a Sender to send a dummy read command, which will either result in success (if a replica on that Node has the lease), in a NotLeaseHolderError pointing to the current lease holder (if there is an active lease), or in the replica on that store acquiring the lease (if there isn't an active lease). If an active lease existed for the range, it's extended as a side-effect.

func (*TestCluster) LookupRange Uses

func (tc *TestCluster) LookupRange(key roachpb.Key) (roachpb.RangeDescriptor, error)

LookupRange returns the descriptor of the range containing key.

func (*TestCluster) NumServers Uses

func (tc *TestCluster) NumServers() int

NumServers is part of TestClusterInterface.

func (*TestCluster) RemoveReplicas Uses

func (tc *TestCluster) RemoveReplicas(
    startKey roachpb.Key, targets ...ReplicationTarget,
) (*roachpb.RangeDescriptor, error)

RemoveReplicas removes one or more replicas from a range.

func (*TestCluster) Server Uses

func (tc *TestCluster) Server(idx int) serverutils.TestServerInterface

Server is part of TestClusterInterface.

func (*TestCluster) ServerConn Uses

func (tc *TestCluster) ServerConn(idx int) *gosql.DB

ServerConn is part of TestClusterInterface.

func (*TestCluster) SplitRange Uses

func (tc *TestCluster) SplitRange(
    splitKey roachpb.Key,
) (*roachpb.RangeDescriptor, *roachpb.RangeDescriptor, error)

SplitRange splits the range containing splitKey. The right range created by the split starts at the split key and extends to the original range's end key. Returns the new descriptors of the left and right ranges.

splitKey must correspond to a SQL table key (it must end with a family ID / col ID).

func (*TestCluster) Stopper Uses

func (tc *TestCluster) Stopper() *stop.Stopper

Stopper returns a Stopper to be used to stop the TestCluster.

func (*TestCluster) Target Uses

func (tc *TestCluster) Target(serverIdx int) ReplicationTarget

Target returns a ReplicationTarget for the specified server.

func (*TestCluster) TransferRangeLease Uses

func (tc *TestCluster) TransferRangeLease(
    rangeDesc *roachpb.RangeDescriptor, dest ReplicationTarget,
) error

TransferRangeLease transfers the lease for a range from whoever has it to a particular store. That store must already have a replica of the range. If that replica already has the (active) lease, this method is a no-op.

func (*TestCluster) WaitForFullReplication Uses

func (tc *TestCluster) WaitForFullReplication() error

WaitForFullReplication waits until all stores in the cluster have no ranges with replication pending.

Package testcluster imports 19 packages (graph). Updated 2017-03-13. Refresh now. Tools for package owners. This is a dead-end fork (no commits since the fork).