feat(spanner): add method (#8945)

Co-authored-by: rahul2393 <[email protected]>
googleapis · Nov 3, 2023 · 5c778b2 · 5c778b2
1 parent 4dcd7dd
commit 5c778b2
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
+}