Retornar datos de una consulta y un ciclo anidado

[anacona16]

Hola expertos de la Web, verán me surge esta necesidad, tengo una función que consulta la tabla tb_menus y esta a la vez tiene sub-menus y para ello consulta la tabla tb_submenus las consultas son sencillas si hago todo el PHP y el HTML juntos pero quiero lo que quiero hacer es que la funcion retorne los datos en un arreglo para luego recorrerlos.

Código PHP:

function menus(){
$consultaMenus = $conn->query("SELECT * FROM tb_menus");
$datosMenus = $consultaMenus->fetch(PDO::FETCH_ASSOC);
 
while($menu = $consultaMenus){
$consultaSubMenus = $conn->query("SELECT * FROM tb_submenus WHERE codimenu = '".$datosMenus['codimenu']."'");
}
return array("menus" => $consultaMenus->fetch(PDO::FETCH_ASSOC), "submenus" => $consultaSubMenus->fetch(PDO::FETCH_ASSOC));
}
 

Ahora, al recorrer los arreglos retornados por la consulta, es donde se presenta el problema, en algunos casos se imprimen todos los menus, pero no los submenus y en otros casos pasa lo contrario, ya he intentado de todo pero no he logrado solucionarlo por eso acudo a ustedes.

Espero me hayan entendido y me puedan ayudar.