AlterDatabaseToMultiByteCharset.php

Go to the documentation of this file.

<?php

namespace Elgg\Upgrades;

use Elgg\Upgrade\AsynchronousUpgrade;
use Elgg\Upgrade\Result;

class AlterDatabaseToMultiByteCharset implements AsynchronousUpgrade {

    private $utf8mb4_tables = [
        // InnoDB
        'access_collection_membership',
        'access_collections',
        'annotations',
        'api_users',
        'config',
        'entities',
        'entity_relationships',
        'metadata',
        'private_settings',
        'queue',
        'river',
        'system_log',
        'users_remember_me_cookies',
        'users_sessions',
        // MEMORY
        'hmac_cache',
        'users_apisessions',
    ];

    // Columns with utf8 encoding and utf8_general_ci collation
    // $table => [
    //   $column => $index
    // ]

    private $non_mb4_columns = [
        'config' => [
            'name' => [
                'primary' => true,
                'name' => 'name',
                'unique' => false,
            ],
        ],
        'entities' => [
            'subtype' => [
                'primary' => false,
                'name' => 'subtype',
                'unique' => false,
            ],
        ],
        'queue' => [
            'name' => [
                'primary' => false,
                'name' => "name",
                'unique' => false,
            ],
        ],
        'users_sessions' => [
            'session' => [
                'primary' => true,
                'name' => 'session',
                'unique' => false,
            ],
        ],
        'hmac_cache' => [
            'hmac' => [
                'primary' => true,
                'name' => 'hmac',
                'unique' => false,
            ],
        ],
        'system_log' => [
            'object_class' => [
                'primary' => false,
                'name' => 'object_class',
                'unique' => false,
            ],
            'object_type' => [
                'primary' => false,
                'name' => 'object_type',
                'unique' => false,
            ],
            'object_subtype' => [
                'primary' => false,
                'name' => 'object_subtype',
                'unique' => false,
            ],
            'event' => [
                'primary' => false,
                'name' => 'event',
                'unique' => false,
            ],
            'river_key' => [
                'primary' => false,
                'name' => 'river_key',
                'unique' => false,
                'columns' => ['object_type', 'object_subtype', 'event']
            ],
        ]
    ];

    public function getVersion() {
        return 2017080900;
    }

    public function needsIncrementOffset() {
        return false;
    }

    public function shouldBeSkipped() {

        $config = _elgg_services()->dbConfig->getConnectionConfig();
        $rows = elgg()->db->getData("SHOW TABLE STATUS FROM `{$config['database']}`");

        $prefixed_table_names = array_map(function ($t) use ($config) {
            return "{$config['prefix']}{$t}";
        }, $this->utf8mb4_tables);

        foreach ($rows as $row) {
            if (in_array($row->Name, $prefixed_table_names) && $row->Collation !== 'utf8mb4_general_ci') {
                return false;
            }
        }

        return true;
    }

    public function countItems() {
        return 1;
    }

    public function run(Result $result, $offset) {

        $config = _elgg_services()->dbConfig->getConnectionConfig();

        try {
            // check if we need to change a global variable
            $row = elgg()->db->getDataRow("SHOW GLOBAL VARIABLES LIKE 'innodb_large_prefix'");
            
            if (empty($row) || $row->Value === 'OFF') {
                // required to allow bigger index sizes required for utf8mb4
                elgg()->db->updateData("SET GLOBAL innodb_large_prefix = 'ON'");
            }
        } catch (\Exception $e) {
            // something went wrong, maybe database permissions, or version
            $result->addFailures();
            $result->addError("Failure to set 'innodb_large_prefix'. Ask your database administrator for more information.");
            $result->addError("Alternatively ask the database administrator to (temporarily) set 'innodb_large_prefix' to 'ON'.");
            $result->addError($e->getMessage());
            
            return $result;
        }
        
        try {
            // alter table structure
            elgg()->db->updateData("
                ALTER DATABASE
                `{$config['database']}`
                CHARACTER SET = utf8mb4
                COLLATE = utf8mb4_unicode_ci
            ");

            foreach ($this->utf8mb4_tables as $table) {
                if (!empty($this->non_mb4_columns[$table])) {
                    foreach ($this->non_mb4_columns[$table] as $column => $index) {
                        if ($index) {
                            if ($index['primary']) {
                                elgg()->db->updateData("
                                    ALTER TABLE {$config['prefix']}{$table}
                                    DROP PRIMARY KEY
                                ");
                            } else {
                                elgg()->db->updateData("
                                    ALTER TABLE {$config['prefix']}{$table}
                                    DROP KEY {$index['name']}
                                ");
                            }
                        }
                    }
                }

                elgg()->db->updateData("
                    ALTER TABLE {$config['prefix']}{$table}
                    ROW_FORMAT=DYNAMIC
                ");

                elgg()->db->updateData("
                    ALTER TABLE {$config['prefix']}{$table}
                    CONVERT TO CHARACTER SET utf8mb4
                    COLLATE utf8mb4_general_ci
                ");

                if (!empty($this->non_mb4_columns[$table])) {
                    foreach ($this->non_mb4_columns[$table] as $column => $index) {
                        if (empty($index['columns'])) {
                            // Alter table only if the key is not composite
                            elgg()->db->updateData("
                                ALTER TABLE {$config['prefix']}{$table}
                                MODIFY $column VARCHAR(255)
                                CHARACTER SET utf8
                                COLLATE utf8_unicode_ci
                            ");
                        }

                        if (!$index) {
                            continue;
                        }

                        $sql = "ADD";
                        if ($index['unique']) {
                            $sql .= " UNIQUE ({$index['name']})";
                        } else if ($index['primary']) {
                            $sql .= " PRIMARY KEY ({$index['name']})";
                        } else {
                            $key_columns = elgg_extract('columns', $index, [$column]);
                            $key_columns = implode(',', $key_columns);
                            $sql .= " KEY {$index['name']} ($key_columns)";
                        }

                        elgg()->db->updateData("
                            ALTER TABLE {$config['prefix']}{$table}
                            $sql
                        ");
                    }
                }
            }
        } catch (\Exception $e) {
            $result->addFailures();
            $result->addError($e->getMessage());
            
            return $result;
        }

        $result->addSuccesses();

        return $result;
    }
}