Skip to content

125.兼容postgresql

大家好~我是米洛
我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的教程,希望大家多多支持。
欢迎关注我的公众号米洛的测开日记,一起交流学习!

回顾

上一节我们给自己代码做了一次review,在review的过程中我注意到我们的数据库还没有完全支持postgresql,所以今天我们就来试试看,切换底层数据库,pity能不能顶得住。

定义枚举

如我所说,对于具体的sql_type,我们应该用枚举类给定义起来,所以我们来定义一下枚举:

接着我们去替换对应的地方,也就是上一节所说的magic number:

这里如果不满足我们的条件,则直接抛异常,提示用户没有这个数据库类型。

安装pg

pg官网

进入下载页,下载对应操作系统的pg即可。傻瓜式安装后,如果没有启动,则需要在安装目录进行如下操作:

// 在bin目录
// 数据路径是pg在磁盘的存储数据路径

pg_ctl -D 数据存储路径 start

启动之后我们可以用DataGrip连接数据库,然后创建pity数据库。这一步类似于mysql的

create database pity;

一切都是傻瓜式的,嘿嘿。

下载异步pg引擎

经过sqlalchemy官网的指示,我们下载asyncpg引擎:

pip install asyncpg

修改jdbc

因为我们是临时配置测试一下,所以直接写死了,这些都是我本地的数据库账号密码和端口等数据。

本次测试并不是要切换到pg,而是给只能选择pg(比如我上家公司就没有mysql)的人一个选择,顺便也看看pity能不能无缝切换到pg。

重新建表

创建好数据库以后,我们第一步就是要测试数据表是否能够建立。我们启动一下pity试试。

一上来就给咱一个猛的报错,可以看到,我们的ddl里面包含的是datetime,但pg里面并没有datetime,确实,我之前使用的时候也都用的timestamp

所以这意味着,如果我们要兼容pg,则需要把datetime类型全部换成TIMESTAMP,换是可以换,但我就怕我们现在的代码会受影响。

这里我已经批量替换过了,发现sql里面反射出来的依旧是datetime类型(Python的datetime),那就说明代码没啥问题。

pg也没有tinytext和longtext,所以我们需要统一改成text兼容pg。

总体来说改动并不大,这就是我们用sqlalchemy(orm)的好处。


来看看改造后,是否好用:

可以看到,数据表基本都建好了,但是少了一只:apscheduler。忘了这个硬茬了!!!

适配apscheduler

总体来说,由于apscheduler的存在,我们还是得弄一套同步session,至于pg的同步session,我们需要安装psycopg2:

pip install psycopg2

再次启动服务,发现表格建好了!

试试pity吧

问题一: deleted_at

出师不太利,一上来就报了个deleted_at的错,原来我有的deleted_at的字段定义的是timestamp,但实际上我查询的时候用的是否等于0。

所以我统一把deleted_at改成bigint了。

问题二: date_format

接着又遇到了一个date_format的错,我们试着解决下吧:

对于mysql这句话是没毛病的,格式化日期,但是pg不太支持呀,所以我们得换通用点的方式。

这里我们就不采用sql的func了,直接用Python的方法,这样都兼容,按照图里这么改。改好了之后mysql继续测试一下,发现没毛病。

问题三: timestamp不支持直接比较字符串

我们之前比较日期,用between或者大于小于的时候,传入字符串的时间格式都莫得问题,但是换pg之后她就傲娇了。原因是,它不能把text直接转成timestamp,所以我们得手动转一下,MySQL倒是2种都支持。

问题四: pg的数据传入id字段,哪怕为None insert的时候会把id置为0

为了解决这个问题,我们就不赋予id属性了:

也许跟用的异步pg引擎有关,我试了下这个也不影响MySQL,所以就迁就下吧。


以上就是我总结的,数据库切换引发的一系列的问题,希望对大家有所帮助

我是米洛,一直陪伴各位学习!免费的小黄心,帮我点一个吧!

项目地址: https://github.com/wuranxu/pity