save connection status when resetting file manager service

Author: sean-breen

internal/file/file_manager_service.go

@@ -54,6 +54,7 @@ type (
 		UpdateCurrentFilesOnDisk(updateFiles map[string]*mpi.File)
 		DetermineFileActions(currentFiles, modifiedFiles map[string]*mpi.File) (map[string]*mpi.File,
 			map[string][]byte, error)
+		IsConnected() bool
 		SetIsConnected(isConnected bool)
 	}
 )
@@ -271,6 +272,10 @@ func (fms *FileManagerService) UpdateFile(
 	return err
 }
 
+func (fms *FileManagerService) IsConnected() bool {
+	return fms.isConnected.Load()
+}
+
 func (fms *FileManagerService) SetIsConnected(isConnected bool) {
 	fms.isConnected.Store(isConnected)
 }

internal/file/file_plugin.go

@@ -97,12 +97,17 @@ func (fp *FilePlugin) Subscriptions() []string {
 func (fp *FilePlugin) handleConnectionReset(ctx context.Context, msg *bus.Message) {
 	slog.DebugContext(ctx, "File plugin received connection reset message")
 	if newConnection, ok := msg.Data.(grpc.GrpcConnectionInterface); ok {
+		var reconnect bool
 		err := fp.conn.Close(ctx)
 		if err != nil {
 			slog.ErrorContext(ctx, "File plugin: unable to close connection", "error", err)
 		}
 		fp.conn = newConnection
+
+		reconnect = fp.fileManagerService.IsConnected()
 		fp.fileManagerService = NewFileManagerService(fp.conn.FileServiceClient(), fp.config)
+		fp.fileManagerService.SetIsConnected(reconnect)
+
 		slog.DebugContext(ctx, "File plugin: client reset successfully")
 	}
 }