puller: fix remove path (#973)

pingcap · Feb 8, 2025 · 1ef9ddc · 1ef9ddc
1 parent d8a1f14
commit 1ef9ddc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/logservice/logpuller/subscription_client.go b/logservice/logpuller/subscription_client.go
@@ -324,7 +324,6 @@ func (s *SubscriptionClient) Unsubscribe(subID SubscriptionID) {
 		log.Warn("unknown subscription", zap.Uint64("subscriptionID", uint64(subID)))
 		return
 	}
-	s.ds.RemovePath(rt.subID)
 	s.setTableStopped(rt)
 
 	log.Info("unsubscribe span success",
@@ -398,6 +397,7 @@ func (s *SubscriptionClient) onTableDrained(rt *subscribedSpan) {
 
 	s.totalSpans.Lock()
 	defer s.totalSpans.Unlock()
+	s.ds.RemovePath(rt.subID)
 	delete(s.totalSpans.spanMap, rt.subID)
 }