diff --git a/pkg/postgres/azure.go b/pkg/postgres/azure.go index 5edbc5095..b15c8438a 100644 --- a/pkg/postgres/azure.go +++ b/pkg/postgres/azure.go @@ -3,6 +3,8 @@ package postgres import ( "fmt" "strings" + + "github.com/go-logr/logr" ) type azurepg struct { @@ -47,3 +49,8 @@ func (azpg *azurepg) CreateDB(dbname, role string) error { return azpg.pg.CreateDB(dbname, role) } + +func (azpg *azurepg) DropRole(role, newOwner, database string, logger logr.Logger) error { + azNewOwner := azpg.GetRoleForLogin(newOwner) + return azpg.pg.DropRole(role, azNewOwner, database, logger) +}