升级生成dao的工具

2023-02-28 16:18:20 +08:00
parent 0dd215a1b9
commit e2aed27ba7
4 changed files with 27 additions and 7 deletions
--- a/sql2go/generate_dao.go
+++ b/sql2go/generate_dao.go
@ -16,11 +16,11 @@ import (
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 10:15 上午 2021/11/17
-func GenerateDao(sql string, packageName string) (string, error) {
+func GenerateDao(sql string, packageName string, withGetTableNameFunc bool) (string, error) {
 	if len(packageName) == 0 {
 		packageName = "dao"
 	}
-	dataStruct, basic, err := ParseCreateTableSql(sql)
+	dataStruct, basic, err := ParseCreateTableSql(sql, withGetTableNameFunc)
 	if nil != err {
 		return "", err
 	}