MetastringsTable.php

Go to the documentation of this file.

<?php

namespace Elgg\Database;

use Elgg\Cache\Pool;
use Elgg\Database;

class MetastringsTable {

    protected $cache;

    protected $db;

    public function __construct(Pool $cache, Database $db) {
        $this->cache = $cache;
        $this->db = $db;
    }

    function getId($string, $case_sensitive = true) {
        if ($case_sensitive) {
            return $this->getIdCaseSensitive($string);
        } else {
            return $this->getIdCaseInsensitive($string);
        }
    }

    function getMap(array $string_keys) {
        if (!$string_keys) {
            return [];
        }
        if (count($string_keys) === 1) {
            $key = reset($string_keys);
            return [$key => $this->getIdCaseSensitive($key)];
        }

        $missing = array_fill_keys($string_keys, true);

        $set_element = array_map(function ($string) {
            return "BINARY '" . $this->db->sanitizeString($string) . "'";
        }, $string_keys);

        $set = implode(',', $set_element);

        $query = "SELECT * FROM {$this->getTableName()} WHERE string IN ($set)";
        $ret = [];

        foreach ($this->db->getData($query) as $row) {
            $ret[$row->string] = (int) $row->id;
            unset($missing[$row->string]);
        }
        foreach (array_keys($missing) as $string) {
            $ret[$string] = $this->getIdCaseSensitive($string);
        }

        return $ret;
    }

    private function getIdCaseSensitive($string) {
        $string = (string) $string;
        return $this->cache->get($string, function() use ($string) {

            $memcache = _elgg_get_memcache('metastrings_memcache');

            // Stash can't handle arbitrary keys
            $result = $memcache->load(md5($string));
            if ($result !== false) {
                return $result;
            }

            $query = "SELECT id FROM {$this->getTableName()} WHERE string = BINARY :string";
            $params = [
                ':string' => $string,
            ];

            $result = $this->db->getDataRow($query, null, $params);
            if ($result) {
                $id = $result->id;
            } else {
                $id = $this->add($string);
            }

            $memcache->save(md5($string), $id);

            return $id;
        });
    }

    private function getIdCaseInsensitive($string) {
        $string = (string) $string;
        // caching doesn't work for case insensitive requests
        $query = "SELECT id FROM {$this->getTableName()} WHERE string = :string";
        $params = [
            ':string' => $string,
        ];
        $results = $this->db->getData($query, null, $params);
        $ids = array();
        foreach ($results as $result) {
            $ids[] = $result->id;
        }
        if (empty($ids)) {
            $ids[] = $this->add($string);
        }
        return $ids;
    }

    public function add($string) {
        $sql = "INSERT INTO {$this->getTableName()} (string) VALUES (:string)";
        $params = [
            ':string' => trim((string) $string),
        ];
        return $this->db->insertData($sql, $params);
    }

    public function getTableName() {
        return $this->db->prefix . "metastrings";
    }

}