[GitHub] [airflow] turbaszek commented on issue #11983: airflow.contrib.operators.pubsub_operator.PubSubPublishOperator can't handle multiple messages in a template

[GitBox <gi...@apache.org>]

turbaszek commented on issue #11983:
URL: https://github.com/apache/airflow/issues/11983#issuecomment-759269669


   It seems that in 2.0 you can use `output` attribute to do what you want:
   ```py
   class CustomOp(BaseOperator):
       @apply_defaults
       def __init__(self, **kwargs):
           super().__init__(**kwargs)
   
       def execute(self, context: Any):
           return [{"a": 1, "b": 2} for _ in range(5)]
   
   
   class CustomOp2(BaseOperator):
       template_fields = ["xs"]
   
       @apply_defaults
       def __init__(self, xs, **kwargs):
           super().__init__(**kwargs)
           self.xs = xs
   
       def execute(self, context: Any):
           self.log.info(self.xs)
           for x in self.xs:
               self.log.info(x)
   
   
   with DAG(
       'test_xcom_native',
       default_args=default_args
   ) as dag2:
       op = CustomOp(task_id="generate")
       CustomOp2(task_id="print", xs=op.output)
   ```
   
   and you get:
   ```
   [2021-01-13 07:44:12,266] {xcom.py:52} INFO - [{'a': 1, 'b': 2}, {'a': 1, 'b': 2}, {'a': 1, 'b': 2}, {'a': 1, 'b': 2}, {'a': 1, 'b': 2}]
   [2021-01-13 07:44:12,267] {xcom.py:54} INFO - {'a': 1, 'b': 2}
   [2021-01-13 07:44:12,269] {xcom.py:54} INFO - {'a': 1, 'b': 2}
   [2021-01-13 07:44:12,270] {xcom.py:54} INFO - {'a': 1, 'b': 2}
   [2021-01-13 07:44:12,271] {xcom.py:54} INFO - {'a': 1, 'b': 2}
   [2021-01-13 07:44:12,272] {xcom.py:54} INFO - {'a': 1, 'b': 2}
   ```


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org