Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

etcdctl: create root role on auth enable if it does not yet exist

Browse Source 
Kind of tedious to add the root role when enabling auth; can just add
it automatically.
This commit is contained in:
Anthony Romano 2017-01-09 16:16:30 -08:00
parent 8adfc06084
commit d20f23c795
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
etcdctl/ctlv3/command/auth_command.go
View File

@ -17,6 +17,7 @@ package command
import (
"fmt"
"github.com/coreos/etcd/etcdserver/api/v3rpc/rpctypes"
"github.com/spf13/cobra"
)
@ -48,7 +49,21 @@ func authEnableCommandFunc(cmd *cobra.Command, args []string) {
}
ctx, cancel := commandCtx(cmd)
_, err := mustClientFromCmd(cmd).Auth.AuthEnable(ctx)
cli := mustClientFromCmd(cmd)
var err error
for err == nil {
if _, err = cli.AuthEnable(ctx); err == nil {
break
}
if err == rpctypes.ErrRootRoleNotExist {
if _, err = cli.RoleAdd(ctx, "root"); err != nil {
break
}
if _, err = cli.UserGrantRole(ctx, "root", "root"); err != nil {
break
}
}
}
cancel()
if err != nil {
ExitWithError(ExitError, err)