Add an entitySave() method to DataProviderEntity.

RESTful-Drupal · Apr 27, 2018 · faaea06 · faaea06
1 parent a148298
commit faaea06
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/src/Plugin/resource/DataProvider/DataProviderEntity.php b/src/Plugin/resource/DataProvider/DataProviderEntity.php
@@ -346,6 +346,13 @@ public function canonicalPath($path) {
    */
   public function entityPreSave(\EntityDrupalWrapper $wrapper) {}
 
+  /**
+   * {@inheritdoc}
+   */
+  public function entitySave(\EntityDrupalWrapper $wrapper) {
+    $wrapper->save();
+  }
+
   /**
    * {@inheritdoc}
    */
@@ -1057,8 +1064,7 @@ protected function setPropertyValues(\EntityDrupalWrapper $wrapper, $object, $re
     $this->entityPreSave($interpreter->getWrapper());
 
     $this->entityValidate($interpreter->getWrapper());
-
-    $wrapper->save();
+    $this->entitySave($interpreter->getWrapper());
   }
 
   /**

diff --git a/src/Plugin/resource/DataProvider/DataProviderEntityInterface.php b/src/Plugin/resource/DataProvider/DataProviderEntityInterface.php
@@ -19,6 +19,16 @@ interface DataProviderEntityInterface extends DataProviderInterface {
    */
   public function entityPreSave(\EntityDrupalWrapper $wrapper);
 
+  /**
+   * Save an entity.
+   *
+   * @param \EntityDrupalWrapper $wrapper
+   *   The wrapped entity.
+   *
+   * @throws BadRequestException
+   */
+  public function entitySave(\EntityDrupalWrapper $wrapper);
+
   /**
    * Validate an entity before it is saved.
    *