SubtypeTable.php

Go to the documentation of this file.

<?php
namespace Elgg\Database;

use Elgg\Database;

class SubtypeTable {

    protected $cache = null;

    protected $db;

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

    public function setCachedValues(array $values) {
        $this->cache = $values;
    }

    public function getId($type, $subtype) {
        if (!$subtype) {
            return false;
        }

        $obj = $this->retrieveFromCache($type, $subtype);

        return $obj ? $obj->id : false;
    }
    
    public function getSubtype($subtype_id) {
        if (!$subtype_id) {
            return '';
        }

        $cache = $this->getPopulatedCache();

        return isset($cache[$subtype_id]) ? $cache[$subtype_id]->subtype : false;
    }
    
    public function retrieveFromCache($type, $subtype) {
        foreach ($this->getPopulatedCache() as $obj) {
            if ($obj->type === $type && $obj->subtype === $subtype) {
                return $obj;
            }
        }

        return null;
    }

    public function getClass($type, $subtype) {
        $obj = $this->retrieveFromCache($type, $subtype);

        return $obj ? $obj->class : null;
    }

    public function getClassFromId($subtype_id) {
        if (!$subtype_id) {
            return null;
        }

        $cache = $this->getPopulatedCache();

        return isset($cache[$subtype_id]) ? $cache[$subtype_id]->class : null;
    }
    
    public function add($type, $subtype, $class = "") {
        if (!$subtype) {
            return 0;
        }
    
        $id = $this->getId($type, $subtype);
    
        if (!$id) {
            $sql = "
                INSERT INTO {$this->db->prefix}entity_subtypes
                    (type,  subtype,  class) VALUES
                    (:type, :subtype, :class)
            ";
            $params = [
                ':type' => $type,
                ':subtype' => $subtype,
                ':class' => $class,
            ];
            $id = $this->db->insertData($sql, $params);

            $this->invalidateCache();
        }
    
        return $id;
    }
    
    public function remove($type, $subtype) {
        $sql = "
            DELETE FROM {$this->db->prefix}entity_subtypes
            WHERE type = :type AND subtype = :subtype
        ";
        $params = [
            ':type' => $type,
            ':subtype' => $subtype,
        ];
        if (!$this->db->deleteData($sql, $params)) {
            return false;
        }

        $this->invalidateCache();
        
        return true;
    }
    
    public function update($type, $subtype, $class = '') {
        $id = $this->getId($type, $subtype);
        if (!$id) {
            return false;
        }
    
        $sql = "
            UPDATE {$this->db->prefix}entity_subtypes
            SET type = :type, subtype = :subtype, class = :class
            WHERE id = :id
        ";
        $params = [
            ':type' => $type,
            ':subtype' => $subtype,
            ':class' => $class,
            ':id' => $id,
        ];
        if (!$this->db->updateData($sql, false, $params)) {
            return false;
        }

        $this->invalidateCache();

        return true;
    }

    protected function invalidateCache() {
        $this->cache = null;
        _elgg_services()->boot->invalidateCache();
        _elgg_services()->entityCache->clear();
    }

    protected function getPopulatedCache() {
        if ($this->cache === null) {
            $rows = $this->db->getData("
                SELECT *
                FROM {$this->db->prefix}entity_subtypes
            ");

            $this->cache = [];
            foreach ($rows as $row) {
                $this->cache[$row->id] = $row;
            }
        }

        return $this->cache;
    }
}