From a62450acd6f2a2bb8becf8159d25487bff3eaa2d Mon Sep 17 00:00:00 2001
From: Norbert Hu <norbert@temporal.io>
Date: Tue, 20 Jun 2023 16:11:40 -0700
Subject: [PATCH] Bring back branch token for branch deletion (#4521)

---
 common/persistence/history_manager.go      | 1 +
 common/persistence/persistenceInterface.go | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/common/persistence/history_manager.go b/common/persistence/history_manager.go
index 7defabfaa75..f058f90f440 100644
--- a/common/persistence/history_manager.go
+++ b/common/persistence/history_manager.go
@@ -213,6 +213,7 @@ findDeleteRanges:
 	}
 
 	req := &InternalDeleteHistoryBranchRequest{
+		BranchToken:  request.BranchToken,
 		BranchInfo:   branch,
 		ShardID:      request.ShardID,
 		BranchRanges: deleteRanges,
diff --git a/common/persistence/persistenceInterface.go b/common/persistence/persistenceInterface.go
index 7e2ee10f1c5..e23ce20f1ae 100644
--- a/common/persistence/persistenceInterface.go
+++ b/common/persistence/persistenceInterface.go
@@ -563,6 +563,8 @@ type (
 
 	// InternalDeleteHistoryBranchRequest is used to remove a history branch
 	InternalDeleteHistoryBranchRequest struct {
+		// The raw branch token
+		BranchToken []byte
 		// The branch
 		BranchInfo *persistencespb.HistoryBranch
 		// Used in sharded data stores to identify which shard to use