Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\comment;
Chris@0: 
Chris@0: use Drupal\Core\Entity\ContentEntityTypeInterface;
Chris@0: use Drupal\Core\Entity\Sql\SqlContentEntityStorageSchema;
Chris@0: use Drupal\Core\Field\FieldStorageDefinitionInterface;
Chris@18: use Drupal\Core\Field\RequiredFieldStorageDefinitionInterface;
Chris@0: 
Chris@0: /**
Chris@0:  * Defines the comment schema handler.
Chris@0:  */
Chris@0: class CommentStorageSchema extends SqlContentEntityStorageSchema {
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function getEntitySchema(ContentEntityTypeInterface $entity_type, $reset = FALSE) {
Chris@0:     $schema = parent::getEntitySchema($entity_type, $reset);
Chris@0: 
Chris@17:     if ($data_table = $this->storage->getDataTable()) {
Chris@17:       $schema[$data_table]['indexes'] += [
Chris@17:         'comment__status_pid' => ['pid', 'status'],
Chris@17:         'comment__num_new' => [
Chris@17:           'entity_id',
Chris@17:           'entity_type',
Chris@17:           'comment_type',
Chris@17:           'status',
Chris@17:           'created',
Chris@17:           'cid',
Chris@17:           'thread',
Chris@17:         ],
Chris@17:         'comment__entity_langcode' => [
Chris@17:           'entity_id',
Chris@17:           'entity_type',
Chris@17:           'comment_type',
Chris@17:           'default_langcode',
Chris@17:         ],
Chris@17:       ];
Chris@17:     }
Chris@0: 
Chris@0:     return $schema;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function getSharedTableFieldSchema(FieldStorageDefinitionInterface $storage_definition, $table_name, array $column_mapping) {
Chris@0:     $schema = parent::getSharedTableFieldSchema($storage_definition, $table_name, $column_mapping);
Chris@0:     $field_name = $storage_definition->getName();
Chris@0: 
Chris@0:     if ($table_name == 'comment_field_data') {
Chris@0:       // Remove unneeded indexes.
Chris@0:       unset($schema['indexes']['comment_field__pid__target_id']);
Chris@0:       unset($schema['indexes']['comment_field__entity_id__target_id']);
Chris@0: 
Chris@0:       switch ($field_name) {
Chris@0:         case 'thread':
Chris@0:           // Improves the performance of the comment__num_new index defined
Chris@0:           // in getEntitySchema().
Chris@0:           $schema['fields'][$field_name]['not null'] = TRUE;
Chris@0:           break;
Chris@0: 
Chris@18:         case 'entity_type':
Chris@18:         case 'field_name':
Chris@18:           assert($storage_definition instanceof RequiredFieldStorageDefinitionInterface);
Chris@18:           if ($storage_definition->isStorageRequired()) {
Chris@18:             // The 'entity_type' and 'field_name' are required so they also need
Chris@18:             // to be marked as NOT NULL.
Chris@18:             $schema['fields'][$field_name]['not null'] = TRUE;
Chris@18:           }
Chris@18:           break;
Chris@18: 
Chris@0:         case 'created':
Chris@0:           $this->addSharedTableFieldIndex($storage_definition, $schema, TRUE);
Chris@0:           break;
Chris@0: 
Chris@0:         case 'uid':
Chris@0:           $this->addSharedTableFieldForeignKey($storage_definition, $schema, 'users', 'uid');
Chris@0:       }
Chris@0:     }
Chris@0: 
Chris@0:     return $schema;
Chris@0:   }
Chris@0: 
Chris@0: }