updated workflow loading

viant · Oct 10, 2022 · c22f7e3 · c22f7e3
1 parent 72f44cb
commit c22f7e3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/service/endpoint/service.go b/service/endpoint/service.go
@@ -64,9 +64,11 @@ func (s *Service) initModelHandler(datastores map[string]*datastore.Service, poo
 	var lock sync.Mutex
 	for i := range s.config.ModelList.Models {
 		go func(model *config.Model) {
-			waitGroup.Done()
+			defer waitGroup.Done()
 			if e := s.initModel(datastores, pool, mux, model, &lock); e != nil {
+				lock.Lock()
 				err = e
+				lock.Unlock()
 			}
 		}(s.config.ModelList.Models[i])
 	}