Airflow Custom MySQL Оператор get_records с параметрами сеанса - PullRequest
0 голосов
/ 06 мая 2020

Я пытаюсь выполнить запрос MySQL в воздушном потоке, который требует, чтобы параметр сеанса был установлен до выполнения запроса. Если я передам оба оператора вместе, оператор get_records будет ошибаться, потому что первый оператор не возвращает никаких данных. Как я могу установить этот параметр сеанса, а затем вернуть результаты моего запроса?

MySQL Пример:

SET SESSION group_concat_max_len=2294867295;
SELECT ...query to return data...;

Это соответствующий код воздушного потока:

    def get_records(self, query, hook):
    results = list(hook.get_records(query))
...