Skip secret if row not found (#284)

beam-cloud · Jun 6, 2024 · 7f0191b · 7f0191b
1 parent 99b29b8
commit 7f0191b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/gateway/services/stub.go b/pkg/gateway/services/stub.go
@@ -2,6 +2,7 @@ package gatewayservices
 
 import (
 	"context"
+	"database/sql"
 
 	"github.com/beam-cloud/beta9/pkg/auth"
 	"github.com/beam-cloud/beta9/pkg/common"
@@ -40,6 +41,10 @@ func (gws *GatewayService) GetOrCreateStub(ctx context.Context, in *pb.GetOrCrea
 	for _, secret := range in.Secrets {
 		secret, err := gws.backendRepo.GetSecretByName(ctx, authInfo.Workspace, secret.Name)
 		if err != nil {
+			if err == sql.ErrNoRows {
+				continue // Skip secret if not found
+			}
+
 			return &pb.GetOrCreateStubResponse{
 				Ok: false,
 			}, nil