drainer: escape quote in value of column (#1132)

cartersz · web-flow · commit b607f343b539 · 2022-01-21T13:59:46.000+08:00
ref #1131
diff --git a/pkg/loader/model.go b/pkg/loader/model.go
@@ -519,6 +519,10 @@ func genOracleValue(column *model.ColumnInfo, value interface{}) string {
 		mysql.TypeYear, mysql.TypeFloat, mysql.TypeDouble, mysql.TypeNewDecimal:
 		return fmt.Sprintf("%v", value)
 	default:
-		return fmt.Sprintf("'%v'", value)
+		return fmt.Sprintf("'%s'", processOracleQuoteStringValue(fmt.Sprintf("%v", value)))
 	}
 }
+
+func processOracleQuoteStringValue(data string) string {
+	return strings.ReplaceAll(data, "'", "''")
+}

Original file line number	Diff line number	Diff line change
`@@ -519,6 +519,10 @@ func genOracleValue(column *model.ColumnInfo, value interface{}) string {`
`519`	`519`	`mysql.TypeYear, mysql.TypeFloat, mysql.TypeDouble, mysql.TypeNewDecimal:`
`520`	`520`	`return fmt.Sprintf("%v", value)`
`521`	`521`	`default:`
`522`		`- return fmt.Sprintf("'%v'", value)`
	`522`	`+ return fmt.Sprintf("'%s'", processOracleQuoteStringValue(fmt.Sprintf("%v", value)))`
`523`	`523`	`}`
`524`	`524`	`}`
	`525`	`+`
	`526`	`+func processOracleQuoteStringValue(data string) string {`
	`527`	`+ return strings.ReplaceAll(data, "'", "''")`
	`528`	`+}`