association-1.0.0-alpha2/src/Entity/Storage/AssociationStorage.php

<?php

namespace Drupal\association\Entity\Storage;

use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Entity\Sql\SqlContentEntityStorage;

/**
 * Defines the storage handler class for association entities.
 *
 * This extends the base storage class, adding the ability to generate the
 * complementary association pages when the type dictates that it has pages.
 *
 * @ingroup association
 */
class AssociationStorage extends SqlContentEntityStorage {

  /**
   * {@inheritdoc}
   */
  protected function doSave($id, EntityInterface $entity) {
    /** @var \Drupal\association\Entity\AssociationInterface $entity */

    // The "new" flag is removed after the save is completed.
    $isNew = $entity->isNew();
    $status = parent::doSave($id, $entity);

    // Update the landing page based on the association updates. Needs to be
    // called after the save to ensure association has an ID on creation.
    $method = $isNew ? 'onCreate' : 'onUpdate';
    $entity->getLandingPageHandler()->{$method}($entity);

    return $status;
  }

}