CN114968214B

CN114968214B - 代码自动生成方法、装置、存储介质及终端设备

Info

Publication number: CN114968214B
Application number: CN202210811775.1A
Authority: CN
Inventors: 王俊月; 刘鑫; 李盟
Original assignee: Accumulus Technologies Tianjin Co Ltd
Current assignee: Accumulus Technologies Tianjin Co Ltd
Priority date: 2022-07-12
Filing date: 2022-07-12
Publication date: 2022-10-28
Anticipated expiration: 2042-07-12
Also published as: CN114968214A

Abstract

本发明公开了一种代码自动生成方法、装置、存储介质及终端设备，方法包括如下步骤：预定义基本数据类型映射关系；配置需要生成映射关系的表的信息及特殊场景的自定义数据类型；读取需要生成映射关系的表的信息及特殊场景的自定义数据类型保存；获取数据库读取权限；查询配置指定的需要生成映射关系的数据表的表结构信息；获取扩展的表和字段及字段映射后的自定义数据类型；迭代表结构信息，判断是否和扩展字段匹配，直到所有表结构信息完成映射；使用预置的代码模版生成业务逻辑代码。本发明可以在保留数据库与go语言基本数据类型映射关系的基础上，通过配置信息更为便捷的为特殊场景需求相应修改为自定义数据类型的映射关系。

Description

代码自动生成方法、装置、存储介质及终端设备

技术领域

本发明涉及计算机技术技术领域，尤其涉及一种代码自动生成方法、装置、存储介质及终端设备。

背景技术

服务端处理业务逻辑通常依赖于对数据库表的增删改查操作，其中mysql数据库的基本数据类型有bigint、float、varchar、text、datetime，分别对应于go语言中的基本数据类型int64、float、string、string、time.Time。使用ORM技术自动生成代码时，可以简便的构造数据库与go语言数据类型间的映射关系，在业务代码中调用sql语句查询数据库表时可以直接转换到对应的go语言数据类型。

但是业务实现及数据库存储有特殊场景的需求，如淘宝登录需要用户提供手机号及验证码，手机号作为用户隐私数据，不能直接存储在数据库中，会造成数据的泄漏，有一定的安全隐患。此时，需要将手机号进行掩码处理后再存储，业务读取时进行掩码解析得到真实手机号，这样手机号不会直接暴露出来。

在这种场景下，掩码处理后的手机号在数据库中仍使用varchar数据类型进行存储，业务中直接读取到的数据类型仍为string时，需要业务代码单独实现进行掩码解析。如果将数据库的varchar数据类型直接转换到go语言已自定义的数据类型中，该自定义数据类型已封装解析代码，使用起来将会更加安全便捷，业务代码与解析代码也可以解耦合。

传统的使用ORM技术自动生成代码的方法中，仅构造了基本数据类型的映射，未实现自定义数据类型的映射关系。

发明内容

本发明的目的在于提供一种代码自动生成方法、装置、存储介质及终端设备，可以在保留数据库与go语言基本数据类型映射关系的基础上，通过配置信息更为便捷的为特殊场景需求相应修改为自定义数据类型的映射关系。

为实现本发明的目的，本发明提供的技术方案具体如下：

第一方面

本发明提供了一种代码自动生成方法，所述方法包括如下步骤：

接收代码生成请求指令；

执行如下步骤：

步骤a.预定义基本数据类型映射关系；

步骤b.配置需要生成映射关系的表的信息及特殊场景的自定义数据类型，包括数据库连接地址、数据库名、需要生成映射关系的表名、需要扩展的表和字段，及字段映射后的自定义数据类型；

步骤c.读取需要生成映射关系的表的信息及特殊场景的自定义数据类型保存至内存中；

步骤d.根据数据库连接地址及数据库名，连接到指定的数据库，获取数据库读取权限；

步骤e.查询配置指定的需要生成映射关系的数据表的表结构信息；

步骤f.获取扩展的表和字段，及字段映射后的自定义数据类型，及需要引用的外部包链接；

步骤g.迭代步骤e查询到的表结构信息，判断是否和扩展字段匹配，若不匹配，则将表结构映射为基本数据类型；若匹配，则将表结构映射为自定义的数据类型，并保存外部包链接；直到所有表结构信息完成映射；

步骤h.使用预置的代码模版生成业务逻辑代码。

