diff --git a/library/traits/model/Relation.php b/library/traits/model/Relation.php
index 3585c8ef..1372ab97 100644
--- a/library/traits/model/Relation.php
+++ b/library/traits/model/Relation.php
@@ -97,9 +97,10 @@ trait Relation
      * 对保存到数据库的数据进行处理
      * @access protected
      * @param mixed $data 要操作的数据
-     * @return boolean
+     * @param string $type insert 或者 update
+     * @return array
      */
-    protected function _write_data($data)
+    protected function _write_data($data, $type)
     {
         $this->_before_write($data);
         return $data;