Ensure that a file overview update is performed even if the initial create connection request takes a long time

internal/file/file_manager_service.go

@@ -244,6 +244,7 @@ func (fms *FileManagerService) ConfigUpdate(ctx context.Context,
 			"error", err,
 		)
 	}
+	slog.InfoContext(ctx, "Finished updating file overview")
 }
 
 func (fms *FileManagerService) ConfigUpload(ctx context.Context, configUploadRequest *mpi.ConfigUploadRequest) error {

internal/file/file_service_operator.go

@@ -132,8 +132,31 @@ func (fso *FileServiceOperator) UpdateOverview(
 		},
 	}
 
-	backOffCtx, backoffCancel := context.WithTimeout(newCtx, fso.agentConfig.Client.Backoff.MaxElapsedTime)
-	defer backoffCancel()
+	backoffSettings := &config.BackOff{
+		InitialInterval:     fso.agentConfig.Client.Backoff.InitialInterval,
+		MaxInterval:         fso.agentConfig.Client.Backoff.MaxInterval,
+		MaxElapsedTime:      fso.agentConfig.Client.Backoff.MaxElapsedTime,
+		RandomizationFactor: fso.agentConfig.Client.Backoff.RandomizationFactor,
+		Multiplier:          fso.agentConfig.Client.Backoff.Multiplier,
+	}
+
+	// If the create connection takes a long time that we wait indefinitely to do
+	// the initial file overview update to ensure that the management plane has a file overview
+	// on agent startup.
+	if !fso.isConnected.Load() {
+		slog.DebugContext(
+			newCtx,
+			"Not connected to management plane yet, "+
+				"retrying indefinitely to update file overview until connection is created",
+		)
+		backoffSettings = &config.BackOff{
+			InitialInterval:     fso.agentConfig.Client.Backoff.InitialInterval,
+			MaxInterval:         fso.agentConfig.Client.Backoff.MaxInterval,
+			MaxElapsedTime:      0,
+			RandomizationFactor: fso.agentConfig.Client.Backoff.RandomizationFactor,
+			Multiplier:          fso.agentConfig.Client.Backoff.Multiplier,
+		}
+	}
 
 	sendUpdateOverview := func() (*mpi.UpdateOverviewResponse, error) {
 		if fso.fileServiceClient == nil {
@@ -162,10 +185,9 @@ func (fso *FileServiceOperator) UpdateOverview(
 		return response, nil
 	}
 
-	backoffSettings := fso.agentConfig.Client.Backoff
 	response, err := backoff.RetryWithData(
 		sendUpdateOverview,
-		backoffHelpers.Context(backOffCtx, backoffSettings),
+		backoffHelpers.Context(newCtx, backoffSettings),
 	)
 	if err != nil {
 		return err
@@ -174,11 +196,13 @@ func (fso *FileServiceOperator) UpdateOverview(
 	slog.DebugContext(newCtx, "UpdateOverview response", "response", response)
 
 	if response.GetOverview() == nil {
-		slog.DebugContext(ctx, "UpdateOverview response is empty")
+		slog.DebugContext(newCtx, "UpdateOverview response is empty")
 		return nil
 	}
 	delta := files.ConvertToMapOfFiles(response.GetOverview().GetFiles())
 
+	// Make sure that the original context is used if a file upload is required so that original correlation ID
+	// can be used again for update file overview request
 	if len(delta) != 0 {
 		return fso.updateFiles(ctx, delta, instanceID, configPath, iteration)
 	}

internal/file/file_service_operator_test.go

@@ -12,6 +12,7 @@ import (
 	"sync"
 	"sync/atomic"
 	"testing"
+	"time"
 
 	mpi "github.com/nginx/agent/v3/api/grpc/mpi/v1"
 	"github.com/nginx/agent/v3/api/grpc/mpi/v1/v1fakes"
@@ -97,6 +98,30 @@ func TestFileServiceOperator_UpdateOverview_MaxIterations(t *testing.T) {
 	assert.Equal(t, "too many UpdateOverview attempts", err.Error())
 }
 
+func TestFileServiceOperator_UpdateOverview_NoConnection(t *testing.T) {
+	ctx, cancel := context.WithTimeout(t.Context(), 500*time.Millisecond)
+	defer cancel()
+
+	filePath := filepath.Join(t.TempDir(), "nginx.conf")
+	fileMeta := protos.FileMeta(filePath, "")
+
+	fakeFileServiceClient := &v1fakes.FakeFileServiceClient{}
+
+	agentConfig := types.AgentConfig()
+	agentConfig.Client.Backoff.MaxElapsedTime = 200 * time.Millisecond
+
+	fileServiceOperator := NewFileServiceOperator(types.AgentConfig(), fakeFileServiceClient, &sync.RWMutex{})
+	fileServiceOperator.SetIsConnected(false)
+
+	err := fileServiceOperator.UpdateOverview(ctx, "123", []*mpi.File{
+		{
+			FileMeta: fileMeta,
+		},
+	}, filePath, 0)
+
+	assert.ErrorIs(t, err, context.DeadlineExceeded)
+}
+
 func TestFileManagerService_UpdateFile(t *testing.T) {
 	tests := []struct {
 		name   string