其中，所述方法包括如下步骤：所述步骤a具体为预定义数据库基本数据类型与go语言基本数据类型映射关系。

其中，在步骤e中，利用sql语句查询。

第二方面

本发明提供了一种代码自动生成装置，包括代码生成请求指令接收模块、基本数据类型映射关系模块、配置信息模块、信息保存模块、数据库读取模块、表结构信息查询模块以及配置信息获取模块、自定义的数据类型映射模块、代码生成模块；

其中，所述代码生成请求指令接收模块用于接收代码生成请求指令；

所述基本数据类型映射关系模块用于预定义基本数据类型映射关系；

所述配置信息模块用于配置需要生成映射关系的表的信息及特殊场景的自定义数据类型，包括数据库连接地址、数据库名、需要生成映射关系的表名、需要扩展的表和字段，及字段映射后的自定义数据类型；

所述信息保存模块用于读取需要生成映射关系的表的信息及特殊场景的自定义数据类型保存至内存中；

所述数据库读取模块用于根据数据库连接地址及数据库名，连接到指定的数据库，获取数据库读取权限；

所述表结构信息查询模块用于查询配置指定的需要生成映射关系的数据表的表结构信息；

所述配置信息获取模块用于获取扩展的表和字段，及字段映射后的自定义数据类型，及需要引用的外部包链接；

所述自定义的数据类型映射模块用于迭代查询到的表结构信息，判断是否和扩展字段匹配，若不匹配，则将表结构映射为基本数据类型；若匹配，则将表结构映射为自定义的数据类型，并保存外部包链接；直到所有表结构信息完成映射；

所述代码生成模块用于使用预置的代码模版生成业务逻辑代码。

其中，所述基本数据类型映射关系模块具体用于预定义数据库基本数据类型与go语言基本数据类型映射关系。

其中，所述表结构信息查询模块具体利用sql语句查询。

第三方面

本发明提供了一种存储介质，所述存储介质中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由处理器加载并执行以实现所述的一种基于ORM的自定义数据类型的代码自动生成方法。

第四方面

本发明提供了一种终端设备，所述终端设备包括处理器和存储器，所述存储器中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由所述处理器加载并执行以实现所述的一种基于ORM的自定义数据类型的代码自动生成方法。

与现有技术相比，本发明的有益效果为，本发明使用配置信息指定数据类型以适配特殊需求场景，可以在保留数据库与go语言基本数据类型映射关系的基础上，通过配置信息更为便捷的为特殊场景需求相应修改为自定义数据类型的映射关系。

附图说明

图1为本申请实施例的方法流程示意图。

具体实施方式

以下结合附图和具体实施例对本发明作进一步详细说明。应当理解，此处所描述的具体实施例仅仅用以解释本发明，并不用于限定本发明。

如图1所示，为本发明实施例提供的一种代码自动生成方法，是一种基于ORM的自定义数据类型的代码自动生成方法，所述方法包括如下步骤：

接收代码生成请求指令；

执行如下步骤：

步骤a.预定义基本数据类型映射关系；具体地，预定义数据库基本数据类型与go语言基本数据类型映射关系；

具体实现如下：

var dataType2goType=map[string]string{

“bigint”: “int64”,

“tinyint”: “int32”,

“varchar”: “string”,

“ text”: “string”,

“int”: “int32”,

“smallint”: “int32”,

“datetime”: “time.Time”,

“date”: “time.Time”,

}

具体实现如下：

connection：“ root:*/ormexample”

database: “ormexample” # 数据库

includetables：#需要生成映射关系的表名

-“example”

exttables: # 需要扩展的表

-name: “example”

columns: # 需要扩展的字段

-name: “I_CUSTOM_FILED”

Customtype: “code.*.com/infra/yorm/example/custom,custom.CustomInt”

-name: “CH_TXET”

Customtype: “gopkg.in/guregu/null.V4,null.String”

步骤e.利用sql语句查询配置指定的需要生成映射关系的数据表的表结构信息；

数据表的表结构如下：

