AirFlow 中使用的 HiveServer2Hook 底层使用的 pyhive，所以先来看下 pyhive 的使用：

1. PyHive连接Hive

  1. from pyhive import hive
  3. conn = hive.connect(host='1.1.1.1',
  4.                     port=2000,
  5.                     auth="LDAP",
  6.                     database='test',
  7.                     username='yumingmin',
  8.                     password='yumingmin')
  9. cursor = conn.cursor()
  10. cursor.execute('select * from test.students limit 10')

2. 使用HiveServer2Hook

首先常见一个 connection，注意 authMechanism 不可以使用 PLAIN，这个和 impyla 是不一样的。

  1. $ airflow connections add 'hive_db' --conn-uri 'hiveserver2://yumingmin:yumingmin@1.1.1.1:20000/test?authMechanism=LDAP'

执行代码：

  1. from airflow.providers.apache.hive.hooks.hive import HiveServer2Hook
  3. hh = HiveServer2Hook(hiveserver2_conn_id="impala_db")
  4. sql = "SELECT * FROM cszc.sop_phone_recom_num_m2small_aftn LIMIT 1"
  5. print(len(hh.get_records(sql)))