[发明专利]代码生成方法、装置、电子设备及存储介质

权利要求书

1.一种代码生成方法，其特征在于，包括：

响应于代码生成指令，执行代码生成工具对应的代码生成逻辑，并调用预设插件；所述预设插件包含多个代码编辑方法，所述多个代码编辑方法被执行后实现将已生成代码中的目标字符串替换为预设字符串；所述多个代码编辑方法为基于所述代码生成逻辑中的多个目标片段，分别进行重写得到的方法；

在所述代码生成逻辑的执行过程中，当检测到当前待执行片段属于任一所述目标片段时，转为执行所述插件中与所述当前待执行片段对应的代码编辑方法，并在该代码编辑方法执行结束后，返回执行所述当前待执行片段的下一片段；

返回执行上一步骤，直到所述代码生成逻辑执行完毕，得到生成的代码。

2.根据权利要求1所述的方法，其特征在于，所述当检测到当前待执行片段属于任一所述目标片段时，转为执行所述插件中与所述当前待执行片段对应的代码编辑方法，包括：

当检测到当前待执行片段属于第一目标片段时，转为执行第一代码编辑方法，以获取所述预设字符串；所述第一代码编辑方法为针对所述第一目标片段重写的方法；

当检测到当前待执行片段属于第二目标片段，转为执行第二代码编辑方法，以更新所述已生成代码中包含所述目标字符串的类名和接口方法名，将所述类名和所述接口方法名中的目标字符串替换为预设字符串；所述第二代码编辑方法为针对所述第二目标片段重写的方法；

当检测到当前待执行片段属于第三目标片段，转为执行第三代码编辑方法，以更新已生成代码中包含所述目标字符串的接口参数和注释信息，将所述接口参数和注释信息中的目标字符串替换为预设字符串；所述第三代码编辑方法为针对所述第三目标片段重写的方法；

当检测到当前待执行片段属于第四目标片段，转为执行第四代码编辑方法，以更新已生成代码中包含所述目标字符串的接口配置文件中的配置信息，将所述配置信息中的目标字符串替换为预设字符串；所述第四代码编辑方法为针对所述第四目标片段重写的方法。

3.根据权利要求2所述的方法，其特征在于，所述代码生成工具为MyBatis的代码生成器，所述代码生成逻辑为操作数据库接口的生成逻辑；所述第一代码编辑方法为重写后的参数验证方法；

所述执行所述第一代码编辑方法，以获取所述预设字符串，包括：

调用所述重写后的参数验证方法包含的变量获取方法函数，以获取以下预设字符串：

用于更新查询条件对象类的类名的第一字符串；

用于更新操作数据库接口的接口方法名的第二字符串；

用于更新接口参数与注释信息的第三字符串；

用于更新操作数据库接口配置文件中的配置信息的第四字符串。

4.根据权利要求3所述的方法，其特征在于，所述第二代码编辑方法为重写后的初始化方法；

所述执行所述第二代码编辑方法，以更新所述已生成代码中包含所述目标字符串的类名和接口方法名，将所述类名和所述接口方法名中的目标字符串替换为预设字符串，包括：

调用所述重写后的初始化方法包含的第一参数替换方法函数，将所述已生成查询条件对象类的类名中包含的所述目标字符串，替换为所述第一字符串；

调用所述重写后的初始化方法包含的第二参数替换方法函数，将所述已生成操作数据库接口代码中接口方法名包含的所述目标字符串，替换为所述第二字符串。

5.根据权利要求3所述的方法，其特征在于，所述第三代码编辑方法为重写后的生成调用方法；

所述执行所述第三代码编辑方法，以更新已生成代码中包含所述目标字符串的接口参数和注释信息，将所述接口参数和注释信息中的目标字符串替换为预设字符串，包括：

调用所述重写后的生成调用方法中包含的参数获取方法函数，将所述已生成操作数据库接口代码中接口参数包含的所述目标字符串，替换为所述第三字符串；

调用所述重写后的生成调用方法中包含的注释获取方法函数，将所述已生成操作数据库接口代码中注释信息包含的所述目标字符串，替换为所述第三字符串。