单元测试 分表处理

src/Common/Entity/Split.php

@@ -8,6 +8,7 @@
 namespace Hdll\Services\Common\Entity;
 namespace Hdll\Services\Common\Entity;
 
 
 use Hdll\Services\Common\Bean\Collector\SplitCollector;
 use Hdll\Services\Common\Bean\Collector\SplitCollector;
+use Hdll\Services\Common\Enum\TestEnum;
 use Hdll\Services\Common\Exception\CommonException;
 use Hdll\Services\Common\Exception\CommonException;
 use Swoft\Core\RequestContext;
 use Swoft\Core\RequestContext;
 use Swoft\Db\Bean\Collector\EntityCollector;
 use Swoft\Db\Bean\Collector\EntityCollector;
@@ -23,32 +24,36 @@ class Split
             $map = array_flip($collector['map']);
             $map = array_flip($collector['map']);
         }
         }
 
 
-        //如果是实体且为分表实体
-        if (strpos($tableName, '\\') !== false && isset($collector[$tableName])) {
-            $className = $tableName;
-            $tableName = $collector[$tableName]['table'];
+        if ( ! self::isTesting() ) {
+            //如果是实体且为分表实体
+            if (strpos($tableName, '\\') !== false && isset($collector[$tableName])) {
+                $className = $tableName;
+                $tableName = $collector[$tableName]['table'];
 
 
-            $entityCollector = EntityCollector::getCollector();
+                $entityCollector = EntityCollector::getCollector();
 
 
-            //设置分表对应的className
-            EntityCollector::setCollector($tableName.'_'.self::getDbNum($className), $tableName);
+                //设置分表对应的className
+                EntityCollector::setCollector($tableName.'_'.self::getDbNum($className), $tableName);
 
 
-            //获取分表名
-            $tableName = $tableName.'_'.self::getDbNum($className);
+                //获取分表名
+                $tableName = $tableName.'_'.self::getDbNum($className);
 
 
-            //设置分表类型
-            $query->selectInstance($entityCollector[$className]['instance']);
+                //设置分表类型
+                $query->selectInstance($entityCollector[$className]['instance']);
 
 
 
 
-        } elseif ( isset($map[$tableName]) ) { //非实体且分表
-            $className = $map[$tableName];
-            $defaultTbleName = $tableName;
-            $tableName = $tableName.'_'.self::getDbNum($className);
-            //设置分表对应的className
-            EntityCollector::setCollector($tableName, $defaultTbleName);
+            } elseif ( isset($map[$tableName]) ) { //非实体且分表
+                $className = $map[$tableName];
+                $defaultTbleName = $tableName;
+                $tableName = $tableName.'_'.self::getDbNum($className);
+                //设置分表对应的className
+                EntityCollector::setCollector($tableName, $defaultTbleName);
+
+            }
 
 
         }
         }
 
 
+
         return $tableName;
         return $tableName;
     }
     }
 
 
@@ -63,4 +68,12 @@ class Split
         }
         }
         return $storeId % $splitNum;
         return $storeId % $splitNum;
     }
     }
+
+    private function isTesting()
+    {
+        $data = RequestContext::getContextData();
+        $storeId = $data['userInfo']['storeId'];
+
+        return intval($storeId) === TestEnum::TEST_STORE_ID;
+    }
 }
 }
\ No newline at end of file

src/Common/Enum/TestEnum.php

+<?php
+/**
+ * Created by PhpStorm.
+ * User: Administrator
+ * Date: 2018/9/25
+ * Time: 13:22
+ */
+namespace Hdll\Services\Common\Enum;
+
+class TestEnum
+{
+    const TEST_STORE_ID = -3;
+
+}
+