mysqli заявка връща само първия ред

Използвайте този код:

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

Използвате връщането вътре в while и return прекратява цикъла while след първата итерация, поради което получавате само един ред.