use time.After instead of sleep, incase of goroutine leak

Signed-off-by: jsvisa <delweng@gmail.com>

core/state/statesize.go

@@ -155,18 +155,19 @@ func (g *StateSizeGenerator) initialize() chan struct{} {
 	LOOP:
 		// Wait for snapshot generator to complete first
 		for {
+			root, done := g.triedb.SnapshotCompleted()
+			if done {
+				g.metrics.Root = root
+				g.buffered.Root = root
+				break LOOP
+			}
+
 			select {
 			case <-g.abort:
 				log.Info("State size initialization aborted during snapshot wait")
 				return
-			default:
+			case <-time.After(10 * time.Second):
-				root, done := g.triedb.SnapshotCompleted()
+				// Continue checking for snapshot completion
-				if done {
-					g.metrics.Root = root
-					g.buffered.Root = root
-					break LOOP
-				}
-				time.Sleep(10 * time.Second)
 			}
 		}