redash使用与开发(四):执行多句SQL

红太狼 2023-07-04 14:21 21阅读 0赞

在使用了几天后，我逐渐熟悉了使用SQL语言进行复杂的查询，但是受限于redash的查询逻辑，每次都只能执行一句SQL查询，也就是说只能把想获得的字段全部压进一个SELECT语句中，这样写起来没有结构，很乱，所以需要解决的问题是让它可以执行以分号为分隔的多句SQL, 这样就可以方便地创建临时表、设定临时变量。

为此, 需要对负责MySQL查询的函数进行更改，将查询语句按分号分列，同时要注意这其中的一些细节，比如查询语句与非查询语句的一些判断，我是按照最前面的几个字符中是否含有`SELECT`关键字进行判断，为了防止系统自动在查询语句最前面附加的注释造成影响，需要设定`should_annotate_query = False`，该参数位于`redash/redash/query_runner/__init__.py`

关键代码如下

query_list = query.split(';')
    query_length = len(query_list)  # 获取查询的句数，用以判断是否为最后一句
    for query_index in range(0, query_length):
        query_content = query_list[query_index]
        try:
            cursor.execute(query_content)
            data = cursor.fetchall()
            desc = cursor.description
    	    if desc is not None:
    	    	...前面内容基本不变
    	    	break  # 一旦有查询结果，终止查询
    	    else:
    	    	# 查询为空设置结果，但是不返回，继续下一轮执行(如果有的话)
    	    	r.json_data = None
    	        r.error = "没有数据返回..."
    	except:... ...
    	finally:
    		...关闭连接
    		ev.set()

*  注意ev.set()函数的位置，一旦该函数执行，主函数将不再阻塞
 *  当遇到有返回结果的查询(SELECT)或者所有查询执行完毕，当即关闭连接，结束线程，函数的设计应该保证无论什么情况参数r(结果)都有被赋值