Supervisar el estado del trabajo con notificaciones de Pub/Sub y BigQuery

En este documento, se explica cómo puedes supervisar los cambios de estado de los trabajos y las tareas configurando notificaciones de Pub/Sub. Puedes usar las notificaciones para casos de uso; p. ej., en este documento, se explica cómo transmitir notificaciones a una tabla de BigQuery, que puedes usar para analizar el de datos no estructurados. Para obtener más información sobre los casos de uso de las notificaciones de Pub/Sub, consulta Elige un tipo de suscripción en la documentación de Pub/Sub.

Como alternativa, si quieres exportar toda la información sobre un trabajo a BigQuery, consulta Exporta información del trabajo. en su lugar.

Antes de comenzar

  1. Si nunca usaste Batch, revisa Comienza a usar Batch y habilitar Batch completando el requisitos previos para los proyectos y usuarios.
  2. Si deseas configurar notificaciones de Pub/Sub para tus trabajos por lotes, haz lo siguiente:
    1. Enable the Pub/Sub API.

      Enable the API

    2. Para obtener los permisos que necesitas para configurar un tema y una suscripción de Pub/Sub, pídele a tu administrador que te otorgue el rol de IAM de editor de Pub/Sub (roles/pubsub.editor) en un proyecto. Para obtener más información sobre cómo otorgar roles, consulta Administra el acceso a proyectos, carpetas y organizaciones.

      También puedes obtener los permisos necesarios mediante roles personalizados o cualquier otro rol predefinido.

  3. Los requisitos adicionales varían según el tipo de suscripción que decidas configurar. Por ejemplo, si deseas transmitir y analizar notificaciones en una tabla de BigQuery, debes hacer lo siguiente:
    1. Enable the BigQuery API.

      Enable the API

    2. Para garantizar que la cuenta de servicio de Pub/Sub tenga los permisos necesarios para transmitir notificaciones de Pub/Sub a BigQuery, pídele a tu administrador que le otorgue a la cuenta de servicio de Pub/Sub el rol de IAM de Editor de datos de BigQuery (roles/bigquery.dataEditor) en el proyecto del tema de Pub/Sub.

      Para obtener más información sobre cómo asignar este rol, consulta Asigna roles de BigQuery a la cuenta de servicio de Pub/Sub.

    3. Asegúrate de que los usuarios para los que deseas analizar las notificaciones en BigQuery tengan los permisos necesarios.

      Para obtener los permisos que necesitas para consultar notificaciones de Pub/Sub en BigQuery, solicita a tu administrador que te otorgue el los siguientes roles de IAM:

Configura notificaciones de Pub/Sub para Batch

Por lotes, admite notificaciones de Pub/Sub para cambios en los estados de trabajos y tareas, que puedes usar para alertas, observabilidad o análisis.

Para configurar notificaciones de Pub/Sub para uno o más trabajos de lotes, completa los pasos que se describen en las siguientes secciones:

  1. Crea o identifica un tema de Pub/Sub para tus trabajos de Batch.
  2. Crea al menos una suscripción para que tu tema reciba y use sus notificaciones.
  3. Configura trabajos para enviar notificaciones a tu tema.

Después de configurar las notificaciones de Pub/Sub para Batch, puedes configurar de manera opcional cada trabajo para que envíe notificaciones específicas. Puedes recibir notificaciones cuando los estados de los trabajos o tareas cambien, o puedes filtrar el notificaciones para estados específicos de trabajos o tareas. En particular, Pub/Sub no Enviar notificaciones del primer estado en el que ingresa un trabajo o una tarea Pub/Sub solo envía notificaciones cuando un trabajo pasa de un estado a otro. Para obtener más información sobre los atributos que puedes configurar para las notificaciones de Pub/Sub, consulta Cómo crear y ejecutar una tarea que envíe notificaciones de estado de Pub/Sub.

Crea o identifica un tema de Pub/Sub para lotes

Crea un tema de Pub/Sub nuevo o identifica un tema de Pub/Sub existente que quieras usar para las notificaciones por lotes.

Como se explica más adelante en este documento, los trabajos no pueden enviar notificaciones a este tema de Pub/Sub hasta que el administrador otorgue permisos que lo permiten.

Crea una suscripción para recibir y usar las notificaciones

