fix: wait for ipam grpc server to be Ready

Author: fra98

cmd/liqo-controller-manager/main.go

@@ -58,6 +58,7 @@ import (
 	dynamicutils "github.com/liqotech/liqo/pkg/utils/dynamic"
 	liqoerrors "github.com/liqotech/liqo/pkg/utils/errors"
 	flagsutils "github.com/liqotech/liqo/pkg/utils/flags"
+	grpcutils "github.com/liqotech/liqo/pkg/utils/grpc"
 	"github.com/liqotech/liqo/pkg/utils/indexer"
 	ipamips "github.com/liqotech/liqo/pkg/utils/ipam/mapping"
 	"github.com/liqotech/liqo/pkg/utils/mapper"
@@ -256,7 +257,15 @@ func main() {
 				klog.Errorf("failed to establish a connection to the IPAM %q", *ipamServer)
 				os.Exit(1)
 			}
+
+			if err := grpcutils.WaitForConnectionReady(ctx, conn, 10*time.Second); err != nil {
+				klog.Errorf("failed to establish a connection to the IPAM server %q", *ipamServer)
+				os.Exit(1)
+			}
+			klog.Infof("IPAM connected: %s", conn.GetState())
+
 			defer conn.Close()
+
 			ipamClient = ipam.NewIPAMClient(conn)
 		}
 
@@ -278,7 +287,8 @@ func main() {
 
 			GenevePort: *genevePort,
 		}); err != nil {
-			klog.Fatalf("Unable to setup the networking module: %v", err)
+			klog.Errorf("Unable to setup the networking module: %v", err)
+			os.Exit(1)
 		}
 	}
 

pkg/utils/grpc/connection.go

@@ -0,0 +1,15 @@
+// Copyright 2019-2024 The Liqo Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package grpc

pkg/utils/grpc/doc.go

@@ -0,0 +1,49 @@
+// Copyright 2019-2024 The Liqo Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Package grpc provides a set of utilities to manage grpc connections.
+package grpc
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/connectivity"
+)
+
+// WaitForConnectionReady waits for the connection to a gRPC server to be Ready.
+func WaitForConnectionReady(ctx context.Context, conn *grpc.ClientConn, timeout time.Duration) error {
+	// Add a contxt with a timeout to avoid waiting indefinitely.
+	ctx, cancel := context.WithTimeout(ctx, timeout)
+	defer cancel()
+
+	// Attempt to connect to the gRPC server.
+	conn.Connect()
+
+	for {
+		state := conn.GetState()
+		if state == connectivity.Ready {
+			break
+		}
+
+		changed := conn.WaitForStateChange(ctx, state)
+		if !changed {
+			return fmt.Errorf("timeout connection to the gRPC server")
+		}
+	}
+
+	return nil
+}