chore: block for controller to undeploy before delete

instill-ai · Aug 18, 2023 · 2481315 · 2481315
1 parent dd6f100
commit 2481315
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/service/service.go b/pkg/service/service.go
@@ -8,6 +8,7 @@ import (
 	"path/filepath"
 	"strconv"
 	"strings"
+	"time"
 
 	"cloud.google.com/go/longrunning/autogen/longrunningpb"
 	"github.com/go-redis/redis/v9"
@@ -706,6 +707,13 @@ func (s *service) DeleteModel(ctx context.Context, owner string, modelID string)
 		}
 	}
 
+	for state.String() == modelPB.Model_STATE_ONLINE.String() {
+		if state, err = s.GetResourceState(ctx, modelInDB.UID); err != nil {
+			return err
+		}
+		time.Sleep(100 * time.Millisecond)
+	}
+
 	if err := s.DeleteResourceState(ctx, modelInDB.UID); err != nil {
 		return err
 	}