基于Trae开发的自动表关联查询工具

QdcCNuwxWOXo

对于复杂的一些业务，会涉及很多张表，其间有各种各样的关联关系，在开发&测试过程中，随时需要查看这些表中的数据状态，这种情况下需要我们写一些关联查询的SQL或者多条SQL执行来查看结果，个人感觉用起来还是不太方便，所以想开发一个关联表自动进行查询的工具
后端基于spring boot自行开发；前端部分使用了Trae，让大模型来进行的开发～
后端项目代码、 前端项目代码、 体验地址
服务主要还是本地部署个人使用，为了简化，本地数据库使用了sqlite的文件形式存储，同时采用了将前端项目打包结果复制到后端项目中同步部署，所以没有其他任务外部依赖
在后端项目中启动后，直接访问 http://localhost:8080 即可使用，或者打包后通过 java -jar xxx.jar 执行
注意：

• 需要 jdk 版本大于等于 21
  
• 只支持MySQL数据库
  
• 没有做鉴权及SQL注入等处理，所以只建议本地安装个人使用
  

下面进行一下对应功能介绍
数据源管理

用来管理数据库（MySQL）的连接（需要服务所在的机器可以连接到数据库）

表字段关系管理

这里主要进行字段关系的维护，如果使用了外键的话，会自动进行同步，支持跨库关联

比如员工表(employees)的部门编码字段(dept_no)对应了，部门表(departments)的(dept_no)字段，那么我们可以这样配置（主表可以理解为包含类似外键的表）
其中的关联条件部分，使用场景为一张表根据条件关联不同的表
假设有多张表，它们的操作日志都记录在了一个叫做操作日志的表里面，同时操作日志表有一个类型字段来区分对应的表是哪个，那么我们可以在关联条件中输入对应的条件即可，如（type=1），如果没有这种场景就可以不填写

ER图查看

在创建好数据源以及维护好字段关系后，我们可以通过查看ER图来确认一下配置是否正确，并且也可以让新人快速熟悉表间关系
需要输入一下对应的db和table即可，会查找所有关联的表进行展示，并同时展示关联的字段关系，支持跨库关联

数据查询

最后就是数据的查询，选择db和表后，需要输入对应的查询条件，这时会查询对应的数据，同时会将关联的表和数据同时在下方进行展示（目前限制了单表数据最多10条）
目前是会根据选择的表向外查询关联表，是广度优先，并且对于相同的表只会查询一次，所以选择的表不同，结果可能会有所差异

后续扩展

在有个表关系的ER图后，我们可以比较容易的让大模型来帮助我们根据自然语言来生成查询SQL了
将memaid格式的ER图文本和自然语言提供给大模型即可，这里举一个小例子(langchain)：
class ExecutableSql(BaseModel):    """可执行的SQL信息"""    sql: str = Field(description="查询结果的SQL语句")def sql_generation(er_diagram: str, description: str, llm: BaseChatModel) -> str:    """    根据用户的请求，和er图，生成最终的SQL语句    """    # 构造解析器    parser = PydanticOutputParser(pydantic_object=ExecutableSql)    # 构造提示词模版    prompt = PromptTemplate(        template="根据如何mermaid格式的ER图:\n{er_diagram}\n\n为这个请求生成SQL查询语句:\n{description}\n\n，{format_instructions}\n",        input_variables=["description", "er_diagram"],        partial_variables={"format_instructions": parser.get_format_instructions()},    )    # 将提示此、llm、结果解析器构造成链    chain = prompt | llm | parser    # 使用参数实际调用获取结果    sqlInfo = chain.invoke({"er_diagram": er_diagram, "description": description})    return sqlInfo.sql这里使用了mysql 例子中的employees数据库来实际看下，生成的ER图如下：

将左侧ER图的文本和问题，如：查询一下员工编号为10002的基本信息、所在部门，以及不同时间对应的薪资情况提供给大模型，可以得到如下结果：
SELECT     e.emp_no,    e.birth_date,    e.first_name,    e.last_name,    e.gender,    e.hire_date,    d.dept_name,    s.salary,    s.from_date AS salary_from_date,    s.to_date AS salary_to_dateFROM employees eJOIN dept_emp de ON e.emp_no = de.emp_noJOIN departments d ON de.dept_no = d.dept_noJOIN salaries s ON e.emp_no = s.emp_noWHERE e.emp_no = 10002ORDER BY s.from_date;我们实际执行一下，获取结果：