助力工业物联网，工业大数据之服务域：Shell调度测试【三十三】

本文链接：https://blog.csdn.net/2401_84160361/article/details/138360853

	```
	- 开发
	
	 
	```
	# import
	from datetime import timedelta
	from airflow import DAG
	from airflow.operators.bash import BashOperator
	from airflow.utils.dates import days_ago
	
	# define args
	default_args = {
	    'owner': 'airflow',
	    'email': ['airflow@example.com'],
	    'email\_on\_failure': True,
	    'email\_on\_retry': True,
	    'retries': 1,
	    'retry\_delay': timedelta(minutes=1),
	}
	
	# define dag
	dag = DAG(
	    'second\_airflow\_dag',
	    default_args=default_args,
	    description='first airflow task DAG',
	    schedule_interval=timedelta(days=1),
	    start_date=days_ago(1),
	    tags=['itcast\_bash'],
	)
	
	# define task1
	say_hello_task = BashOperator(
	    task_id='say\_hello\_task',
	    bash_command='echo "start task"',
	    dag=dag,
	)
	
	# define task2
	print_date_format_task2 = BashOperator(
	    task_id='print\_date\_format\_task2',
	    bash_command='date +"%F %T"',
	    dag=dag,
	)
	
	# define task3
	print_date_format_task3 = BashOperator(
	    task_id='print\_date\_format\_task3',
	    bash_command='date +"%F %T"',
	    dag=dag,
	)
	
	# define task4
	end_task4 = BashOperator(
	    task_id='end\_task',
	    bash_command='echo "end task"',
	    dag=dag,
	)
	
	say_hello_task >> [print_date_format_task2,print_date_format_task3] >> end_task4
	
	```
+ **提交**

 
```
python second_bash_operator.py 

```
+ **查看**

 ![image-20211005131800085](https://img-blog.csdnimg.cn/img_convert/9fabfc124be52e0e364d495bddb90df4.png)

小结
- 实现AirFlow的依赖调度测试

知识点09：Python调度测试

目标：实现Python代码的调度测试

实施

需求：调度Python代码Task的运行

代码

创建

cd /root/airflow/dags
vim python_etl_airflow.py

开发

# import package
from airflow import DAG
from airflow.operators.python import PythonOperator
from airflow.utils.dates import days_ago
import json

# define args
default_args = {
    'owner': 'airflow',
}

# define the dag
with DAG(
    'python\_etl\_dag',
    default_args=default_args,
    description='DATA ETL DAG',
    schedule_interval=None,
    start_date=days_ago(2),
    tags=['itcast'],
) as dag:
    # function1
    def extract(\*\*kwargs):
        ti = kwargs['ti']
        data_string = '{"1001": 301.27, "1002": 433.21, "1003": 502.22, "1004": 606.65, "1005": 777.03}'
        ti.xcom_push('order\_data', data_string)
        
    # function2
    def transform(\*\*kwargs):
        ti = kwargs['ti']
        extract_data_string = ti.xcom_pull(task_ids='extract', key='order\_data')
        order_data = json.loads(extract_data_string)
        total_order_value = 0
        for value in order_data.values():
            total_order_value += value
        total_value = {"total\_order\_value": total_order_value}
        total_value_json_string = json.dumps(total_value)
        ti.xcom_push('total\_order\_value', total_value_json_string)
        
    # function3
    def load(\*\*kwargs):
        ti = kwargs['ti']
        total_value_string = ti.xcom_pull(task_ids='transform', key='total\_order\_value')
        total_order_value = json.loads(total_value_string)
        print(total_order_value)
        
    # task1
    extract_task = PythonOperator(
        task_id='extract',
        python_callable=extract,
    )
    extract_task.doc_md = """\
#### Extract task
A simple Extract task to get data ready for the rest of the data pipeline.
In this case, getting data is simulated by reading from a hardcoded JSON string.
This data is then put into xcom, so that it can be processed by the next task.
"""
	# task2
    transform_task = PythonOperator(
        task_id='transform',
        python_callable=transform,
    )
    transform_task.doc_md = """\
#### Transform task
A simple Transform task which takes in the collection of order data from xcom
and computes the total order value.
This computed value is then put into xcom, so that it can be processed by the next task.
"""
	# task3
    load_task = PythonOperator(
        task_id='load',
        python_callable=load,
    )
    load_task.doc_md = """\
#### Load task
A simple Load task which takes in the result of the Transform task, by reading it
from xcom and instead of saving it to end user review, just prints it out.
"""

# run
extract_task >> transform_task >> load_task

提交

python python_etl_airflow.py

查看

小结
- 实现Python代码的调度测试

知识点10：Oracle与MySQL调度方法

目标：了解Oracle与MySQL的调度方法

实施

Oracle调度：参考《oracle任务调度详细操作文档.md》

step1：本地安装Oracle客户端
step2：安装AirFlow集成Oracle库
step3：创建Oracle连接
step4：开发测试

query_oracle_task = OracleOperator(
    task_id = 'oracle\_operator\_task',
    sql = 'select \* from ciss4.ciss\_base\_areas',
    oracle_conn_id = 'oracle-airflow-connection',
    autocommit = True,
    dag=dag
)

MySQL调度：《MySQL任务调度详细操作文档.md》

step1：本地安装MySQL客户端
step2：安装AirFlow集成MySQL库
step3：创建MySQL连接

step4：开发测试

方式一：指定SQL语句

query_table_mysql_task = MySqlOperator(
    task_id='query\_table\_mysql', 
    mysql_conn_id='mysql\_airflow\_connection', 
    sql=r"""select \* from test.test\_airflow\_mysql\_task;""",
    dag=dag
)

  + 方式二：指定SQL文件
  
   
  ```
  query_table_mysql_task = MySqlOperator(
      task_id='query\_table\_mysql\_second', 
      mysql_conn_id='mysql-airflow-connection', 
      sql='test\_airflow\_mysql\_task.sql',
      dag=dag
  )
  
  ```

方式三：指定变量

insert_sql = r"""
INSERT INTO `test`.`test\_airflow\_mysql\_task`(`task\_name`) VALUES ( 'test airflow mysql task3');
INSERT INTO `test`.`test\_airflow\_mysql\_task`(`task\_name`) VALUES ( 'test airflow mysql task4');
INSERT INTO `test`.`test\_airflow\_mysql\_task`(`task\_name`) VALUES ( 'test airflow mysql task5');
"""

insert_table_mysql_task = MySqlOperator(
    task_id='mysql\_operator\_insert\_task', 
    mysql_conn_id='mysql-airflow-connection', 
    sql=insert_sql,
    dag=dag
)