From 1432dad358c7c757128fc6d657186ebf2beb094b Mon Sep 17 00:00:00 2001 From: zhanghan2021 Date: Thu, 23 May 2024 09:38:19 +0800 Subject: [PATCH] add permission for a-tune plugin --- server/main.go | 1 + server/service/extentions.go | 22 ++++++++++++++++++---- 2 files changed, 19 insertions(+), 4 deletions(-) diff --git a/server/main.go b/server/main.go index 315e392..16eaf29 100644 --- a/server/main.go +++ b/server/main.go @@ -33,6 +33,7 @@ func main() { plugin.GlobalClient = client.DefaultClient(plugin.Init(config.Config().PluginAtune)) service.GetTags() service.AddExtentions() + service.AddPermission() err := router.HttpServerInit(config.Config().HttpServer) if err != nil { diff --git a/server/service/extentions.go b/server/service/extentions.go index db2952f..43ad6e8 100644 --- a/server/service/extentions.go +++ b/server/service/extentions.go @@ -11,26 +11,40 @@ func AddExtentions() { Type: common.ExtentionMachine, Name: "安装a-tune", URL: "/plugin/atune/atune_install", - Permission: "plugin.atune.agent/install", + Permission: "plugin.atune/agent_install", } me2 := &common.MachineExtention{ Type: common.ExtentionMachine, Name: "卸载a-tune", URL: "/plugin/atune/atune_uninstall", - Permission: "plugin.atune.agent/uninstall", + Permission: "plugin.atune/agent_uninstall", } pe1 := &common.PageExtention{ Type: common.ExtentionPage, Name: "任务列表", URL: "/task", - Permission: "plugin.prometheus.page/menu", + Permission: "plugin.atune.page/menu", } pe2 := &common.PageExtention{ Type: common.ExtentionPage, Name: "调优模板", URL: "/template", - Permission: "plugin.prometheus.page/menu", + Permission: "plugin.atune.page/menu", } ex = append(ex, me1, me2, pe1, pe2) plugin.GlobalClient.RegisterExtention(ex) } + +func AddPermission() { + var ps []common.Permission + p1 := common.Permission{ + Resource: "plugin.atune", + Operate: "agent_install", + } + p2 := common.Permission{ + Resource: "plugin.atune", + Operate: "agent_uninstall", + } + ps = append(ps, p1, p2) + plugin.GlobalClient.RegisterPermission(ps) +} -- Gitee