public_git
/
postgres-operator
mirror of https://github.com/zalando/postgres-operator.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

reflect code review

This commit is contained in:
Felix Kunde 2021-04-21 17:16:43 +02:00
parent f18cc4913d
commit 5eac3e86ac
3 changed files with 11 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/cluster/cluster.go
View File

@ -646,11 +646,10 @@ func (c *Cluster) Update(oldSpec, newSpec *acidv1.Postgresql) error {
c.logger.Debugf("initialize users")
// save current state of pgUsers to check for deleted roles later
if len(c.pgUsers) > 0 {
usersCache := map[string]spec.PgUser{}
c.pgUsersCache = map[string]spec.PgUser{}
for k, v := range c.pgUsers {
usersCache[k] = v
c.pgUsersCache[k] = v
}
c.pgUsersCache = usersCache
}
if err := c.initUsers(); err != nil {
c.logger.Errorf("could not init users: %v", err)

7
pkg/cluster/sync.go
View File

@ -39,11 +39,10 @@ func (c *Cluster) Sync(newSpec *acidv1.Postgresql) error {
// save current state of pgUsers to check for deleted roles later
if len(c.pgUsers) > 0 {
usersCache := map[string]spec.PgUser{}
c.pgUsersCache = map[string]spec.PgUser{}
for k, v := range c.pgUsers {
usersCache[k] = v
c.pgUsersCache[k] = v
}
c.pgUsersCache = usersCache
}
if err = c.initUsers(); err != nil {
err = fmt.Errorf("could not init users: %v", err)
@ -586,7 +585,7 @@ func (c *Cluster) syncRoles() (err error) {
for _, u := range c.pgUsers {
userNames = append(userNames, u.Name)
// add team user name with rename suffix in case we need to rename it back
// add team member role name with rename suffix in case we need to rename it back
if u.Origin == spec.RoleOriginTeamsAPI {
userNames = append(userNames, u.Name+constants.RoleRenameSuffix)
}

20
pkg/util/users/users.go
View File

@ -72,8 +72,8 @@ func (strategy DefaultUserSyncStrategy) ProduceSyncRequests(dbUsers spec.PgUserM
}
}
// No existing roles are deleted or stripped of role memebership/flags
// but they will be renamed acting as a simple blocker
// No existing roles are deleted or stripped of role membership/flags
// but they will be renamed acting as a simple login blocker
for name, dbUser := range dbUsers {
if _, exists := newUsers[name]; !exists {
reqs = append(reqs, spec.PgSyncUserRequest{Kind: spec.PGSyncUserRename, User: dbUser})
@ -141,18 +141,10 @@ func (strategy DefaultUserSyncStrategy) alterPgUserSet(user spec.PgUser, db *sql
func (strategy DefaultUserSyncStrategy) alterPgUserRename(user spec.PgUser, db *sql.DB) error {
var query string
renamedBack := false
nameSuffixDiff := len(user.Name) - len(constants.RoleRenameSuffix)
if nameSuffixDiff > 0 {
if user.Name[nameSuffixDiff:] == constants.RoleRenameSuffix {
query = fmt.Sprintf(alterUserRenameSQL, user.Name, user.Name[:nameSuffixDiff], "")
renamedBack = true
}
}
if !renamedBack {
if strings.HasSuffix(user.Name, constants.RoleRenameSuffix) {
newName := strings.TrimSuffix(user.Name, constants.RoleRenameSuffix)
query = fmt.Sprintf(alterUserRenameSQL, user.Name, newName, "")
} else {
query = fmt.Sprintf(alterUserRenameSQL, user.Name, user.Name, constants.RoleRenameSuffix)
}