plugin: support dynamic enable/disable plugins(#11122)(#10923)(#10506)

executor/admin_plugins.go

@@ -0,0 +1,52 @@
+// Copyright 2019 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package executor
+
+import (
+	"context"
+
+	"github.com/pingcap/tidb/domain"
+	"github.com/pingcap/tidb/planner/core"
+	"github.com/pingcap/tidb/plugin"
+	"github.com/pingcap/tidb/util/chunk"
+)
+
+// AdminPluginsExec indicates AdminPlugins executor.
+type AdminPluginsExec struct {
+	baseExecutor
+	Action  core.AdminPluginsAction
+	Plugins []string
+}
+
+// Next implements the Executor Next interface.
+func (e *AdminPluginsExec) Next(ctx context.Context, _ *chunk.Chunk) error {
+	switch e.Action {
+	case core.Enable:
+		return e.changeDisableFlagAndFlush(false)
+	case core.Disable:
+		return e.changeDisableFlagAndFlush(true)
+	}
+	return nil
+}
+
+func (e *AdminPluginsExec) changeDisableFlagAndFlush(disabled bool) error {
+	dom := domain.GetDomain(e.ctx)
+	for _, pluginName := range e.Plugins {
+		err := plugin.ChangeDisableFlagAndFlush(dom, pluginName, disabled)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}

executor/builder.go

@@ -86,6 +86,8 @@ func (b *executorBuilder) build(p plannercore.Plan) Executor {
 		return b.buildChecksumTable(v)
 	case *plannercore.ReloadExprPushdownBlacklist:
 		return b.buildReloadExprPushdownBlacklist(v)
+	case *plannercore.AdminPlugins:
+		return b.buildAdminPlugins(v)
 	case *plannercore.DDL:
 		return b.buildDDL(v)
 	case *plannercore.Deallocate:
@@ -432,6 +434,10 @@ func (b *executorBuilder) buildReloadExprPushdownBlacklist(v *plannercore.Reload
 	return &ReloadExprPushdownBlacklistExec{baseExecutor{ctx: b.ctx}}
 }
 
+func (b *executorBuilder) buildAdminPlugins(v *plannercore.AdminPlugins) Executor {
+	return &AdminPluginsExec{baseExecutor: baseExecutor{ctx: b.ctx}, Action: v.Action, Plugins: v.Plugins}
+}
+
 func (b *executorBuilder) buildDeallocate(v *plannercore.Deallocate) Executor {
 	base := newBaseExecutor(b.ctx, nil, v.ExplainID())
 	base.initCap = chunk.ZeroCapacity

executor/show.go

@@ -823,7 +823,7 @@ func (e *ShowExec) fetchShowPlugins() error {
 	tiPlugins := plugin.GetAll()
 	for _, ps := range tiPlugins {
 		for _, p := range ps {
-			e.appendRow([]interface{}{p.Name, p.State.String(), p.Kind.String(), p.Path, p.License, strconv.Itoa(int(p.Version))})
+			e.appendRow([]interface{}{p.Name, p.StateValue(), p.Kind.String(), p.Path, p.License, strconv.Itoa(int(p.Version))})
 		}
 	}
 	return nil

go.mod

@@ -52,7 +52,7 @@ require (
 	github.com/pingcap/goleveldb v0.0.0-20171020084629-8d44bfdf1030
 	github.com/pingcap/kvproto v0.0.0-20190429124202-32a5ba2af0f7
 	github.com/pingcap/log v0.0.0-20190307075452-bd41d9273596
-	github.com/pingcap/parser v0.0.0-20190703021824-628683061f71
+	github.com/pingcap/parser v0.0.0-20190711031004-8190881c7ec9
 	github.com/pingcap/pd v2.1.12+incompatible
 	github.com/pingcap/tidb-tools v2.1.3-0.20190116051332-34c808eef588+incompatible
 	github.com/pingcap/tipb v0.0.0-20180910045846-371b48b15d93

go.sum

@@ -109,8 +109,8 @@ github.com/pingcap/kvproto v0.0.0-20190429124202-32a5ba2af0f7 h1:+wEqJTc74Jvoxen
 github.com/pingcap/kvproto v0.0.0-20190429124202-32a5ba2af0f7/go.mod h1:0gwbe1F2iBIjuQ9AH0DbQhL+Dpr5GofU8fgYyXk+ykk=
 github.com/pingcap/log v0.0.0-20190307075452-bd41d9273596 h1:t2OQTpPJnrPDGlvA+3FwJptMTt6MEPdzK1Wt99oaefQ=
 github.com/pingcap/log v0.0.0-20190307075452-bd41d9273596/go.mod h1:WpHUKhNZ18v116SvGrmjkA9CBhYmuUTKL+p8JC9ANEw=
-github.com/pingcap/parser v0.0.0-20190703021824-628683061f71 h1:ErX0Ww42LBHTSlK2Q8f8/86CUEg+dUpeyYbMt0iNxkg=
-github.com/pingcap/parser v0.0.0-20190703021824-628683061f71/go.mod h1:1FNvfp9+J0wvc4kl8eGNh7Rqrxveg15jJoWo/a0uHwA=
+github.com/pingcap/parser v0.0.0-20190711031004-8190881c7ec9 h1:89I1Q867PRSxMrBug971kpaV+lGoZM7Q9EqMRTxgkx4=
+github.com/pingcap/parser v0.0.0-20190711031004-8190881c7ec9/go.mod h1:1FNvfp9+J0wvc4kl8eGNh7Rqrxveg15jJoWo/a0uHwA=
 github.com/pingcap/pd v2.1.12+incompatible h1:6N3LBxx2aSZqT+IWEG730EDNDttP7dXO8J6yvBh+HXw=
 github.com/pingcap/pd v2.1.12+incompatible/go.mod h1:nD3+EoYes4+aNNODO99ES59V83MZSI+dFbhyr667a0E=
 github.com/pingcap/tidb-tools v2.1.3-0.20190116051332-34c808eef588+incompatible h1:e9Gi/LP9181HT3gBfSOeSBA+5JfemuE4aEAhqNgoE4k=

planner/core/common_plans.go

@@ -131,6 +131,23 @@ type ReloadExprPushdownBlacklist struct {
 	baseSchemaProducer
 }
 
+// AdminPluginsAction indicate action will be taken on plugins.
+type AdminPluginsAction int
+
+const (
+	// Enable indicates enable plugins.
+	Enable AdminPluginsAction = iota + 1
+	// Disable indicates disable plugins.
+	Disable
+)
+
+// AdminPlugins administrates tidb plugins.
+type AdminPlugins struct {
+	baseSchemaProducer
+	Action  AdminPluginsAction
+	Plugins []string
+}
+
 // Prepare represents prepare plan.
 type Prepare struct {
 	baseSchemaProducer

planner/core/planbuilder.go

@@ -583,6 +583,10 @@ func (b *planBuilder) buildAdmin(as *ast.AdminStmt) (Plan, error) {
 		ret = p
 	case ast.AdminReloadExprPushdownBlacklist:
 		return &ReloadExprPushdownBlacklist{}, nil
+	case ast.AdminPluginEnable:
+		return &AdminPlugins{Action: Enable, Plugins: as.Plugins}, nil
+	case ast.AdminPluginDisable:
+		return &AdminPlugins{Action: Disable, Plugins: as.Plugins}, nil
 	default:
 		return nil, ErrUnsupportedType.GenWithStack("Unsupported ast.AdminStmt(%T) for buildAdmin", as)
 	}

plugin/audit.go

@@ -16,7 +16,6 @@ package plugin
 import (
 	"context"
 
-	"github.com/pingcap/parser/auth"
 	"github.com/pingcap/tidb/sessionctx/variable"
 )
 
@@ -77,7 +76,7 @@ type AuditManifest struct {
 	Manifest
 	// OnConnectionEvent will be called when TiDB receive or disconnect from client.
 	// return error will ignore and close current connection.
-	OnConnectionEvent func(ctx context.Context, identity *auth.UserIdentity, event ConnectionEvent, info *variable.ConnectionInfo) error
+	OnConnectionEvent func(ctx context.Context, event ConnectionEvent, info *variable.ConnectionInfo) error
 	// OnGeneralEvent will be called during TiDB execution.
 	OnGeneralEvent func(ctx context.Context, sctx *variable.SessionVars, event GeneralEvent, cmd string)
 	// OnGlobalVariableEvent will be called when Change GlobalVariable.

plugin/const_test.go

@@ -0,0 +1,42 @@
+// Copyright 2019 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package plugin
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestConstToString(t *testing.T) {
+	kinds := map[fmt.Stringer]string{
+		Audit:                     "Audit",
+		Authentication:            "Authentication",
+		Schema:                    "Schema",
+		Daemon:                    "Daemon",
+		Uninitialized:             "Uninitialized",
+		Ready:                     "Ready",
+		Dying:                     "Dying",
+		Disable:                   "Disable",
+		Connected:                 "Connected",
+		Disconnect:                "Disconnect",
+		ChangeUser:                "ChangeUser",
+		PreAuth:                   "PreAuth",
+		ConnectionEvent(byte(15)): "",
+	}
+	for key, value := range kinds {
+		if key.String() != value {
+			t.Errorf("kind %s != %s", key.String(), kinds)
+		}
+	}
+}

plugin/helper_test.go

@@ -0,0 +1,54 @@
+// Copyright 2019 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package plugin
+
+import "testing"
+
+func TestPluginDeclare(t *testing.T) {
+	auditRaw := &AuditManifest{Manifest: Manifest{}}
+	auditExport := ExportManifest(auditRaw)
+	audit2 := DeclareAuditManifest(auditExport)
+	if audit2 != auditRaw {
+		t.Errorf("declare audit fail")
+	}
+
+	authRaw := &AuthenticationManifest{Manifest: Manifest{}}
+	authExport := ExportManifest(authRaw)
+	auth2 := DeclareAuthenticationManifest(authExport)
+	if auth2 != authRaw {
+		t.Errorf("declare auth fail")
+	}
+
+	schemaRaw := &SchemaManifest{Manifest: Manifest{}}
+	schemaExport := ExportManifest(schemaRaw)
+	schema2 := DeclareSchemaManifest(schemaExport)
+	if schema2 != schemaRaw {
+		t.Errorf("declare schema fail")
+	}
+
+	daemonRaw := &DaemonManifest{Manifest: Manifest{}}
+	daemonExport := ExportManifest(daemonRaw)
+	daemon2 := DeclareDaemonManifest(daemonExport)
+	if daemon2 != daemonRaw {
+		t.Errorf("declare daemon fail")
+	}
+}
+
+func TestDecode(t *testing.T) {
+	failID := ID("fail")
+	_, _, err := failID.Decode()
+	if err == nil {
+		t.Errorf("'fail' should not decode success")
+	}
+}