Properly initialize reload health

viant · Mar 29, 2023 · e0c3830 · e0c3830
1 parent 66b6a42
commit e0c3830
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/service/service.go b/service/service.go
@@ -278,6 +278,9 @@ func (s *Service) reloadIfNeeded(ctx context.Context) error {
 	s.dictionary = dictionary
 	s.inputs = inputs
 	s.config.Modified = snapshot
+
+	atomic.StoreInt32(&s.ReloadOK, 1)
+
 	return nil
 }
 
@@ -429,8 +432,6 @@ func (s *Service) scheduleModelReload() {
 		if err := s.reloadIfNeeded(context.Background()); err != nil {
 			fmt.Printf("failed to reload model: %v, due to %v", s.config.ID, err)
 			atomic.StoreInt32(&s.ReloadOK, 0)
-		} else {
-			atomic.StoreInt32(&s.ReloadOK, 1)
 		}
 
 		if atomic.LoadInt32(&s.closed) != 0 {