[Go] Add scan delete interface (#56)

Signed-off-by: sunby <[email protected]>
milvus-io · Sep 12, 2023 · c847928 · c847928
1 parent 6f78647
commit c847928
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/go/reader/record_reader/multi_files_sequential_reader.go b/go/reader/record_reader/multi_files_sequential_reader.go
@@ -56,6 +56,10 @@ func (m *MultiFilesSequentialReader) Next() bool {
 		if m.currReader.Next() {
 			return true
 		}
+		if m.currReader.Err() != nil {
+			m.err = m.currReader.Err()
+			return false
+		}
 		if m.currReader != nil {
 			m.currReader.Release()
 			m.currReader = nil

diff --git a/go/reader/record_reader/record_reader.go b/go/reader/record_reader/record_reader.go
@@ -74,3 +74,7 @@ func filtersOnlyContainPKAndVersion(s *schema.Schema, filters []filter.Filter) b
 	}
 	return true
 }
+
+func MakeScanDeleteReader(manifest *manifest.Manifest, fs fs.Fs) array.RecordReader {
+	return NewMultiFilesSequentialReader(fs, manifest.GetDeleteFragments(), manifest.GetSchema().DeleteSchema(), option.NewReadOptions())
+}
diff --git a/go/storage/space.go b/go/storage/space.go
@@ -471,3 +471,7 @@ func (s *Space) GetBlobByteSize(name string) (int64, error) {
 func (s *Space) GetCurrentVersion() int64 {
 	return s.manifest.Version()
 }
+
+func (s *Space) ScanDelete() (array.RecordReader, error) {
+	return record_reader.MakeScanDeleteReader(s.manifest, s.fs), nil
+}