Skip to content

Commit 9128ba5

Browse files
authored
fix(cel-shed): start the store (celestiaorg#4566)
1 parent 5782cce commit 9128ba5

File tree

1 file changed

+28
-2
lines changed

1 file changed

+28
-2
lines changed

cmd/cel-shed/header.go

Lines changed: 28 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)