update generator

pirosiki197 · pirosiki197 · commit a33bdba28e6e · 2025-01-25T00:01:43.000+09:00
diff --git a/generator.go b/generator.go
@@ -14,15 +14,16 @@ type Generator struct {
 }
 
 type data struct {
-	PackageName  string
-	CachePlanRaw string
+	PackageName    string
+	CachePlanRaw   string
+	TableSchemaRaw string
 }
 
-func NewGenerator(cachePlanRaw string) *Generator {
+func NewGenerator(cachePlanRaw string, tableSchemaRaw string) *Generator {
 	return &Generator{
 		driverTmpl: template.Must(template.ParseFiles("template/driver.tmpl")),
 		stmtTmpl:   template.Must(template.ParseFiles("template/stmt.tmpl")),
-		data:       data{CachePlanRaw: escapeGoString(cachePlanRaw)},
+		data:       data{CachePlanRaw: escapeGoString(cachePlanRaw), TableSchemaRaw: escapeGoString(tableSchemaRaw)},
 	}
 }
 

Original file line number	Diff line number	Diff line change
`@@ -14,15 +14,16 @@ type Generator struct {`
`14`	`14`	`}`
`15`	`15`
`16`	`16`	`type data struct {`
`17`		`- PackageName string`
`18`		`- CachePlanRaw string`
	`17`	`+ PackageName string`
	`18`	`+ CachePlanRaw string`
	`19`	`+ TableSchemaRaw string`
`19`	`20`	`}`
`20`	`21`
`21`		`-func NewGenerator(cachePlanRaw string) *Generator {`
	`22`	`+func NewGenerator(cachePlanRaw string, tableSchemaRaw string) *Generator {`
`22`	`23`	`return &Generator{`
`23`	`24`	`driverTmpl: template.Must(template.ParseFiles("template/driver.tmpl")),`
`24`	`25`	`stmtTmpl: template.Must(template.ParseFiles("template/stmt.tmpl")),`
`25`		`- data: data{CachePlanRaw: escapeGoString(cachePlanRaw)},`
	`26`	`+ data: data{CachePlanRaw: escapeGoString(cachePlanRaw), TableSchemaRaw: escapeGoString(tableSchemaRaw)},`
`26`	`27`	`}`
`27`	`28`	`}`
`28`	`29`