summaryrefslogtreecommitdiff
path: root/vendor/fguillot/picodb/lib/PicoDb/Hashtable.php
blob: 17afd0e6a69308c13126f67d09cea08e646f9e2c (plain)
1
2
3
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
<?php

namespace PicoDb;

use PDO;

/**
 * HashTable (key/value)
 *
 * @package PicoDb
 * @author  Frederic Guillot
 * @author  Mathias Kresin
 */
class Hashtable extends Table
{
    /**
     * Column for the key
     *
     * @access private
     * @var    string
     */
    private $keyColumn = 'key';

    /**
     * Column for the value
     *
     * @access private
     * @var    string
     */
    private $valueColumn = 'value';

    /**
     * Set the key column
     *
     * @access public
     * @param  string  $column
     * @return $this
     */
    public function columnKey($column)
    {
        $this->keyColumn = $column;
        return $this;
    }

    /**
     * Set the value column
     *
     * @access public
     * @param  string  $column
     * @return $this
     */
    public function columnValue($column)
    {
        $this->valueColumn = $column;
        return $this;
    }

    /**
     * Insert or update
     *
     * @access public
     * @param  array    $hashmap
     * @return boolean
     */
    public function put(array $hashmap)
    {
        return $this->db->getDriver()->upsert($this->getName(), $this->keyColumn, $this->valueColumn, $hashmap);
    }

    /**
     * Hashmap result [ [column1 => column2], [], ...]
     *
     * @access public
     * @return array
     */
    public function get()
    {
        $hashmap = array();

        // setup where condition
        if (func_num_args() > 0) {
            $this->in($this->keyColumn, func_get_args());
        }

        // setup to select columns in case that there are more than two
        $this->columns($this->keyColumn, $this->valueColumn);

        $rq = $this->db->execute($this->buildSelectQuery(), $this->conditionBuilder->getValues());
        $rows = $rq->fetchAll(PDO::FETCH_NUM);

        foreach ($rows as $row) {
            $hashmap[$row[0]] = $row[1];
        }

        return $hashmap;
    }

    /**
     * Shortcut method to get a hashmap result
     *
     * @access public
     * @param  string  $key    Key column
     * @param  string  $value  Value column
     * @return array
     */
    public function getAll($key, $value)
    {
        $this->keyColumn = $key;
        $this->valueColumn = $value;
        return $this->get();
    }
}