update replace

4 years ago · 23aeedabb4
1 changed files with 4 additions and 2 deletions
--- a/dtmcli/dtmimp/utils.go
+++ b/dtmcli/dtmimp/utils.go
@ -239,8 +239,10 @@ func DeferDo(rerr *error, success func() error, fail func() error) {

 // Escape solve CodeQL reported problem
 func Escape(input string) string {
-	replacer := strings.NewReplacer("'", "", "\n", "", "\r", "", " ", "")
-	return replacer.Replace(input)
+	v := strings.Replace(input, "\n", "", -1)
+	v = strings.Replace(v, "\r", "", -1)
+	v = strings.Replace(v, ";", "", -1)
+	return strings.Replace(v, "'", "", -1)
 }

 // EscapeGet escape get