Handling of the decimal column in the build query (#17)

Toru-Takagi · web-flow · commit d3900db1421e · 2024-01-22T13:18:25.000-08:00
diff --git a/internal/core/gen.go b/internal/core/gen.go
@@ -173,6 +173,9 @@ func jdbcGet(t ktType, idx int) string {
 		}
 		return fmt.Sprintf(`results.getObject(%d) as%s %s`, idx, nullCast, t.Name)
 	}
+	if t.IsBigDecimal() {
+		return fmt.Sprintf(`results.getBigDecimal(%d)`, idx)
+	}
 	return fmt.Sprintf(`results.get%s(%d)`, t.Name, idx)
 }
 
@@ -364,6 +367,10 @@ func (t ktType) IsUUID() bool {
 	return t.Name == "UUID"
 }
 
+func (t ktType) IsBigDecimal() bool {
+	return t.Name == "java.math.BigDecimal"
+}
+
 func makeType(req *plugin.GenerateRequest, col *plugin.Column) ktType {
 	typ, isEnum := ktInnerType(req, col)
 	return ktType{

Original file line number	Diff line number	Diff line change
`@@ -173,6 +173,9 @@ func jdbcGet(t ktType, idx int) string {`
`173`	`173`	`}`
`174`	`174`	return fmt.Sprintf(`results.getObject(%d) as%s %s`, idx, nullCast, t.Name)
`175`	`175`	`}`
	`176`	`+ if t.IsBigDecimal() {`
	`177`	+ return fmt.Sprintf(`results.getBigDecimal(%d)`, idx)
	`178`	`+ }`
`176`	`179`	return fmt.Sprintf(`results.get%s(%d)`, t.Name, idx)
`177`	`180`	`}`
`178`	`181`
`@@ -364,6 +367,10 @@ func (t ktType) IsUUID() bool {`
`364`	`367`	`return t.Name == "UUID"`
`365`	`368`	`}`
`366`	`369`
	`370`	`+func (t ktType) IsBigDecimal() bool {`
	`371`	`+ return t.Name == "java.math.BigDecimal"`
	`372`	`+}`
	`373`	`+`
`367`	`374`	`func makeType(req plugin.GenerateRequest, col plugin.Column) ktType {`
`368`	`375`	`typ, isEnum := ktInnerType(req, col)`
`369`	`376`	`return ktType{`