当前位置 : 主页 > 大数据 > 区块链 >

修改gorm支持protobuf

来源:互联网 收集:自由互联 发布时间:2021-06-22
gorm的功能很强大,支持很多很多特性,打算在项目中用上它。 但gorm不支持protobuf,如果idl用的是protobuf,需要对每个message做一个重新定义一个内部的struct,使得可以让这个内部struct和

gorm的功能很强大,支持很多很多特性,打算在项目中用上它。

但gorm不支持protobuf,如果idl用的是protobuf,需要对每个message做一个重新定义一个内部的struct,使得可以让这个内部struct和mysql一一对应,这个是个很无脑的搬砖活;而且,对于message里面有内嵌message的情况的,原生的gorm只能做表关联,这对某些表需要比较多的内嵌数据,比如repeated数据,会使得表的记录数出现膨胀增长。因此,让gorm支持protobuf的内嵌字段,而且不需要写中间的struct模型,是非常有必要的。

我主要从几个文件上修改:

1、model_struct.go,这个文件主要是解析struct字段到mysql字段的,对于pb生成的go文件,每个message默认都会生成类似XXX_格式的几个字段,这几个跟业务无关,不需要db,因此直接忽略掉,在GetModelStruct中修改。

2、callback_create.go:这个文件是插入数据的回调,在这里主要是要对pb中的内嵌字段进行序列化,在createCallback中,简单判断字段是ptr就进行序列化。这里有个问题要需要验证:mysql执行插入操作,如果字段是二进制的,可以直接insert into tables(..) values(..)吗

3、dialect_mysql.go:mysql的字段定义,主要改判断字段是ptr的时候,就定义为longblob类型

4、scope.go:从数据库中检索数据用,需要对内嵌字段反序列化,这里需要先了解下sql.rows.scan的操作作用,scan之后需要对字段重新赋值,对于内嵌字段,需要反序列化,在scan中改。这个文件的另一个改动是创建表结构,在createTable中,只对isNormal,也就是普通字段进行创建,其他字段不创建,这里改下。

经过改造后,基本可以支持protobuf了,今天来不及上代码了,改天整理下,看看能否作为一个插件提交给gorm。

网友评论