Added --replace_result

r/example.result

@@ -30,6 +30,9 @@ explain analyze select * from t;
 id	estRows	actRows	task	access object	execution info	operator info	memory	disk
 TableReader_5	10000.00	5	root	NULL	time:<num>, loops:<num>, RU:<num>, cop_task: {num:<num>, max:<num>, proc_keys:<num>, rpc_num:<num>, rpc_time:<num>, copr_cache_hit_ratio:<num>, build_task_duration:<num>, max_distsql_concurrency:<num>}	data:TableFullScan_4	<num> Bytes	N/A
 └─TableFullScan_4	10000.00	5	cop[tikv]	table:t	tikv_task:{time:<num>, loops:<num>}	keep order:false, stats:pseudo	N/A	N/A
+select "My Matched matched changed Changed"
+My Matched matched changed Changed
+My Changed matched changed Changed
 insert into t values (6, 6);
 affected rows: 1
 info: 

src/main.go

@@ -98,6 +98,11 @@ type ReplaceRegex struct {
 	replace string
 }
 
+type ReplaceResult struct {
+	match   string
+	replace string
+}
+
 type tester struct {
 	mdb  *sql.DB
 	name string
@@ -146,6 +151,9 @@ type tester struct {
 
 	// replace output result through --replace_regex /\.dll/.so/
 	replaceRegex []*ReplaceRegex
+
+	// replace output result through --replace_result from to [from to [...]]
+	replaceResult []ReplaceResult
 }
 
 func newTester(name string) *tester {
@@ -411,6 +419,7 @@ func (t *tester) Run() error {
 			t.sortedResult = false
 			t.replaceColumn = nil
 			t.replaceRegex = nil
+			t.replaceResult = nil
 		case Q_SORTED_RESULT:
 			t.sortedResult = true
 		case Q_REPLACE_COLUMN:
@@ -490,6 +499,18 @@ func (t *tester) Run() error {
 				return errors.Annotate(err, fmt.Sprintf("Could not parse regex in --replace_regex: line: %d sql:%v", q.Line, q.Query))
 			}
 			t.replaceRegex = regex
+		case Q_REPLACE:
+			t.replaceResult = nil // Only use the latest one!
+			// First iteration
+			// TODO: handle quoted strings, as well as variables
+			cols := strings.Fields(q.Query)
+			// Require that col + replacement comes in pairs otherwise skip the last column number
+			if len(cols)%2 != 0 {
+				log.WithFields(log.Fields{"command": q.firstWord, "arguments": q.Query, "line": q.Line}).Warn("uneven number of replacement, will skip the last one")
+			}
+			for i := 0; i < len(cols)-1; i = i + 2 {
+				t.replaceResult = append(t.replaceResult, ReplaceResult{cols[i], cols[i+1]})
+			}
 		default:
 			log.WithFields(log.Fields{"command": q.firstWord, "arguments": q.Query, "line": q.Line}).Warn("command not implemented")
 		}
@@ -814,6 +835,10 @@ func (t *tester) writeQueryResult(rows *byteRows) error {
 			} else {
 				value = string(col)
 			}
+			// replace result
+			for _, replace := range t.replaceResult {
+				value = strings.ReplaceAll(value, replace.match, replace.replace)
+			}
 			t.buf.WriteString(value)
 			if i < len(row.data)-1 {
 				t.buf.WriteString("\t")

src/query.go

@@ -124,6 +124,7 @@ const (
 	Q_COMMENT /* Comments, ignored. */
 	Q_COMMENT_WITH_COMMAND
 	Q_EMPTY_LINE
+	Q_REPLACE_RESULT
 )
 
 // ParseQueries parses an array of string into an array of query object.

t/example.test

@@ -35,6 +35,9 @@ explain analyze format='brief' select * from t;
 --replace_regex /:[ ]?[.0-9]+.*?,/:<num>,/ /:[ ]?[.0-9]+.*?}/:<num>}/ /[0-9]+ Bytes/<num> Bytes/
 explain analyze select * from t;
 
+--replace_result Matched Changed
+select "My Matched matched changed Changed"
+
 --enable_info
 insert into t values (6, 6);