基于 SplPriorityQueue 实现的排序方法

[limingxinleo]

limingxinleo/hyperf-utils

前言

之前我们在进行数组排序时，经常使用 usort ksort 等方法，但当我们排序规则复杂一些后，这些方法使用起来就不是那么的方便了。

后来发现了 \SplPriorityQueue，进行简单的封装后，就更加方便使用了。

基础封装

Laminas\Stdlib\SplPriorityQueue 是对 \SplPriorityQueue 的封装，更加方便使用。

<?php

declare(strict_types=1);
/**
 * This file is part of Hyperf.
 *
 * @link     https://www.hyperf.io
 * @document https://hyperf.wiki
 * @contact  [email protected]
 * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
 */
namespace Han\Utils\Utils;

use Laminas\Stdlib\SplPriorityQueue;

class Sorter
{
    /**
     * @param array|\iterable $items
     */
    public function sort($items, callable $callable): SplPriorityQueue
    {
        $queue = new SplPriorityQueue();
        $serial = PHP_INT_MAX;
        foreach ($items as $item) {
            $priority = (array) $callable($item);
            $priority[] = $serial--;
            $queue->insert($item, $priority);
        }
        return $queue;
    }
}

上述代码，可以看到，我们额外增加了参数 $serial，这个字段会让 $priority 一致的元素可以按照优先插入元素的顺序排序。

代码测试

我们只需要调用 sort 方法，然后在后面的匿名函数中返回对应元素的权重即可。

<?php

$col = new Collection([
    new ModelStub(['id' => 1, 'message' => $id = uniqid()]),
    new ModelStub(['id' => 2, 'message' => $id2 = uniqid()]),
]);

$sorter = new Sorter();
$res = $sorter->sort($col, static function (ModelStub $model) {
    return $model->id;
});

$data = $res->toArray();
$this->assertSame(2, $data[0]->id);
$this->assertSame(1, $data[1]->id);
$this->assertSame($id2, $data[0]->message);
$this->assertSame($id, $data[1]->message);