Utilisez ce code :
$rows = array();
if ($type == 'assoc') {
while($row = $result->fetch_assoc()) {
$rows[] = $row;
}
} else {
while($row = $result->fetch_object()) {
$rows[] = $row;
}
}
return $rows;
Vous utilisez le retour à l'intérieur du while et return termine la boucle while après la première itération, c'est pourquoi vous n'obtenez qu'une seule ligne.