ループ内でブレークコントロールっぽい書き方 (PHP)

Question

たとえば、下記ようなレコードセットをデータベースから取って来て、
企業の一覧と企業に紐づく社員を２名まで表示するような画面を表示したいと思います

社員ID(person_id), 企業ID(company_id), 社員名(person_name), 企業名(company_name),
1, 2, 田中太郎, 山田商事
2, 2, ....
3, 2, ....
4, 2, ....
5, 3, ....
6, 3, ....
7, 4, ....
8, 5, ....
9, 5, ....
10,5, ....



そんなとき自分はPHPで下記のようなコードをよく書いたりするのですが、

$data = [];  // 最終的にテンプレートに渡すデータソース
$count = 0;
$old_company_id = $old_company_name = $person1 = $person2 = null;
while ( $row = $recordset->fetch() ) {
    if (!is_null($old_company_id) && $old_company_id != $row['company_id'] ) {
        $data[$old_company_id]['person1'] = $person1;
        $data[$old_company_id]['person2'] = $person2;
        $data[$old_company_id]['company_name'] = $old_company_name;
        $old_company_name = $person1 = $person2 = null;
        $count = 0;
     }

    if ($count == 0) {
        $person1 = $row['person_name'];
    } else if ( $count == 1 ) {
        $person2 = $row['person_name'];
    }

    $old_company_id = $row['company_id'];
    $old_company_name = $row['company_name'];
    $count++;
}
if (!is_null($person1)) {  // 残り
    $data[$old_company_id]['person1'] = $person1;
    $data[$old_company_id]['person2'] = $person2;
    $data[$old_company_id]['company_name'] = $old_company_name;
}

こういうのをたくさん書いてるのですが、もうちょっと簡潔な書き方はあるでしょうか

Answer 1

動作確認はしていないのでバグってるかもですが、
下記でどうでしょうか？

$data = array();  // 最終的にテンプレートに渡すデータソース
while ( $row = $recordset->fetch() ) {
	$company_id = $row['company_id'];

	if (isset($data[$company_id])) {
		// person2が無い場合設定
		if (!isset($data[$company_id]['person2'])) {
			$data[$company_id]['person2'] = $row['person_name'];
		}
	} else {
		// 初回だからperson1
		$data[$company_id]['person1'] = $row['person_name'];
		$data[$company_id]['company_name'] = $row['company_name'];
	}
}

Comments

[himakuma]

今後表示項目が増えるならhiimさんのやり方がいいかもですね。正直ＳＱＬで一発で取れると思いますが。。。

[redara]

なるほど。キーとなる値が変わったかどうかで判定しようとしてましたが、
$dataにキーが存在するかどうかで判定した方が記述が簡潔になりますね。
ありがとうございます。

[himakuma]

２項目ぐらいならランク関数使って、ＪＯＩＮとかの方がいいかもですね。ＳＱＬの戻りは遅くなりますが、一括全件持ってくるよりは良いかと思います。※条件によるスキャン２回するので。。

[redara]

そうですね。今回はこのような例になりましたが、よく同じような記述をしているので一般的な書き方という意味で質問しました。できるだけSQL側で解決できるようにしたいと思います

Answer 2

sql自体を工夫すればもう少し楽に書けるとは思いますが、あくまでsqlは全てとってきているという前提で

$data = array();// 初期化

while ( $row = $recordset->fetch() ) {

    // 会社名を登録
	if(!isset($data[$row['company_id']]['companyname']))
		$data[$row['company_id']]['companyname'] = $row['company_name'];

    // 人の名前登録
	if(!isset($data[$row['company_id']]['person1']))
		$data[$row['company_id']]['person1'] = $row['person_name'];
	elseif(!isset($data[$row['company_id']]['person2']))
		$data[$row['company_id']]['person2'] = $row['person_name'];
	}
}

でどうでしょうか

Comments

[redara]

$data側に値が入ってるかどうかで判定するのが良さそうですね。
今回の例ではSQLで取ってくる方法もあると思いますが、SQLで取ってこれないようなプログラム側で解決したいケースもあるので参考になります。