Starburst Presto SQL中如何实现用户输入提示功能

Starburst Presto（基于Trino内核的商业发行版）内核本身没有内置面向终端用户的输入提示模块，所有提示能力都可以根据你使用的交互场景，在不改动核心执行逻辑的前提下实现：

场景1：使用官方CLI客户端交互 #

• 轻量实现不需要改源码：直接写Shell包装脚本启动CLI，在脚本里提前做交互提示，收集用户输入的必要参数（比如分区值、查询队列、执行超时配置等），校验通过后再把参数拼接到SQL会话配置或者查询语句里传给CLI执行，简单实现示例：

#!/bin/bash
read -p "请输入本次查询必须指定的dt分区值（格式yyyy-mm-dd）: " input_dt
if [[ ! $input_dt =~ ^[0-9]{4}-[0-9]{2}-[0-9]{2}$ ]]; then
  echo "输入的分区格式不符合要求，请重新输入"
  exit 1
fi
# 传入参数启动CLI
presto --session catalog=hive --schema=dwd --execute "
SET SESSION query_max_run_time='20m';
SELECT * FROM user_behavior WHERE dt='${input_dt}' LIMIT 1000;
"

• 如果要做输入过程中的实时联想提示，需要修改CLI的源码：CLI基于JLine做交互输入，你可以找到对应版本CLI模块下的自动补全逻辑类，把你需要的提示规则（比如自定义函数提示、分区字段必填提示、高危操作拦截提示）加到补全候选集里，编译替换原有CLI的jar包即可生效。

场景2：对接自建查询平台/BI工具 #

• 最通用的方案是在前端查询编辑器层实现提示：提前通过Presto的系统表拉取全量元数据（表名、字段名、内置函数、自定义UDF、常用SQL模板），给编辑器绑定输入事件监听，用户输入时匹配内容前缀弹出对应提示即可，不需要对Presto服务端做任何改造，绝大多数内部数据平台都是用这个方案实现。
• 如果要做服务端强制提示（比如检测到用户写了不带分区的查询、扫描数据量超阈值的查询时弹出提示拦截），可以实现Presto的事件监听器扩展点：在查询提交创建的触发节点，用Presto自带的SQL解析器解析语句逻辑，检测到不符合规范的查询直接返回带提示内容的错误，提示文本会直接展示在用户的客户端界面，比如返回查询dwd层表必须携带dt分区过滤条件，请补充后重试，就可以实现强制的输入规范提示。

注意：不要尝试修改Presto SQL执行内核实现输入提示，输入提示属于交互层能力，放在客户端、前端或者查询接入校验层实现的维护成本远低于改内核。

内容的提问来源于stack exchange，提问作者SillyStraw1022