!pip install jaydebeapi
import jaydebeapi
from airflow.models import DAG
from airflow.operators.python_operator import PythonOperator
from datetime import datetime
def db2_query():
conn = jaydebeapi.connect('com.ibm.db2.jcc.DB2Driver',
'jdbc:db2://db2_server:port/DBNAME',
{'user': 'USERNAME', 'password': 'PASSWORD'})
c = conn.cursor()
c.execute('SELECT * FROM TABLE_NAME')
rows = c.fetchall()
for row in rows:
print(row)
conn.close()
dag = DAG('db2_query', description='Query DB2', schedule_interval='0 0 * * *', start_date=datetime(2021, 9, 1))
db2_task = PythonOperator(task_id='db2_task', python_callable=db2_query, dag=dag)
airflow trigger_dag db2_query
如果没有错误,你应该能看到来自DB2数据库的输出。