视图模型支持关联数据写入

Author: liu21st

src/Entity.php

@@ -308,12 +308,18 @@ public static function __callStatic($method, $args)
             $db = $entity->model()->db();
         }
 
+        if ('with' != $method && !empty(self::$weakMap[$entity]['autoMapping'])) {
+            // 自动关联查询
+            $db->with(self::$weakMap[$entity]['autoMapping']);
+        }
+
         return call_user_func_array([$db, $method], $args);
     }
 
     public function __call($method, $args)
     {
         // 调用Model类方法
-        return call_user_func_array([$this->model(), $method], $args);
+        $result = call_user_func_array([$this->model(), $method], $args);
+        return $result instanceof Model ? $this : $result;
     }
 }

src/model/View.php

@@ -106,7 +106,11 @@ private function fetchViewAttr(string $field)
             $value     = null;
             foreach ($relations as $relation) {
                 if ($model->$relation->hasData($field)) {
-                    $value = $model->$relation->$field;
+                    $value   = $model->$relation->$field;
+                    $mapping = $this->getOption('autoMappingAlias', []);
+
+                    $mapping[$field] = $relation . '->' . $field;
+                    $this->setOption('autoMappingAlias', $mapping);
                     break;
                 }
             }
@@ -295,12 +299,16 @@ protected function convertData(): array
     {
         // 获取实体属性
         $properties = $this->getEntityProperties();
+        $mapping    = $this->getOption('autoMappingAlias', []);
         $data       = $this->getData();
         $item       = [];
         foreach ($properties as $key => $field) {
             if (strpos($field, '->')) {
                 [$relation, $field]      = explode('->', $field);
                 $item[$relation][$field] = $data[$key];
+            } elseif (is_int($key) && isset($mapping[$field])) {
+                [$relation] = explode('->', $mapping[$field]);
+                $this->model()->$relation->$field = $data[$field];
             } else {
                 $item[$field] =  $data[is_int($key) ? $field : $key];
             }