CREATE TABLE ‘example’(

‘I_ID’bigint(20) unsigned NOT NULL AUTO_INCREMENT COMMENT ‘自增主键’，

‘I_USER_ID’bigint(20) NOT NULL DEFAULT ‘0’COMMTET ‘USERID’,

‘CH_NAME’varchar(10) NOT NULL DEFAULT ‘’COMMTET ‘名称’,

‘CH_TEXT’text COMMENT ‘txt’,

‘I_CUSTOM_FIELD’ bigint(20) NOT NULL DEFAULT ‘0’COMMTET ‘自定义类型字段’,

‘D_CREATED_AT’ datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMTET‘创建时间’,

‘D_UPDATED_AT’ datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATECURRENT_TIMESTAMP COMMTET ‘更新时间’,

PRIMARY KEY (‘I_ID’)

)ENGINT=InnoDB DEFAULT CHARSET=utf8mb4 COMMTET ‘example’

步骤g.迭代步骤e查询到的表结构信息，判断是否和扩展字段匹配，若不匹配，则将表结构映射为基本数据类型；若匹配，则将表结构映射为自定义的数据类型，并保存外部包链接；

步骤h.直到所有表结构信息完成映射；

其中，无自定义数据类型的映射结果如下：

import (

“time”

)

//example

Type example struct{

ID int32 ‘db: “I_ID”’ //自增主键

USERID int32 ‘db: “I_ USER_ID”’ // USERID

NAME string ‘db: “CH_NAME”’ //名称

TEXT string ‘db: “CH_TEXT”’ //txt

CUSTOMFIELD int64 ‘db: “I_CUSTOM_FIELD”’ //自定义类型字段

CREATEDAt time.Time ‘db: “D_CREATED_AT”’ //创建时间

UPDATEDAt time.Time ‘db: “D_ UPDATED_AT”’ //更新时间

}

其中，有自定义数据类型的映射结果如下：

import (

“time”

“code.*.com/infra/yorm/example/customt”

)

//example

Type example struct{

ID int32 ‘db: “I_ID”’ //自增主键

USERID int32 ‘db: “I_ USER_ID”’ // USERID

NAME string ‘db: “CH_NAME”’ //名称

TEXT null.string ‘db: “CH_TEXT”’ //txt

CUSTOMFIELD custom.customInt ‘db: “I_CUSTOM_FIELD”’ //自定义类型字段

CREATEDAt time.Time ‘db: “D_CREATED_AT”’ //创建时间

UPDATEDAt time.Time ‘db: “D_ UPDATED_AT”’ //更新时间

}

步骤i.使用预置的代码模版生成业务逻辑代码。

与上述方法相对应地，本发明实施例还提供了一种代码自动生成装置，包括代码生成请求指令接收模块、基本数据类型映射关系模块、配置信息模块、信息保存模块、数据库读取模块、表结构信息查询模块以及配置信息获取模块、自定义的数据类型映射模块、代码生成模块；

所述表结构信息查询模块用于查询配置指定需要生成映射关系的数据表的表结构信息；

其中，所述表结构信息查询模块具体利用sql语句查询。

需说明的是，本发明实施例的装置与上述的方法的具体实施细节及效果相同或相似，在此不再赘述。

本方明另一实施例还提供一种终端设备，所述终端设备包括处理器和存储器，所述存储器中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由所述处理器加载并执行以实现上述任一项代码自动生成方法。

需要说明的是终端设备为计算机设备。

本发明另一实施例还提供一种存储介质，所述存储介质中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由处理器加载并执行以实现上述任一项代码自动生成方法。

对于计算机设备及存储介质中的代码自动生成方法可参考上述实施例进行理解，在此不再赘述。

下面主要结合应用场景对计算机设备及存储介质进行进一步介绍。

本领域内的技术人员应明白，本发明实施例可提供为方法、装置(设备或系统)、或计算机设备、存储介质。因此，本发明实施例可采用完全硬件实施例、完全软件实施例、或结合软件和硬件方面的实施例的形式。而且，本发明实施例可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品的形式。

本发明是参照根据本发明实施例的方法、装置(设备或系统)、和计算机程序产品的流程图和/或方框图来描述的。应理解可由计算机程序指令实现流程图和/或方框图中的每一流程和/或方框、以及流程图和/或方框图中的流程和/或方框的结合。可提供这些计算机程序指令到通用计算机、专用计算机、嵌入式处理机或其他可编程数据处理设备的处理器以产生一个机器，使得通过计算机或其他可编程数据处理设备的处理器执行的指令产生用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的装置。

这些计算机程序指令也可存储在能引导计算机或其他可编程数据处理设备以特定方式工作的计算机可读存储器中，使得存储在该计算机可读存储器中的指令产生包括指令装置的制造品，该指令装置实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能。

