fix datarace

Signed-off-by: lhy1024 <admin@liudos.us>
tikv · Jun 29, 2023 · 89f99e8 · 89f99e8
1 parent 38a1cc5
commit 89f99e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/tso/keyspace_group_manager.go b/pkg/tso/keyspace_group_manager.go
@@ -109,8 +109,12 @@ func (s *state) getKeyspaceGroupMeta(
 ) (*AllocatorManager, *endpoint.KeyspaceGroup) {
 	s.RLock()
 	defer s.RUnlock()
+	am := s.ams[groupID]
+	if s.kgs[groupID] == nil {
+		return am, nil
+	}
 	kg := *s.kgs[groupID] // copy the keyspace group meta to avoid datarace
-	return s.ams[groupID], &kg
+	return am, &kg
 }
 
 // getKeyspaceGroupMetaWithCheck returns the keyspace group meta of the given keyspace.