query mysqli che restituisce solo la prima riga

Usa questo codice:

$rows = array();
if ($type == 'assoc') {
    while($row = $result->fetch_assoc()) {
      $rows[] = $row;
    }
} else {    
    while($row = $result->fetch_object()) {
      $rows[] = $row;
    }   
}
return $rows;

Stai usando return all'interno del while e return termina il ciclo while dopo la prima iterazione, ecco perché ottieni solo una riga.