Skip to content

Commit f17cf57

Browse files
committed
roachpb: move FullReplicaID type in
Define FullReplicaID next to RangeID and ReplicaID. Epic: none Release note: none
1 parent 1b915b1 commit f17cf57

17 files changed

+54
-59
lines changed

pkg/kv/kvserver/client_raft_log_queue_test.go

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ import (
2626
"github.com/cockroachdb/cockroach/pkg/rpc/rpcbase"
2727
"github.com/cockroachdb/cockroach/pkg/server"
2828
"github.com/cockroachdb/cockroach/pkg/settings/cluster"
29-
"github.com/cockroachdb/cockroach/pkg/storage"
3029
"github.com/cockroachdb/cockroach/pkg/storage/fs"
3130
"github.com/cockroachdb/cockroach/pkg/testutils"
3231
"github.com/cockroachdb/cockroach/pkg/testutils/listenerutil"
@@ -263,7 +262,7 @@ func TestCrashWhileTruncatingSideloadedEntries(t *testing.T) {
263262
propFilter := newAtomicFunc(func(kvserverbase.ProposalFilterArgs) *kvpb.Error {
264263
return nil
265264
})
266-
applyThrottle := newAtomicFunc(func(storage.FullReplicaID) {})
265+
applyThrottle := newAtomicFunc(func(roachpb.FullReplicaID) {})
267266
postSideEffects := newAtomicFunc(func(args kvserverbase.ApplyFilterArgs) (int, *kvpb.Error) {
268267
return 0, nil
269268
})
@@ -293,7 +292,7 @@ func TestCrashWhileTruncatingSideloadedEntries(t *testing.T) {
293292
Store: &kvserver.StoreTestingKnobs{
294293
DisableRaftLogQueue: true, // we send a log truncation manually
295294
DisableSyncLogWriteToss: true, // always use async log writes
296-
TestingAfterRaftLogSync: func(id storage.FullReplicaID) { applyThrottle.get()(id) },
295+
TestingAfterRaftLogSync: func(id roachpb.FullReplicaID) { applyThrottle.get()(id) },
297296
TestingProposalFilter: func(args kvserverbase.ProposalFilterArgs) *kvpb.Error {
298297
return propFilter.get()(args)
299298
},
@@ -347,7 +346,7 @@ func TestCrashWhileTruncatingSideloadedEntries(t *testing.T) {
347346
// Before writing more commands, block the raft commands application flow on
348347
// the follower replica.
349348
unblockApply := make(chan struct{})
350-
applyThrottle.set(func(id storage.FullReplicaID) {
349+
applyThrottle.set(func(id roachpb.FullReplicaID) {
351350
if id == follower.ID() {
352351
applyThrottle.reset()
353352
<-unblockApply

pkg/kv/kvserver/kvstorage/datadriven_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ func (e *env) close() {
6060
func (e *env) handleNewReplica(
6161
t *testing.T,
6262
ctx context.Context,
63-
id storage.FullReplicaID,
63+
id roachpb.FullReplicaID,
6464
skipRaftReplicaID bool,
6565
k, ek roachpb.RKey,
6666
) *roachpb.RangeDescriptor {
@@ -154,7 +154,7 @@ func TestDataDriven(t *testing.T) {
154154
d.ScanArgs(t, "skip-raft-replica-id", &skipRaftReplicaID)
155155
}
156156
if desc := e.handleNewReplica(t, ctx,
157-
storage.FullReplicaID{RangeID: roachpb.RangeID(rangeID), ReplicaID: roachpb.ReplicaID(replicaID)},
157+
roachpb.FullReplicaID{RangeID: roachpb.RangeID(rangeID), ReplicaID: roachpb.ReplicaID(replicaID)},
158158
skipRaftReplicaID, keys.MustAddr(roachpb.Key(k)), keys.MustAddr(roachpb.Key(ek)),
159159
); desc != nil {
160160
fmt.Fprintln(&buf, desc)

pkg/kv/kvserver/kvstorage/init.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -388,8 +388,8 @@ type Replica struct {
388388
}
389389

390390
// ID returns the FullReplicaID.
391-
func (r Replica) ID() storage.FullReplicaID {
392-
return storage.FullReplicaID{
391+
func (r Replica) ID() roachpb.FullReplicaID {
392+
return roachpb.FullReplicaID{
393393
RangeID: r.RangeID,
394394
ReplicaID: r.ReplicaID,
395395
}

pkg/kv/kvserver/kvstorage/replica_state.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,8 +71,8 @@ func LoadReplicaState(
7171
return ls, nil
7272
}
7373

74-
func (r LoadedReplicaState) FullReplicaID() storage.FullReplicaID {
75-
return storage.FullReplicaID{RangeID: r.ReplState.Desc.RangeID, ReplicaID: r.ReplicaID}
74+
func (r LoadedReplicaState) FullReplicaID() roachpb.FullReplicaID {
75+
return roachpb.FullReplicaID{RangeID: r.ReplState.Desc.RangeID, ReplicaID: r.ReplicaID}
7676
}
7777

7878
// check makes sure that the replica invariants hold for the loaded state.
@@ -120,7 +120,7 @@ const CreateUninitReplicaTODO = 0
120120
// Returns kvpb.RaftGroupDeletedError if this replica can not be created
121121
// because it has been deleted.
122122
func CreateUninitializedReplica(
123-
ctx context.Context, eng storage.Engine, storeID roachpb.StoreID, id storage.FullReplicaID,
123+
ctx context.Context, eng storage.Engine, storeID roachpb.StoreID, id roachpb.FullReplicaID,
124124
) error {
125125
sl := stateloader.Make(id.RangeID)
126126
// Before creating the replica, see if there is a tombstone which would

pkg/kv/kvserver/replica.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1096,8 +1096,8 @@ func (r *Replica) ReplicaID() roachpb.ReplicaID {
10961096
}
10971097

10981098
// ID returns the FullReplicaID for the Replica.
1099-
func (r *Replica) ID() storage.FullReplicaID {
1100-
return storage.FullReplicaID{RangeID: r.RangeID, ReplicaID: r.replicaID}
1099+
func (r *Replica) ID() roachpb.FullReplicaID {
1100+
return roachpb.FullReplicaID{RangeID: r.RangeID, ReplicaID: r.replicaID}
11011101
}
11021102

11031103
// LogStorageRaftMuLocked returns the Replica's log storage.

pkg/kv/kvserver/replica_init.go

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ import (
2929
"github.com/cockroachdb/cockroach/pkg/roachpb"
3030
"github.com/cockroachdb/cockroach/pkg/rpc/rpcbase"
3131
"github.com/cockroachdb/cockroach/pkg/server/telemetry"
32-
"github.com/cockroachdb/cockroach/pkg/storage"
3332
"github.com/cockroachdb/cockroach/pkg/util"
3433
"github.com/cockroachdb/cockroach/pkg/util/buildutil"
3534
"github.com/cockroachdb/cockroach/pkg/util/envutil"
@@ -100,7 +99,7 @@ func newInitializedReplica(
10099
//
101100
// TODO(#94912): we actually have another initialization path which should be
102101
// refactored: Replica.initFromSnapshotLockedRaftMuLocked().
103-
func newUninitializedReplica(store *Store, id storage.FullReplicaID) (*Replica, error) {
102+
func newUninitializedReplica(store *Store, id roachpb.FullReplicaID) (*Replica, error) {
104103
r := newUninitializedReplicaWithoutRaftGroup(store, id)
105104
r.raftMu.Lock()
106105
defer r.raftMu.Unlock()
@@ -117,7 +116,7 @@ func newUninitializedReplica(store *Store, id storage.FullReplicaID) (*Replica,
117116
// newUninitializedReplica() instead. This only exists for
118117
// newInitializedReplica() to avoid creating the Raft group twice (once when
119118
// creating the uninitialized replica, and once when initializing it).
120-
func newUninitializedReplicaWithoutRaftGroup(store *Store, id storage.FullReplicaID) *Replica {
119+
func newUninitializedReplicaWithoutRaftGroup(store *Store, id roachpb.FullReplicaID) *Replica {
121120
uninitState := stateloader.UninitializedReplicaState(id.RangeID)
122121
r := &Replica{
123122
AmbientContext: store.cfg.AmbientCtx,

pkg/kv/kvserver/replica_raft.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2817,7 +2817,7 @@ func (r *Replica) acquireSplitLock(
28172817
ctx context.Context, split *roachpb.SplitTrigger,
28182818
) (func(), error) {
28192819
rightReplDesc, _ := split.RightDesc.GetReplicaDescriptor(r.StoreID())
2820-
rightRepl, _, err := r.store.getOrCreateReplica(ctx, storage.FullReplicaID{
2820+
rightRepl, _, err := r.store.getOrCreateReplica(ctx, roachpb.FullReplicaID{
28212821
RangeID: split.RightDesc.RangeID,
28222822
ReplicaID: rightReplDesc.ReplicaID,
28232823
}, nil /* creatingReplica */)
@@ -2853,7 +2853,7 @@ func (r *Replica) acquireMergeLock(
28532853
// complete, after which the replica will realize it has been destroyed and
28542854
// reject the snapshot.
28552855
rightReplDesc, _ := merge.RightDesc.GetReplicaDescriptor(r.StoreID())
2856-
rightRepl, _, err := r.store.getOrCreateReplica(ctx, storage.FullReplicaID{
2856+
rightRepl, _, err := r.store.getOrCreateReplica(ctx, roachpb.FullReplicaID{
28572857
RangeID: merge.RightDesc.RangeID,
28582858
ReplicaID: rightReplDesc.ReplicaID,
28592859
}, nil /* creatingReplica */)

pkg/kv/kvserver/snapshot_apply_prepare.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ import (
2323
// snapWriteBuilder contains the data needed to prepare the on-disk state for a
2424
// snapshot.
2525
type snapWriteBuilder struct {
26-
id storage.FullReplicaID
26+
id roachpb.FullReplicaID
2727

2828
todoEng storage.Engine
2929
sl stateloader.StateLoader

pkg/kv/kvserver/snapshot_apply_prepare_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ func TestPrepareSnapApply(t *testing.T) {
6969
}
7070
}
7171

72-
id := storage.FullReplicaID{RangeID: 123, ReplicaID: 4}
72+
id := roachpb.FullReplicaID{RangeID: 123, ReplicaID: 4}
7373
descA := desc(101, "a", "b")
7474
descB := desc(102, "b", "z")
7575
createRangeData(t, eng, *descA)

pkg/kv/kvserver/store_create_replica.go

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ import (
1212
"github.com/cockroachdb/cockroach/pkg/kv/kvpb"
1313
"github.com/cockroachdb/cockroach/pkg/kv/kvserver/kvstorage"
1414
"github.com/cockroachdb/cockroach/pkg/roachpb"
15-
"github.com/cockroachdb/cockroach/pkg/storage"
1615
"github.com/cockroachdb/cockroach/pkg/util/log"
1716
"github.com/cockroachdb/cockroach/pkg/util/retry"
1817
"github.com/cockroachdb/errors"
@@ -52,7 +51,7 @@ var errRetry = errors.New("retry: orphaned replica")
5251
//
5352
// The caller must not hold the store's lock.
5453
func (s *Store) getOrCreateReplica(
55-
ctx context.Context, id storage.FullReplicaID, creatingReplica *roachpb.ReplicaDescriptor,
54+
ctx context.Context, id roachpb.FullReplicaID, creatingReplica *roachpb.ReplicaDescriptor,
5655
) (_ *Replica, created bool, _ error) {
5756
if id.ReplicaID == 0 {
5857
log.Fatalf(ctx, "cannot construct a Replica for range %d with 0 id", id.RangeID)
@@ -85,7 +84,7 @@ func (s *Store) getOrCreateReplica(
8584
// removed. Returns errRetry error if the replica is in a transitional state and
8685
// its retrieval needs to be retried. Other errors are permanent.
8786
func (s *Store) tryGetReplica(
88-
ctx context.Context, id storage.FullReplicaID, creatingReplica *roachpb.ReplicaDescriptor,
87+
ctx context.Context, id roachpb.FullReplicaID, creatingReplica *roachpb.ReplicaDescriptor,
8988
) (*Replica, error) {
9089
repl, found := s.mu.replicasByRangeID.Load(id.RangeID)
9190
if !found {
@@ -149,7 +148,7 @@ func (s *Store) tryGetReplica(
149148
// tryGetOrCreateReplica will likely succeed, hence the loop in
150149
// getOrCreateReplica.
151150
func (s *Store) tryGetOrCreateReplica(
152-
ctx context.Context, id storage.FullReplicaID, creatingReplica *roachpb.ReplicaDescriptor,
151+
ctx context.Context, id roachpb.FullReplicaID, creatingReplica *roachpb.ReplicaDescriptor,
153152
) (_ *Replica, created bool, _ error) {
154153
// The common case: look up an existing replica.
155154
if repl, err := s.tryGetReplica(ctx, id, creatingReplica); err != nil {

0 commit comments

Comments
 (0)