mirror of
https://github.com/zalando/postgres-operator.git
synced 2026-09-30 10:42:14 +02:00
fix team member deprecation (#2072)
This commit is contained in:
@@ -231,7 +231,8 @@ func (c *Cluster) readPgUsersFromDatabase(userNames []string) (users spec.PgUser
|
||||
parameters[fields[0]] = fields[1]
|
||||
}
|
||||
|
||||
if strings.HasSuffix(rolname, c.OpConfig.RoleDeletionSuffix) {
|
||||
// consider NOLOGIN roles with deleted suffix as deprecated users
|
||||
if strings.HasSuffix(rolname, c.OpConfig.RoleDeletionSuffix) && !rolcanlogin {
|
||||
roldeleted = true
|
||||
}
|
||||
|
||||
|
||||
+19
-13
@@ -104,18 +104,19 @@ func (c *Cluster) Sync(newSpec *acidv1.Postgresql) error {
|
||||
if !(c.databaseAccessDisabled() || c.getNumberOfInstances(&newSpec.Spec) <= 0 || c.Spec.StandbyCluster != nil) {
|
||||
c.logger.Debug("syncing roles")
|
||||
if err = c.syncRoles(); err != nil {
|
||||
err = fmt.Errorf("could not sync roles: %v", err)
|
||||
return err
|
||||
// remember all cached users in c.pgUsers
|
||||
for cachedUserName, cachedUser := range c.pgUsersCache {
|
||||
c.pgUsers[cachedUserName] = cachedUser
|
||||
}
|
||||
c.logger.Errorf("could not sync roles: %v", err)
|
||||
}
|
||||
c.logger.Debug("syncing databases")
|
||||
if err = c.syncDatabases(); err != nil {
|
||||
err = fmt.Errorf("could not sync databases: %v", err)
|
||||
return err
|
||||
c.logger.Errorf("could not sync databases: %v", err)
|
||||
}
|
||||
c.logger.Debug("syncing prepared databases with schemas")
|
||||
if err = c.syncPreparedDatabases(); err != nil {
|
||||
err = fmt.Errorf("could not sync prepared database: %v", err)
|
||||
return err
|
||||
c.logger.Errorf("could not sync prepared database: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -933,10 +934,7 @@ func (c *Cluster) syncRoles() (err error) {
|
||||
}
|
||||
}
|
||||
|
||||
// copy map for ProduceSyncRequests to include also system users
|
||||
for userName, pgUser := range c.pgUsers {
|
||||
newUsers[userName] = pgUser
|
||||
}
|
||||
// search also for system users
|
||||
for _, systemUser := range c.systemUsers {
|
||||
userNames = append(userNames, systemUser.Name)
|
||||
newUsers[systemUser.Name] = systemUser
|
||||
@@ -950,13 +948,21 @@ func (c *Cluster) syncRoles() (err error) {
|
||||
// update pgUsers where a deleted role was found
|
||||
// so that they are skipped in ProduceSyncRequests
|
||||
for _, dbUser := range dbUsers {
|
||||
if originalUser, exists := deletedUsers[dbUser.Name]; exists {
|
||||
recreatedUser := c.pgUsers[originalUser]
|
||||
originalUsername, foundDeletedUser := deletedUsers[dbUser.Name]
|
||||
// check if original user does not exist in dbUsers
|
||||
_, originalUserAlreadyExists := dbUsers[originalUsername]
|
||||
if foundDeletedUser && !originalUserAlreadyExists {
|
||||
recreatedUser := c.pgUsers[originalUsername]
|
||||
recreatedUser.Deleted = true
|
||||
c.pgUsers[originalUser] = recreatedUser
|
||||
c.pgUsers[originalUsername] = recreatedUser
|
||||
}
|
||||
}
|
||||
|
||||
// last but not least copy pgUsers to newUsers to send to ProduceSyncRequests
|
||||
for _, pgUser := range c.pgUsers {
|
||||
newUsers[pgUser.Name] = pgUser
|
||||
}
|
||||
|
||||
pgSyncRequests := c.userSyncStrategy.ProduceSyncRequests(dbUsers, newUsers)
|
||||
if err = c.userSyncStrategy.ExecuteSyncRequests(pgSyncRequests, c.pgDb); err != nil {
|
||||
return fmt.Errorf("error executing sync statements: %v", err)
|
||||
|
||||
+17
-10
@@ -43,7 +43,8 @@ func (strategy DefaultUserSyncStrategy) ProduceSyncRequests(dbUsers spec.PgUserM
|
||||
|
||||
var reqs []spec.PgSyncUserRequest
|
||||
for name, newUser := range newUsers {
|
||||
// do not create user that exists in DB with deletion suffix
|
||||
// do not create user when there exists a user with the same name plus deletion suffix
|
||||
// instead request a renaming of the deleted user back to the original name (see * below)
|
||||
if newUser.Deleted {
|
||||
continue
|
||||
}
|
||||
@@ -82,22 +83,28 @@ func (strategy DefaultUserSyncStrategy) ProduceSyncRequests(dbUsers spec.PgUserM
|
||||
}
|
||||
}
|
||||
|
||||
// No existing roles are deleted or stripped of role membership/flags
|
||||
// no existing roles are deleted or stripped of role membership/flags
|
||||
// but team roles will be renamed and denied from LOGIN
|
||||
for name, dbUser := range dbUsers {
|
||||
if _, exists := newUsers[name]; !exists {
|
||||
// toggle LOGIN flag based on role deletion
|
||||
userFlags := make([]string, len(dbUser.Flags))
|
||||
userFlags = append(userFlags, dbUser.Flags...)
|
||||
if dbUser.Deleted {
|
||||
dbUser.Flags = util.StringSliceReplaceElement(dbUser.Flags, constants.RoleFlagNoLogin, constants.RoleFlagLogin)
|
||||
// * user with deletion suffix and NOLOGIN found in database
|
||||
// grant back LOGIN and rename only if original user is wanted and does not exist in database
|
||||
originalName := strings.TrimSuffix(name, strategy.RoleDeletionSuffix)
|
||||
_, originalUserWanted := newUsers[originalName]
|
||||
_, originalUserAlreadyExists := dbUsers[originalName]
|
||||
if !originalUserWanted || originalUserAlreadyExists {
|
||||
continue
|
||||
}
|
||||
// a deleted dbUser has no NOLOGIN flag, so we can add the LOGIN flag
|
||||
dbUser.Flags = append(dbUser.Flags, constants.RoleFlagLogin)
|
||||
} else {
|
||||
// user found in database and not wanted in newUsers - replace LOGIN flag with NOLOGIN
|
||||
dbUser.Flags = util.StringSliceReplaceElement(dbUser.Flags, constants.RoleFlagLogin, constants.RoleFlagNoLogin)
|
||||
}
|
||||
if !util.IsEqualIgnoreOrder(userFlags, dbUser.Flags) {
|
||||
reqs = append(reqs, spec.PgSyncUserRequest{Kind: spec.PGsyncUserAlter, User: dbUser})
|
||||
}
|
||||
|
||||
// request ALTER ROLE to grant or revoke LOGIN
|
||||
reqs = append(reqs, spec.PgSyncUserRequest{Kind: spec.PGsyncUserAlter, User: dbUser})
|
||||
// request RENAME which will happen on behalf of the pgUser.Deleted field
|
||||
reqs = append(reqs, spec.PgSyncUserRequest{Kind: spec.PGSyncUserRename, User: dbUser})
|
||||
}
|
||||
}
|
||||
|
||||
@@ -62,6 +62,8 @@ var substractTest = []struct {
|
||||
}{
|
||||
{[]string{"a", "b", "c", "d"}, []string{"a", "b", "c", "d"}, []string{}, true},
|
||||
{[]string{"a", "b", "c", "d"}, []string{"a", "bb", "c", "d"}, []string{"b"}, false},
|
||||
{[]string{""}, []string{"b"}, []string{""}, false},
|
||||
{[]string{"a"}, []string{""}, []string{"a"}, false},
|
||||
}
|
||||
|
||||
var sliceContaintsTest = []struct {
|
||||
|
||||
Reference in New Issue
Block a user