From dc9729849f424e34f8d512ffba6980d4fbd645b8 Mon Sep 17 00:00:00 2001 From: wangyunbo Date: Mon, 21 Sep 2020 18:09:44 +0800 Subject: [PATCH] fix return database name Signed-off-by: wangyunbo --- go/mysql/query.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/go/mysql/query.go b/go/mysql/query.go index 94449457b10..fce6d960db0 100644 --- a/go/mysql/query.go +++ b/go/mysql/query.go @@ -948,8 +948,16 @@ func (c *Conn) writeFields(result *sqltypes.Result) error { return err } + var returnDatabaseName string + if len(result.Fields) >= 1 && strings.HasPrefix(result.Fields[0].Database, "vt_") { + returnDatabaseName = strings.TrimPrefix(result.Fields[0].Database, "vt_") + } else { + returnDatabaseName = result.Fields[0].Database + } + // Now send each Field. for _, field := range result.Fields { + field.Database = returnDatabaseName if err := c.writeColumnDefinition(field); err != nil { return err }