@@ -51,6 +51,11 @@ var headerStoreReset = &cobra.Command{
5151 if err != nil {
5252 return err
5353 }
54+ defer func () {
55+ if err := s .Close (); err != nil {
56+ fmt .Printf ("Error closing node store: %v\n " , err )
57+ }
58+ }()
5459
5560 ds , err := s .Datastore ()
5661 if err != nil {
@@ -61,6 +66,14 @@ var headerStoreReset = &cobra.Command{
6166 if err != nil {
6267 return fmt .Errorf ("opening header store: %w" , err )
6368 }
69+ if err = hstore .Start (ctx ); err != nil {
70+ return err
71+ }
72+ defer func () {
73+ if err := hstore .Stop (ctx ); err != nil {
74+ fmt .Printf ("Error stopping header store: %v\n " , err )
75+ }
76+ }()
6477
6578 if head != 0 {
6679 err := store .UnsafeResetHead (ctx , hstore , head )
@@ -77,7 +90,7 @@ var headerStoreReset = &cobra.Command{
7790 fmt .Println ("Reset header store tail to height" , tail )
7891 }
7992
80- return s . Close ()
93+ return nil
8194 },
8295}
8396
@@ -97,6 +110,11 @@ var headerStoreRecover = &cobra.Command{
97110 if err != nil {
98111 return err
99112 }
113+ defer func () {
114+ if err := s .Close (); err != nil {
115+ fmt .Printf ("Error closing node store: %v\n " , err )
116+ }
117+ }()
100118 ds , err := s .Datastore ()
101119 if err != nil {
102120 return err
@@ -106,6 +124,14 @@ var headerStoreRecover = &cobra.Command{
106124 if err != nil {
107125 return fmt .Errorf ("opening header store: %w" , err )
108126 }
127+ if err = hstore .Start (ctx ); err != nil {
128+ return err
129+ }
130+ defer func () {
131+ if err := hstore .Stop (ctx ); err != nil {
132+ fmt .Printf ("Error stopping header store: %v\n " , err )
133+ }
134+ }()
109135
110136 startFrom , err := cmd .Flags ().GetUint64 (startFromFlag )
111137 if err != nil {
@@ -124,7 +150,7 @@ var headerStoreRecover = &cobra.Command{
124150 }
125151 fmt .Println ("Reset header store tail to height" , hdr .Height ())
126152
127- return s . Close ()
153+ return nil
128154 },
129155}
130156
0 commit comments