Override DropRole method for Azure (#62)

* Override DropRole method for Azure, the newOwner can be an invalid format for postgres * Refactor
movetokube · Jun 1, 2021 · 9cad050 · 9cad050
1 parent f2d8b5b
commit 9cad050
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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)
+}