这些计算机程序指令也可装载到计算机或其他可编程数据处理设备上，使得在计算机或其他可编程设备上执行一系列操作步骤以产生计算机实现的处理，从而在计算机或其他可编程设备上执行的指令提供用于实现在流程图一个流程或多个流程和/或方框图一个方框或多个方框中指定的功能的步骤。

在一个典型的配置中，计算机设备包括一个或多个处理器(CPU)、存储器、输入/输出接口、网络接口和内存。存储器可能包括计算机可读介质中的非永久性存储器，随机存取存储器(RAM)和/或非易失性内存等形式，如只读存储器(ROM)或闪存(flash RAM)。存储器是计算机可读介质的示例。

计算机可读介质包括永久性和非永久性、可移动和非可移动媒体可以由任何方法或技术来实现信息存储。信息可以是计算机可读指令、数据结构、程序的模块或其他数据。计算机的存储介质的例子包括，但不限于相变内存(PRAM)、静态随机存取存储器(SRAM)、动态随机存取存储器(DRAM)、其他类型的随机存取存储器(RAM)、只读存储器(ROM)、电可擦除可编程只读存储器(EEPROM)、快闪记忆体或其他内存技术、只读光盘只读存储器(CD-ROM)、数字多功能光盘(DVD)或其他光学存储、磁盒式磁带，磁带磁磁盘存储或其他磁性存储设备或任何其他非传输介质，可用于存储可以被计算设备访问的信息。按照本文中的界定，计算机可读介质不包括暂存电脑可读媒体(transitory media)，如调制的数据信号和载波。

本领域技术人员应明白，本发明实施例可提供为方法、系统或计算机程序产品。因此，本发明实施例可采用完全硬件实施例、完全软件实施例或结合软件和硬件方面的实施例的形式。而且，本发明实施例可采用在一个或多个其中包含有计算机可用程序代码的计算机可用存储介质(包括但不限于磁盘存储器、CD-ROM、光学存储器等)上实施的计算机程序产品的形式。

以上结合附图详细描述了本发明实施例的可选实施方式，但是，本发明实施例并不限于上述实施方式中的具体细节，在本发明实施例的技术构思范围内，可以对本发明实施例的技术方案进行多种简单变型，这些简单变型均属于本发明实施例的保护范围。

另外需要说明的是，在上述具体实施方式中所描述的各个具体技术特征，在不矛盾的情况下，可以通过任何合适的方式进行组合。为了避免不必要的重复，本发明实施例对各种可能的组合方式不再另行说明。

Claims

1.一种代码自动生成方法，其特征在于，所述方法为基于ORM的自定义数据类型的代码自动生成方法，包括如下步骤：

接收代码生成请求指令；

执行如下步骤：

步骤a.预定义基本数据类型映射关系；

步骤h.使用预置的代码模版生成业务逻辑代码。

2.根据权利要求1所述的一种代码自动生成方法，其特征在于，所述方法包括如下步骤：所述步骤a具体为预定义数据库基本数据类型与go语言基本数据类型映射关系。

3.根据权利要求1所述的一种代码自动生成方法，其特征在于，在步骤e中，利用sql语句查询。

4.一种代码自动生成装置，其特征在于，所述装置为基于ORM的自定义数据类型的代码自动生成装置，包括代码生成请求指令接收模块、基本数据类型映射关系模块、配置信息模块、信息保存模块、数据库读取模块、表结构信息查询模块以及配置信息获取模块、自定义的数据类型映射模块、代码生成模块；

所述配置信息模块用于配置需要生成映射关系的表的信息及特殊场景的自定义数据类型，包括数据库连接地址、数据库名、需需要生成映射关系的表名、需要扩展的表和字段，及字段映射后的自定义数据类型；

5.根据权利要求4所述的一种代码自动生成装置，其特征在于，所述基本数据类型映射关系模块具体用于预定义数据库基本数据类型与go语言基本数据类型映射关系。

6.根据权利要求4所述的一种代码自动生成装置，其特征在于，所述表结构信息查询模块具体利用sql语句查询。

7.一种存储介质，其特征在于，所述存储介质中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由处理器加载并执行以实现如权利要求1至3中任一项权利要求所述的一种代码自动生成方法。

8.一种终端设备，其特征在于，所述终端设备包括处理器和存储器，所述存储器中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由所述处理器加载并执行以实现如权利要求1至3中任一项权利要求所述的一种代码自动生成方法。