Add Revert in DataCoord meta catalog (milvus-io#19313)

See also: milvus-io#19072 Signed-off-by: yangxuan <[email protected]> Signed-off-by: yangxuan <[email protected]>
sunby · Sep 23, 2022 · 253fb51 · 253fb51
1 parent 432b5dd
commit 253fb51
Show file tree

Hide file tree

Showing 5 changed files with 445 additions and 39 deletions.
diff --git a/Makefile b/Makefile
@@ -259,3 +259,9 @@ rpm: install
 	@cp -r configs ~/rpmbuild/BUILD/
 	@cp -r build/rpm/services ~/rpmbuild/BUILD/
 	@QA_RPATHS="$$[ 0x001|0x0002|0x0020 ]" rpmbuild -ba ./build/rpm/milvus.spec
+
+mock-datanode:
+	mockery --name=DataNode --dir=$(PWD)/internal/types --output=$(PWD)/internal/mocks --filename=mock_datanode.go --with-expecter
+
+mock-tnx-kv:
+	mockery --name=TxnKV --dir=$(PWD)/internal/kv --output=$(PWD)/internal/kv/mocks --filename=TxnKV.go --with-expecter
diff --git a/internal/kv/mocks/TxnKV.go b/internal/kv/mocks/TxnKV.go
diff --git a/internal/metastore/catalog.go b/internal/metastore/catalog.go
@@ -79,6 +79,8 @@ type DataCoordCatalog interface {
 	MarkChannelDeleted(ctx context.Context, channel string) error
 	IsChannelDropped(ctx context.Context, channel string) bool
 	DropChannel(ctx context.Context, channel string) error
+
+	RevertAlterSegmentsAndAddNewSegment(ctx context.Context, segments []*datapb.SegmentInfo, removalSegment *datapb.SegmentInfo) error
 }
 
 type IndexCoordCatalog interface {