OnStop

sei-protocol · Apr 24, 2023 · 023605a · 023605a
1 parent 18c9af7
commit 023605a
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/internal/consensus/reactor_test.go b/internal/consensus/reactor_test.go
@@ -232,7 +232,7 @@ func waitForAndValidateBlockWithTx(
 ) {
 	t.Helper()
 
-	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Minute)
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Minute)
 	defer cancel()
 
 	fn := func(j int) {

diff --git a/internal/dbsync/reactor.go b/internal/dbsync/reactor.go
@@ -223,6 +223,8 @@ func (r *Reactor) OnStart(ctx context.Context) error {
 func (r *Reactor) OnStop() {
 	// tell the dispatcher to stop sending any more requests
 	r.dispatcher.Close()
+	// clear up half-populated directories
+	r.syncer.Stop()
 }
 
 func (r *Reactor) handleMetadataRequest(ctx context.Context, req *dstypes.MetadataRequest, from types.NodeID) (err error) {

diff --git a/internal/dbsync/syncer.go b/internal/dbsync/syncer.go
@@ -180,6 +180,15 @@ func (s *Syncer) Process(ctx context.Context) {
 	}
 }
 
+func (s *Syncer) Stop() {
+	s.mtx.Lock()
+	defer s.mtx.Unlock()
+	if s.active {
+		s.resetDirFn(s)
+		s.active = false
+	}
+}
+
 func (s *Syncer) processFile(ctx context.Context, file *dstypes.FileResponse) error {
 	s.mtx.Lock()
 	defer s.mtx.Unlock()