plugin: add "exec start-time" to GeneralEvent (#11293) (#11322)

pingcap · Jul 18, 2019 · 3d7da06 · 3d7da06
1 parent 7ef017f
commit 3d7da06
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/executor/adapter.go b/executor/adapter.go
@@ -596,7 +596,8 @@ func (a *ExecStmt) logAudit() {
 		audit := plugin.DeclareAuditManifest(p.Manifest)
 		if audit.OnGeneralEvent != nil {
 			cmd := mysql.Command2Str[byte(atomic.LoadUint32(&a.Ctx.GetSessionVars().CommandValue))]
-			audit.OnGeneralEvent(context.Background(), sessVars, plugin.Log, cmd)
+			ctx := context.WithValue(context.Background(), plugin.ExecStartTimeCtxKey, a.StartTime)
+			audit.OnGeneralEvent(ctx, sessVars, plugin.Log, cmd)
 		}
 		return nil
 	})

diff --git a/plugin/audit.go b/plugin/audit.go
@@ -84,3 +84,8 @@ type AuditManifest struct {
 	// OnParseEvent will be called around parse logic.
 	OnParseEvent func(ctx context.Context, sctx *variable.SessionVars, event ParseEvent) error
 }
+
+const (
+	// ExecStartTimeCtxKey indicates stmt start execution time.
+	ExecStartTimeCtxKey = "ExecStartTime"
+)