feat(spanner): add method (#8945)

Co-authored-by: rahul2393 <irahul@google.com>
googleapis · Nov 2, 2023 · 411a51e · 411a51e
1 parent cc76ab3
commit 411a51e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/spanner/batch.go b/spanner/batch.go
@@ -449,3 +449,11 @@ func (p *Partition) UnmarshalBinary(data []byte) error {
 	}
 	return err
 }
+
+// GetPartitionToken returns partition token
+func (p *Partition) GetPartitionToken() []byte {
+	if p != nil {
+		return p.pt
+	}
+	return nil
+}