ListModel.php 1.5 KB
Newer Older
lzc828's avatar
lzc828 已提交
1 2
<?php

lzc828's avatar
lzc828 已提交
3
namespace One\Database\Pgsql;
lzc828's avatar
lzc828 已提交
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79

class ListModel implements \Iterator, \JsonSerializable
{
    private $index = 0;
    private $data = [];

    public function pluck($column, $is_key = false, $unique = false)
    {
        if ($is_key) {
            $r = [];
            if ($unique) {
                foreach ($this->data as $v) {
                    $r[$v[$column]][] = $v;
                }
            } else {
                foreach ($this->data as $v) {
                    $r[$v[$column]] = $v;
                }
            }
            return $r;
        } else {
            $r = [];
            foreach ($this->data as $v) {
                $r[] = $v[$column];
            }
            return $r;
        }
    }

    public function jsonSerialize()
    {
        return $this->data;
    }

    public function __construct($data)
    {
        $this->data = $data;
    }

    public function rewind()
    {
        $this->index = 0;
    }

    public function valid()
    {
        return $this->index < count($this->data);
    }

    public function current()
    {
        return $this->data[$this->index];
    }

    public function next()
    {
        return $this->index++;
    }

    public function key()
    {
        return $this->index;
    }

    public function toArray()
    {
        $r = [];
        foreach ($this->data as $val) {
            $r[] = $val->toArray();
            unset($val);
        }
        $this->data = [];
        return $r;
    }

}