MapField.php 8.37 KB
Newer Older
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
<?php

// Protocol Buffers - Google's data interchange format
// Copyright 2008 Google Inc.  All rights reserved.
// https://developers.google.com/protocol-buffers/
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//     * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
//     * Neither the name of Google Inc. nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

/**
 * MapField and MapFieldIter are used by generated protocol message classes to
 * manipulate map fields.
 */

namespace Google\Protobuf\Internal;

/**
 * MapField is used by generated protocol message classes to manipulate map
 * fields. It can be used like native PHP array.
 */
class MapField implements \ArrayAccess, \IteratorAggregate, \Countable
{
    /**
     * @ignore
     */
    private $container;
    /**
     * @ignore
     */
    private $key_type;
    /**
     * @ignore
     */
    private $value_type;
    /**
     * @ignore
     */
61 62 63 64 65
    private $klass;
    /**
     * @ignore
     */
    private $legacy_klass;
66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81

    /**
     * Constructs an instance of MapField.
     *
     * @param long $key_type Type of the stored key element.
     * @param long $value_type Type of the stored value element.
     * @param string $klass Message/Enum class name of value instance
     * (message/enum fields only).
     * @ignore
     */
    public function __construct($key_type, $value_type, $klass = null)
    {
        $this->container = [];
        $this->key_type = $key_type;
        $this->value_type = $value_type;
        $this->klass = $klass;
82 83 84 85 86 87 88 89 90 91 92

        if ($this->value_type == GPBType::MESSAGE) {
            $pool = DescriptorPool::getGeneratedPool();
            $desc = $pool->getDescriptorByClassName($klass);
            if ($desc == NULL) {
                new $klass;  // No msg class instance has been created before.
                $desc = $pool->getDescriptorByClassName($klass);
            }
            $this->klass = $desc->getClass();
            $this->legacy_klass = $desc->getLegacyClass();
        }
93 94
    }

95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
    /**
     * @ignore
     */
    public function getKeyType()
    {
        return $this->key_type;
    }

    /**
     * @ignore
     */
    public function getValueType()
    {
        return $this->value_type;
    }

    /**
     * @ignore
     */
    public function getValueClass()
    {
        return $this->klass;
    }

119 120 121 122 123 124 125 126
    /**
     * @ignore
     */
    public function getLegacyValueClass()
    {
        return $this->legacy_klass;
    }

127 128 129 130 131 132 133
    /**
     * Return the element at the given key.
     *
     * This will also be called for: $ele = $arr[$key]
     *
     * @param object $key The key of the element to be fetched.
     * @return object The stored element at given key.
134 135
     * @throws \ErrorException Invalid type for index.
     * @throws \ErrorException Non-existing index.
136 137 138 139 140 141 142 143 144 145 146 147 148 149
     */
    public function offsetGet($key)
    {
        return $this->container[$key];
    }

    /**
     * Assign the element at the given key.
     *
     * This will also be called for: $arr[$key] = $value
     *
     * @param object $key The key of the element to be fetched.
     * @param object $value The element to be assigned.
     * @return void
150 151 152
     * @throws \ErrorException Invalid type for key.
     * @throws \ErrorException Invalid type for value.
     * @throws \ErrorException Non-existing key.
153 154 155
     */
    public function offsetSet($key, $value)
    {
156
        $this->checkKey($this->key_type, $key);
157 158

        switch ($this->value_type) {
Bo Yang's avatar
Bo Yang committed
159 160
            case GPBType::SFIXED32:
            case GPBType::SINT32:
161
            case GPBType::INT32:
Bo Yang's avatar
Bo Yang committed
162
            case GPBType::ENUM:
163 164
                GPBUtil::checkInt32($value);
                break;
Bo Yang's avatar
Bo Yang committed
165
            case GPBType::FIXED32:
166 167 168
            case GPBType::UINT32:
                GPBUtil::checkUint32($value);
                break;
Bo Yang's avatar
Bo Yang committed
169 170
            case GPBType::SFIXED64:
            case GPBType::SINT64:
171 172 173
            case GPBType::INT64:
                GPBUtil::checkInt64($value);
                break;
Bo Yang's avatar
Bo Yang committed
174
            case GPBType::FIXED64:
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
            case GPBType::UINT64:
                GPBUtil::checkUint64($value);
                break;
            case GPBType::FLOAT:
                GPBUtil::checkFloat($value);
                break;
            case GPBType::DOUBLE:
                GPBUtil::checkDouble($value);
                break;
            case GPBType::BOOL:
                GPBUtil::checkBool($value);
                break;
            case GPBType::STRING:
                GPBUtil::checkString($value, true);
                break;
            case GPBType::MESSAGE:
191 192 193
                if (is_null($value)) {
                  trigger_error("Map element cannot be null.", E_USER_ERROR);
                }
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
                GPBUtil::checkMessage($value, $this->klass);
                break;
            default:
                break;
        }

        $this->container[$key] = $value;
    }

    /**
     * Remove the element at the given key.
     *
     * This will also be called for: unset($arr)
     *
     * @param object $key The key of the element to be removed.
     * @return void
210
     * @throws \ErrorException Invalid type for key.
211 212 213
     */
    public function offsetUnset($key)
    {
214
        $this->checkKey($this->key_type, $key);
215 216 217 218 219 220 221 222 223 224
        unset($this->container[$key]);
    }

    /**
     * Check the existence of the element at the given key.
     *
     * This will also be called for: isset($arr)
     *
     * @param object $key The key of the element to be removed.
     * @return bool True if the element at the given key exists.
225
     * @throws \ErrorException Invalid type for key.
226 227 228
     */
    public function offsetExists($key)
    {
229
        $this->checkKey($this->key_type, $key);
230 231 232 233 234 235 236 237
        return isset($this->container[$key]);
    }

    /**
     * @ignore
     */
    public function getIterator()
    {
238
        return new MapFieldIter($this->container, $this->key_type);
239 240 241 242 243 244 245 246 247 248 249 250 251
    }

    /**
     * Return the number of stored elements.
     *
     * This will also be called for: count($arr)
     *
     * @return integer The number of stored elements.
     */
    public function count()
    {
        return count($this->container);
    }
252 253 254 255 256 257 258

    /**
     * @ignore
     */
    private function checkKey($key_type, &$key)
    {
        switch ($key_type) {
Bo Yang's avatar
Bo Yang committed
259 260
            case GPBType::SFIXED32:
            case GPBType::SINT32:
261 262 263
            case GPBType::INT32:
                GPBUtil::checkInt32($key);
                break;
Bo Yang's avatar
Bo Yang committed
264
            case GPBType::FIXED32:
265 266 267
            case GPBType::UINT32:
                GPBUtil::checkUint32($key);
                break;
Bo Yang's avatar
Bo Yang committed
268 269
            case GPBType::SFIXED64:
            case GPBType::SINT64:
270 271 272 273
            case GPBType::INT64:
                GPBUtil::checkInt64($key);
                break;
            case GPBType::FIXED64:
Bo Yang's avatar
Bo Yang committed
274
            case GPBType::UINT64:
275 276 277 278 279 280 281 282 283 284 285 286 287 288 289
                GPBUtil::checkUint64($key);
                break;
            case GPBType::BOOL:
                GPBUtil::checkBool($key);
                break;
            case GPBType::STRING:
                GPBUtil::checkString($key, true);
                break;
            default:
                trigger_error(
                    "Given type cannot be map key.",
                    E_USER_ERROR);
                break;
        }
    }
290
}