Airflow / minio:如何将minio用作从Airflow发送的数据的本地S3代理?

时间:2019-04-05 06:26:52

标签: amazon-s3 google-cloud-storage airflow minio

一个简单的问题:

我不想使用S3或GCS,而是想知道如何将minio用作本地S3代理来保存发送气流的数据。我该怎么做呢?我可以使用FileToGoogleCloudStorageOperator吗?

如果不是这种用于本地存储(大图像而不是数据库行)的路由,您会推荐什么?

谢谢!

1 个答案:

答案 0 :(得分:1)

similar answer为基础,这是我在撰写本文时使用最新版本的Airflow(1.10.7)的目的:

首先,使用以下信息创建S3连接:

Connection Name: '<your connection name>' #  e.g. local_minio
Connection Type: S3
Extra: a JSON object with the following properties: 
 {
    "aws_access_key_id":"your_minio_access_key",
    "aws_secret_access_key": "your_minio_secret_key",
    "host": "http://127.0.0.1:9000"
 }

接下来,在DAG中,使用S3Hook与数据进行交互来创建任务。这是一个可以满足您的需求的示例:

from datetime import datetime, timedelta

from airflow import DAG
from airflow.operators.python_operator import PythonOperator
from airflow.hooks.S3_hook import S3Hook

DEFAULT_ARGS = {
    'owner': 'Airflow',
    'depends_on_past': False,
    'start_date': datetime(2020, 1, 13),
    'email': ['airflow@example.com'],
    'email_on_failure': False,
    'email_on_retry': False,
    'retries': 1,
    'retry_delay': timedelta(minutes=5),
}

dag = DAG('create_date_dimension', default_args=DEFAULT_ARGS,
          schedule_interval="@once")


def write_text_file(ds, **kwargs):
    with open("/tmp/test.txt", "w") as fp:
        # Add file generation/processing step here, E.g.:
        fp.write(ds)

        # Upload generated file to Minio
        s3 = S3Hook('local_minio')
        s3.load_file("/tmp/test.txt",
                     key=f"my-test-file.txt",
                     bucket_name="my-bucket")


# Create a task to call your processing function
t1 = PythonOperator(
    task_id='generate_and_upload_to_s3',
    provide_context=True,
    python_callable=write_text_file,
    dag=dag
)