Merge pull request #807 from ydb-platform/ReadFromSnapshot

Added `table/options.ReadFromSnapshot(bool)` option for `session.StreamReadTable()`

Author: asmyasnikov

CHANGELOG.md

@@ -1,3 +1,5 @@
+* Added `table/options.ReadFromSnapshot(bool)` option for `session.StreamReadTable()`
+
 ## v3.51.1
 * Added checking condition for `tx.Rollback()` in `retry.DoTx`
 

table/options/options.go

@@ -1012,6 +1012,7 @@ type (
 
 	readColumnsOption        []string
 	readOrderedOption        struct{}
+	readSnapshotOption       bool
 	readKeyRangeOption       KeyRange
 	readGreaterOrEqualOption struct{ types.Value }
 	readLessOrEqualOption    struct{ types.Value }
@@ -1053,10 +1054,18 @@ func (columns readColumnsOption) ApplyReadTableOption(desc *ReadTableDesc, a *al
 	desc.Columns = append(desc.Columns, columns...)
 }
 
-func (r readOrderedOption) ApplyReadTableOption(desc *ReadTableDesc, a *allocator.Allocator) {
+func (readOrderedOption) ApplyReadTableOption(desc *ReadTableDesc, a *allocator.Allocator) {
 	desc.Ordered = true
 }
 
+func (b readSnapshotOption) ApplyReadTableOption(desc *ReadTableDesc, a *allocator.Allocator) {
+	if b {
+		desc.UseSnapshot = FeatureEnabled.ToYDB()
+	} else {
+		desc.UseSnapshot = FeatureDisabled.ToYDB()
+	}
+}
+
 func (x readKeyRangeOption) ApplyReadTableOption(desc *ReadTableDesc, a *allocator.Allocator) {
 	desc.initKeyRange()
 	if x.From != nil {
@@ -1086,10 +1095,14 @@ func ReadColumns(names ...string) readColumnsOption {
 	return names
 }
 
-func ReadOrdered() readOrderedOption {
+func ReadOrdered() ReadTableOption {
 	return readOrderedOption{}
 }
 
+func ReadFromSnapshot(b bool) ReadTableOption {
+	return readSnapshotOption(b)
+}
+
 // ReadKeyRange returns ReadTableOption which makes ReadTable read values
 // in range [x.From, x.To).
 //