From b2899975657ee40ba858e7c35351edc361307856 Mon Sep 17 00:00:00 2001
From: thinkphp <thinkphp@qq.com>
Date: Mon, 23 Oct 2017 12:05:11 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9BMorphTo=E5=85=B3=E8=81=94?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 library/think/model/relation/MorphTo.php | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/library/think/model/relation/MorphTo.php b/library/think/model/relation/MorphTo.php
index 9136d78e..e1a9db5b 100644
--- a/library/think/model/relation/MorphTo.php
+++ b/library/think/model/relation/MorphTo.php
@@ -238,17 +238,18 @@ class MorphTo extends Relation
     /**
      * 添加关联数据
      * @access public
-     * @param Model $model       关联模型对象
+     * @param Model     $model  关联模型对象
+     * @param string    $type   多态类型
      * @return Model
      */
-    public function associate($model)
+    public function associate($model, $type = '')
     {
         $morphKey  = $this->morphKey;
         $morphType = $this->morphType;
         $pk        = $model->getPk();
 
         $this->parent->setAttr($morphKey, $model->$pk);
-        $this->parent->setAttr($morphType, get_class($model));
+        $this->parent->setAttr($morphType, $type ?: get_class($model));
         $this->parent->save();
 
         return $this->parent->setRelation($this->relation, $model);