session: set Sleep state for process info (#7826) (#7839)

pingcap · Oct 8, 2018 · 8823f12 · 8823f12
1 parent 6a1e94f
commit 8823f12
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/server/tidb_test.go b/server/tidb_test.go
@@ -526,3 +526,17 @@ func (ts *TidbTestSuite) TestSumAvg(c *C) {
 	c.Parallel()
 	runTestSumAvg(c)
 }
+
+func (ts *TidbTestSuite) TestShowProcess(c *C) {
+	qctx, err := ts.tidbdrv.OpenCtx(uint64(0), 0, uint8(tmysql.DefaultCollationID), "test", nil)
+	c.Assert(err, IsNil)
+	ctx := context.Background()
+	results, err := qctx.Execute(ctx, "select 1")
+	c.Assert(err, IsNil)
+	pi := qctx.ShowProcess()
+	c.Assert(pi.Command, Equals, "Query")
+	results[0].Close()
+	pi = qctx.ShowProcess()
+	c.Assert(pi.Command, Equals, "Sleep")
+	qctx.Close()
+}
diff --git a/session/session.go b/session/session.go
@@ -712,6 +712,9 @@ func (s *session) SetProcessInfo(sql string) {
 		State:   s.Status(),
 		Info:    sql,
 	}
+	if sql == "" {
+		pi.Command = "Sleep"
+	}
 	if s.sessionVars.User != nil {
 		pi.User = s.sessionVars.User.Username
 		pi.Host = s.sessionVars.User.Hostname