修改gorm支持protobuf

Posted 2021-01-09 mixerzhan

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。