Para recibir y usar mensajes publicados en un tema de Pub/Sub, debes crear al menos una suscripción a ese tema.

Por ejemplo, una opción es configurar tu tema para que transmita sus notificaciones a BigQuery. Luego, cuando ejecutes trabajos publicar notificaciones, puedes recibirlas y analizarlas en un en la tabla de BigQuery. Para obtener más información, consulta Ejemplo: Recibe y usa notificaciones de Pub/Sub en BigQuery en este documento.

De lo contrario, para obtener más información sobre las suscripciones y otras opciones, consulta la descripción general de las suscripciones en la documentación de Pub/Sub.

Configura trabajos para enviar notificaciones a tu tema

Para configurar trabajos que envíen notificaciones de Pub/Sub a tu tema, sigue las instrucciones que se indican en la página Crea y ejecuta un trabajo que envíe notificaciones de estado de Pub/Sub. En particular, haz lo siguiente:

  1. Si aún no lo hiciste, pídele al administrador que otorgue todos los permisos necesarios para que las tareas envíen notificaciones a tu tema de Pub/Sub. Estos permisos se enumeran en el Sección Roles obligatorios de esa página.
  2. Crear y ejecutar un trabajo que habilite las notificaciones de Pub/Sub Definiendo el campo notifications del trabajo

Ejemplo: Recibe y usa notificaciones de Pub/Sub en BigQuery

En las siguientes secciones, se explica un caso de uso de ejemplo para recibir y usar Notificaciones de Pub/Sub en BigQuery. Con la configuración de un suscripción que transmite las notificaciones de Pub/Sub de tu tema a un tabla de BigQuery, puedes retener el historial del estado de los trabajos y las tareas y consultarlos.

Configurar una suscripción que transmita notificaciones a BigQuery

Para configurar una suscripción a Pub/Sub que transmita notificaciones de un tema a una tabla de BigQuery, haz lo siguiente:

  1. Si aún no lo has hecho, Crea o identifica un tema de Pub/Sub para Batch.

  2. Si aún no tienes uno, crea un conjunto de datos de BigQuery.

  3. Crea una tabla de BigQuery con la siguiente definición de esquema:

    [
      {
        "name": "data",
        "type": "STRING"
      },
      {
        "name": "subscription_name",
        "type": "STRING"
      },
      {
        "name": "message_id",
        "type": "STRING"
      },
      {
        "name": "attributes",
        "type": "JSON"
      },
      {
        "name": "publish_time",
        "type": "TIMESTAMP",
        "mode": "NULLABLE"
      }
    ]
    
  4. Crea una suscripción a Pub/Sub que se transmita a BigQuery y especifica las siguientes opciones:

    • Tema: Selecciona el tema de Pub/Sub para Batch.
    • Proyecto: Selecciona el proyecto que contiene tu tabla de BigQuery.
    • Conjunto de datos: Selecciona el conjunto de datos que contiene tu tabla de BigQuery.
    • Tabla: Selecciona la tabla que creaste en el paso anterior.
    • Selecciona Usar el esquema de tema.
    • Selecciona Escribir metadatos.

Ahora, cada vez que se envía una notificación de Pub/Sub a ese tema, Pub/Sub transmite esa notificación a la tabla de BigQuery. Después de configurar trabajos para enviar notificaciones de Pub/Sub, puedes analizar las notificaciones en BigQuery.

Analiza notificaciones en BigQuery

Puedes analizar las notificaciones en BigQuery después de configurar una suscripción a una tabla de BigQuery para tu tema y ejecutar trabajos que envíen notificaciones a tu tema. Para analizar las notificaciones en BigQuery, ejecuta una consulta en la columna attributes de la tabla de BigQuery. Los atributos que puedes consultar dependen de si configuraste notificaciones para tareas o trabajos.

Puedes consultar los siguientes atributos de las notificaciones de tareas:

  • JobUID
  • NewTaskState
  • Region
  • TaskName
  • TaskUID
  • Type

Puedes consultar los siguientes atributos de notificaciones de trabajo:

  • JobUID
  • JobName
  • NewJobState
  • Region
  • Type

Por ejemplo, la siguiente consulta muestra el JobUID asociado a cada notificación:

select attributes.JobUID from `example-table`

¿Qué sigue?