测试平台系列(51) 编写数据库连接相关方法.md

2022/6/13 测试平台接口测试FastApiPythonReact

大家好~我是米洛

我在从0到1打造一个开源平台, 也在编写一套完整的接口测试平台系列教程,希望大家能够多多支持。

欢迎关注我的公众号米洛的测开日记,获取最新文章教程!

# 回顾

上一节我们边写了数据库配置的功能接口,但是前端页面还没有做展示。

如果完成了前端页面增删改查页面,我想我们还需要几个功能:

  • 在线测试连接

    因为有的时候数据库配置会发生变化,如果我们不支持在线测试功能的话,可能大家还需要手动去连接一下,非常难用

  • 在线执行SQL

    这个是为了能够给用户在线运行sql语句,类似于小ide的功能,当然我们只做最核心的部分。

    那么今天我们就来完善前端页面并完成在线连接测试功能

# 前端部分

前端部分还是以往的dispatch模式,service.js用于发送http请求,model用于管理变量,组件通过dispatch去调用具体的方法。

前端页面还是分3个部分,可以理解为3个div:

  • 搜索栏
  • 添加栏
  • 表格栏

这是需要填写的表单

大概的页面展示是这样,我们今天重点讲一下数据库连接这块。

# 思考怎么做

连接部分我们仍然打算采用sqlalchemy,既然已经开始用了,而且他也支持异步session,在没遇到大坑的时候我们就不考虑更换了。

首先我们考虑以下几个点:

# 连接复用

首先要清楚,我们之所以做数据库连接配置,是因为我们在构造数据的时候会用到,那么我们势必要维护一个数据库连接的map,充当缓存的作用。

当我们的账号密码等连接信息没有变化的时候,我们是不是可以把上次连接用的session拿出来继续使用?答案是肯定的!

而且当我们测试连接以后,这个数据库配置就被缓存了起来,如果我们要使用,直接从连接缓存里面拿就可以了。配合sqlalchemy本身自带的连接池,基本上我们要做的就比较简单,只是包一层dict即可。

# 缓存扩张问题

举个例子,我现在配置了A数据库连接,测试了一遍,你给我缓存起来了,接着我把A改成B,由于A和B的配置不一样,所以缓存里面现在需要插入一条B数据,如果改的频繁,就会使得这个dict无限扩张,内存也会占用变大。但因为数据库配置不太可能经常变化,我们也可以在update配置的时候清楚之前的缓存。

# 多线程访问缓存的问题

Python的多线程我不是太熟悉,不太确定Fastapi内部是否会有多个线程同时读写缓存的情况,如果有,那肯定会出问题的,我们需要加🔒,暂时我们可以先观望一下

# 动起手来

我们对之前的models/init.py方法进行一些改造,我们需要增加一个DatabaseHelper类。

为什么要传入这么多参数,是因为传入配置id的话,需要再次查询数据库,比较麻烦

  • init方法

    init方法创建了我们内部的dict缓存,key是数据库的连接信息,value是对应的session。

  • get_connection

    通过数据库配置去获取到对应的连接信息,首先拼接key,拿到唯一的key标识,接着去缓存找有没有初始化这个连接,有的话直接返回,没有就开始建立连接。

  • get_jdbc_url

这里其实是把数据库配置转为了数据库连接url

注意: pg我还没有找到合适的异步库,所以暂时先不支持它,后面再支持,总感觉pg用的人并没有那么多。

  • 删除缓存的方法

先判断是否有该key,有则删除

  • 测试连接的方法

接受一个session,如果为空,则返回错误信息。最后execute一句select 1测试是否能够正常连上数据库,不能则返回错误信息。

最后我们再实例化这个Helper,统一调用这个对象即可

  • 改写update方法

加入删除缓存的部分

  • 编写在线测试方法

如果session没获取到,返回错误信息

# 测试一哈

密码不正确的时候

这是密码正确的情况下

今天的内容到这里就结束了,下一节我们编写在线执行SQL功能。