fixup

provider/grpc/provider.go

@@ -7,6 +7,7 @@ import (
 	"io"
 	"log"
 	"os/exec"
+	"time"
 
 	"github.com/go-logr/logr"
 	reflectClient "github.com/jhump/protoreflect/grpcreflect"
@@ -41,12 +42,9 @@ func NewGRPCClient(config provider.Config, log logr.Logger) (provider.InternalPr
 	}
 	refCltCtx, cancel := context.WithCancel(context.Background())
 	refClt := reflectClient.NewClientAuto(refCltCtx, conn)
+	defer cancel()
 
-	// TODO: reasonable retry here
-	services, err := refClt.ListServices()
-	if err != nil {
-		return nil, err
-	}
+	services, err := checkServicesRunning(refClt, log)
 	foundCodeSnip := false
 	foundDepResolve := false
 	for _, s := range services {
@@ -58,8 +56,6 @@ func NewGRPCClient(config provider.Config, log logr.Logger) (provider.InternalPr
 			foundDepResolve = true
 		}
 	}
-	// Close the reflection service connection
-	cancel()
 	// Always need these
 	provierClient := pb.NewProviderServiceClient(conn)
 	gp := grpcProvider{
@@ -132,6 +128,25 @@ func NewGRPCClient(config provider.Config, log logr.Logger) (provider.InternalPr
 	// }
 }
 
+func checkServicesRunning(refClt *reflectClient.Client, log logr.Logger) ([]string, error) {
+	for {
+		select {
+		default:
+			services, err := refClt.ListServices()
+			if err == nil && len(services) != 0 {
+				log.Info("no services found")
+				return services, nil
+			}
+			if err != nil {
+				log.Info("error for list services: ", "error", err)
+			}
+			time.Sleep(3 * time.Second)
+		case <-time.After(time.Second * 30):
+			return nil, fmt.Errorf("no services found")
+		}
+	}
+}
+
 func (g *grpcProvider) ProviderInit(ctx context.Context) error {
 	for _, c := range g.config.InitConfig {
 		s, err := g.Init(ctx, g.log, c)