Alter sequences owner implementation (#3)

Netcracker · Jan 28, 2025 · 9ae0840 · 9ae0840
1 parent f9ddf70
commit 9ae0840
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/adapter/basic/basic_roles.go b/adapter/basic/basic_roles.go
@@ -201,7 +201,7 @@ func (sa ServiceAdapter) GrantUsersAccordingRoles(ctx context.Context, dbName st
 			// altering owners for DB resources
 			alterOperations := map[string]func(string, string, string) string{
 				getTablesListQuery(schema, isExternalPg):      alterOwnerForTable,
-				getSequenceListQuery(schema, isExternalPg):    alterOwnerForTable,
+				getSequenceListQuery(schema, isExternalPg):    alterOwnerForSequence,
 				getLargeObjectsListQuery(isExternalPg):        alterOwnerForLargeObject,
 				getViewsListQuery(schema, isExternalPg):       alterViewOwnerQuery,
 				getFunctionsListQuery(schema, isExternalPg):   alterFunctionOwnerQuery,

diff --git a/adapter/basic/sql_helper.go b/adapter/basic/sql_helper.go
@@ -112,6 +112,10 @@ func alterOwnerForTable(schema, tableName, userName string) string {
 	return fmt.Sprintf("ALTER TABLE \"%s\".\"%s\" OWNER TO \"%s\"", schema, tableName, userName)
 }
 
+func alterOwnerForSequence(sequenceName, schema, newOwner string) string {
+	return fmt.Sprintf(`ALTER SEQUENCE "%s"."%s" OWNER TO "%s"`, schema, sequenceName, newOwner)
+}
+
 func alterOwnerForLargeObject(schema, oid, userName string) string {
 	return fmt.Sprintf("ALTER LARGE OBJECT %s OWNER TO \"%s\";", oid, userName) //TODO:
 }