function system_update_8901

Update the stored schema data for entity identifier fields.

File

core/modules/system/system.install, line 1550

Code

function system_update_8901() {
    $definition_update_manager = \Drupal::entityDefinitionUpdateManager();
    $entity_type_manager = \Drupal::entityTypeManager();
    $installed_storage_schema = \Drupal::keyValue('entity.storage_schema.sql');
    foreach ($definition_update_manager->getEntityTypes() as $entity_type_id => $entity_type) {
        // Ensure that we are dealing with a non-deleted entity type that uses the
        // default SQL storage.
        if (!$entity_type_manager->hasDefinition($entity_type_id)) {
            continue;
        }
        $storage = $entity_type_manager->getStorage($entity_type_id);
        if (!$storage instanceof SqlContentEntityStorage) {
            continue;
        }
        foreach ([
            'id',
            'revision',
        ] as $key) {
            if (!$entity_type->hasKey($key)) {
                continue;
            }
            $field_name = $entity_type->getKey($key);
            $field_storage_definition = $definition_update_manager->getFieldStorageDefinition($field_name, $entity_type_id);
            if (!$field_storage_definition) {
                continue;
            }
            if ($field_storage_definition->getType() !== 'integer') {
                continue;
            }
            // Retrieve the storage schema in order to use its own method for updating
            // the identifier schema - ::processIdentifierSchema(). This is needed
            // because some storage schemas might not use serial identifiers.
            // @see \Drupal\user\UserStorageSchema::processIdentifierSchema()
            $ref_get_storage_schema = new \ReflectionMethod($storage, 'getStorageSchema');
            $ref_get_storage_schema->setAccessible(TRUE);
            $storage_schema = $ref_get_storage_schema->invoke($storage);
            if ($storage_schema instanceof SqlContentEntityStorageSchema) {
                $field_schema_data = $installed_storage_schema->get($entity_type_id . '.field_schema_data.' . $field_storage_definition->getName(), []);
                $table = $key === 'id' ? $entity_type->getBaseTable() : $entity_type->getRevisionTable();
                $ref_process_identifier_schema = new \ReflectionMethod($storage_schema, 'processIdentifierSchema');
                $ref_process_identifier_schema->setAccessible(TRUE);
                $ref_process_identifier_schema->invokeArgs($storage_schema, [
                    &$field_schema_data[$table],
                    $field_name,
                ]);
                $installed_storage_schema->set($entity_type_id . '.field_schema_data.' . $field_storage_definition->getName(), $field_schema_data);
            }
        }
    }
}