Make model serializable

Close #71
baopham · Aug 12, 2017 · e051c10 · e051c10
1 parent 37dabf1
commit e051c10
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 7 deletions.
diff --git a/src/DynamoDbModel.php b/src/DynamoDbModel.php
@@ -295,4 +295,26 @@ public function setDynamoDbIndexKeys($dynamoDbIndexKeys)
         $this->dynamoDbIndexKeys = $dynamoDbIndexKeys;
     }
 
+    /**
+     * Remove non-serializable properties when serializing.
+     *
+     * @return array
+     */
+    public function __sleep()
+    {
+        return array_keys(
+            array_except(get_object_vars($this), ['client', 'marshaler', 'attributeFilter'])
+        );
+    }
+
+    /**
+     * When a model is being unserialized, check if it needs to be booted and setup DynamoDB.
+     *
+     * @return void
+     */
+    public function __wakeup()
+    {
+        parent::__wakeup();
+        $this->setupDynamoDb();
+    }
 }
diff --git a/tests/DynamoDbModelTest.php b/tests/DynamoDbModelTest.php
@@ -2,6 +2,8 @@
 
 namespace BaoPham\DynamoDb\Tests;
 
+use Illuminate\Support\Facades\Cache;
+
 /**
  * Class DynamoDbModelTest
  *
@@ -384,17 +386,17 @@ public function testStaticMethods()
 
     public function testConditionContainingIndexKeyAndNonIndexKey()
     {
-        $fooItem = $this->seed([
-            'name' => ['S' => 'Foo'],
-            'count' => ['N' => 10],
-        ]);
-
-        $barItem = $this->seed([
+        $this->seed([
             'name' => ['S' => 'Bar'],
             'count' => ['N' => 11],
         ]);
 
-        $expectedItem = $this->testModel->unmarshalItem($fooItem);
+        $item = $this->seed([
+            'name' => ['S' => 'Foo'],
+            'count' => ['N' => 10],
+        ]);
+
+        $expectedItem = $this->testModel->unmarshalItem($item);
 
         $foundItems = $this->testModel
             ->where('count', 10)
@@ -405,6 +407,16 @@ public function testConditionContainingIndexKeyAndNonIndexKey()
         $this->assertEquals($expectedItem, $foundItems->first()->toArray());
     }
 
+    public function testSerialize()
+    {
+        $item = $this->testModel->unmarshalItem($this->seed());
+        $serializedItems = serialize($this->testModel->all());
+        $unserializedItems = unserialize($serializedItems);
+
+        $this->assertEquals([$item], $unserializedItems->toArray());
+        $this->assertInstanceOf(get_class($this->testModel), $unserializedItems->first());
+    }
+
     protected function seed($attributes = [])
     {
         $item = [