From db3e02e921390c89e7191fac509106b411498379 Mon Sep 17 00:00:00 2001
From: ucwong <ucwong@126.com>
Date: Fri, 28 Mar 2025 19:11:48 +0800
Subject: [PATCH] core/txindexer : add wait group

---
 core/txindexer.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/core/txindexer.go b/core/txindexer.go
index d0fce302f3..cd8f5e019d 100644
--- a/core/txindexer.go
+++ b/core/txindexer.go
@@ -19,6 +19,7 @@ package core
 import (
 	"errors"
 	"fmt"
+	"sync"
 
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core/rawdb"
@@ -54,6 +55,8 @@ type txIndexer struct {
 	progress chan chan TxIndexProgress
 	term     chan chan struct{}
 	closed   chan struct{}
+
+	wg sync.WaitGroup
 }
 
 // newTxIndexer initializes the transaction indexer.
@@ -66,6 +69,7 @@ func newTxIndexer(limit uint64, chain *BlockChain) *txIndexer {
 		term:     make(chan chan struct{}),
 		closed:   make(chan struct{}),
 	}
+	indexer.wg.Add(1)
 	go indexer.loop(chain)
 
 	var msg string
@@ -199,6 +203,7 @@ func (indexer *txIndexer) repair(head uint64) {
 // on the received chain event.
 func (indexer *txIndexer) loop(chain *BlockChain) {
 	defer close(indexer.closed)
+	defer indexer.wg.Done()
 
 	// Listening to chain events and manipulate the transaction indexes.
 	var (
@@ -306,4 +311,6 @@ func (indexer *txIndexer) close() {
 		<-ch
 	case <-indexer.closed:
 	}
+
+	indexer.wg.Wait()
 }