fix: correct postgres check if user exits (#8)

anbraten · May 24, 2022 · fe643c8 · fe643c8
1 parent 4194459
commit fe643c8
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/adapters/postgres_adapter.go b/adapters/postgres_adapter.go
@@ -34,12 +34,23 @@ func (adapter postgresAdapter) DeleteDatabase(ctx context.Context, database stri
 }
 
 func (adapter postgresAdapter) HasDatabaseUserWithAccess(ctx context.Context, database string, username string) (bool, error) {
+	var count int
+	query := fmt.Sprintf("SELECT COUNT(*) FROM pg_roles WHERE rolname='%s';", username)
+	err := adapter.db.QueryRow(ctx, query).Scan(&count)
+	if err != nil {
+		return false, err
+	}
+	if count == 0 {
+		return false, nil
+	}
+
 	var hasPrivilege bool
-	query := fmt.Sprintf("SELECT has_database_privilege('%s', '%s', 'CONNECT');", username, database)
-	err := adapter.db.QueryRow(ctx, query).Scan(&hasPrivilege)
+	query = fmt.Sprintf("SELECT has_database_privilege('%s', '%s', 'CONNECT');", username, database)
+	err = adapter.db.QueryRow(ctx, query).Scan(&hasPrivilege)
 	if err != nil {
 		return false, err
 	}
+
 	return hasPrivilege, nil
 }