Gets the tworker queue override from environment or metadata.
()
| 140 | |
| 141 | @memoize.wrap(memoize.FifoInMemory(1)) |
| 142 | def _get_tworker_queue_override() -> str: |
| 143 | """Gets the tworker queue override from environment or metadata.""" |
| 144 | queue_override = environment.get_value('OVERRIDE_TWORKER_QUEUE') |
| 145 | if queue_override: |
| 146 | return queue_override |
| 147 | |
| 148 | if not compute_metadata.is_gce(): |
| 149 | return "" |
| 150 | |
| 151 | try: |
| 152 | queue_override = compute_metadata.get( |
| 153 | 'instance/attributes/override_tworker_queue') |
| 154 | if queue_override: |
| 155 | return queue_override.strip() |
| 156 | except exceptions.RequestException as e: |
| 157 | if not (isinstance(e, exceptions.HTTPError) and |
| 158 | e.response.status_code == 404): |
| 159 | logs.warning(f'Error fetching override_tworker_queue metadata: {e}') |
| 160 | |
| 161 | return "" |
| 162 | |
| 163 | |
| 164 | def task_loop(): |