Я использую azure.kusto.ingest.QueuedIngestClient
для загрузки панд DataFrames в таблицы с помощью метода ingest_from_dataframe
, например.
client = QueuedIngestClient(...)
df = pd.DataFrame(...)
response = client.ingest_from_dataframe(df, ...)
response
имеет тип azure.kusto.ingest.base_ingest_client.IngestionResult
и имеет атрибут status
. Однако это всегда <IngestionStatus.QUEUED: 'QUEUED'>
даже после завершения загрузки (проверяется вручную через запрос к БД). Итак, как я узнаю в своей программе, когда прием данных завершен?
Внизу https://pypi.org/project/azure-kusto-ingest/ вы можете увидеть следующие ссылки:
В Примере загрузки данных вы можете найти полные примеры.
Вот полный рабочий пример (просто укажите URL вашего кластера и имя вашей базы данных) + результаты выполнения.
Этот код был выполнен на бесплатном кластере ADX
Обратите внимание
Проверка состояния асинхронного приема может отрицательно сказаться на производительности приема и кластера, особенно при приеме больших объемов данных.
На стороне кластера ADX:
.create table MyTable(id:long, name:string, value:real)
Код Python:
from azure.kusto.data import KustoConnectionStringBuilder
from azure.kusto.ingest import IngestionProperties, QueuedIngestClient, ReportLevel
from azure.kusto.data.data_format import DataFormat
from azure.kusto.ingest.status import KustoIngestStatusQueues
import pandas
import time
import pprint
cluster = "https://ingest-#.kusto.windows.net/"
database = "MyDatabase"
table = "MyTable"
fields = ["id", "name", "value"]
rows = [[1, "abc", 15.3], [2, "cde", 99.9]]
df = pandas.DataFrame(data=rows, columns=fields)
kcsb = KustoConnectionStringBuilder.with_interactive_login(cluster)
client = QueuedIngestClient(kcsb)
ingestion_properties = IngestionProperties(database=database, table=table, data_format=DataFormat.CSV, report_level=ReportLevel.FailuresAndSuccesses)
client.ingest_from_dataframe(df, ingestion_properties=ingestion_properties)
qs = KustoIngestStatusQueues(client)
MAX_BACKOFF = 180
backoff = 1
while True:
if qs.success.is_empty() and qs.failure.is_empty():
time.sleep(backoff)
backoff = min(backoff * 2, MAX_BACKOFF)
print("No new messages. backing off for {} seconds".format(backoff))
continue
backoff = 1
success_messages = qs.success.pop(10)
failure_messages = qs.failure.pop(10)
pprint.pprint(f"SUCCESS : {success_messages}")
pprint.pprint(f"FAILURE : {failure_messages}")
Результаты выполнения:
No new messages. backing off for 2 seconds
No new messages. backing off for 4 seconds
No new messages. backing off for 8 seconds
No new messages. backing off for 16 seconds
No new messages. backing off for 32 seconds
('SUCCESS : '
'[SuccessMessage({"OperationId":"7cd42ccf-1ed7-4217-b27c-09f9851c0cec","Database":"MyDatabase","Table":"MyTable","SucceededOn":"2022-11-21T20:43:09.3784058Z","IngestionSourceId":"b1bd0c01-2b22-4378-8bcc-0219331be5a4","RootActivityId":"00000000-0000-0000-0000-000000000000","IngestionSourcePath":"https://#.blob.core.windows.net/#-20221121-ingestdata-e5c334ee145d4b4-0/MyDatabase__MyTable__b1bd0c01-2b22-4378-8bcc-0219331be5a4__df_1939259869456_1669063349_4931c16d-bc90-4cd5-b71a-db64e60835d1.csv.gz"})]')
'FAILURE : []'
No new messages. backing off for 2 seconds
No new messages. backing off for 4 seconds
No new messages. backing off for 8 seconds
No new messages. backing off for 16 seconds
No new messages. backing off for 32 seconds
No new messages. backing off for 64 seconds
No new messages. backing off for 128 seconds
No new messages. backing off for 180 seconds
No new messages. backing off for 180 seconds
No new messages. backing off for 180 seconds
No new messages. backing off for 180 seconds
No new messages. backing off for 180 seconds
.
.
.
(1) В этом примере мы делаем только один прием, но теоретически мы могли бы выполнить серию приемов, а затем получить статусы сразу (например, 10 за раз). (2) Конечно. (3) Насколько я знаю, вы должны использовать один и тот же клиент для получения статуса приема.
В коде, на который вы ссылаетесь, есть очень важный комментарий: уровень отчета свойств приема должен быть установлен на ReportLevel.FailuresAndSuccesses
, иначе успехи никогда не будут распознаны.
@actual_panda, да, и это также появляется в моей демонстрации. Проверьте ingestion_properties
Спасибо. Вы знаете, почему
.pop(10)
вызывается с10
в качестве аргумента? Знаете ли вы, будет ли это работать, если несколько клиентов будут получать данные одновременно? Изолированы ли очереди между клиентами или один клиент пропустит сообщения, если я отправлю сообщения другому клиенту?