edit

lib/Relation.php

@@ -7,6 +7,14 @@ use Dever;
 
 class Relation
 {
+    private $table = 'invite/relation';
+    public function __construct()
+    {
+        if (Dever::config('base')->invite) {
+            $this->table = Dever::config('base')->invite;
+        }
+    }
+
 	# 只记录6级关系
 	private $total = 100;
 
@@ -37,9 +45,9 @@ class Relation
     # 更换上级 效率较高
     public function replaceParent($uid, $old_parent, $new_parent, $call = true)
     {
-        $info = Dever::db('invite/relation')->one(array('to_uid' => $uid, 'uid' => $old_parent));
+        $info = Dever::db(Dever::config('base')->invite)->one(array('to_uid' => $uid, 'uid' => $old_parent));
         if ($info) {
-            $state = Dever::db('invite/relation')->update(array('where_id' => $info['id'], 'uid' => $new_parent));
+            $state = Dever::db(Dever::config('base')->invite)->update(array('where_id' => $info['id'], 'uid' => $new_parent));
             if ($state && $call) {
                 $child = $this->getChild($uid, false, false);
                 if ($child) {
@@ -55,7 +63,7 @@ class Relation
     # 重置上级 效率较低
     public function resetParent($uid, $parent)
     {
-        Dever::db('invite/relation')->delete(array('to_uid' => $uid));
+        Dever::db(Dever::config('base')->invite)->delete(array('to_uid' => $uid));
         $this->set($parent, $uid);
 
         $child = $this->getChild($uid, 1, false);
@@ -68,7 +76,7 @@ class Relation
                 if ($child_parent) {
                     foreach ($child_parent as $k1 => $v1) {
                         if ($v1['level'] > $v['level']) {
-                            Dever::db('invite/relation')->update(array('where_id' => $v['id'], 'uid' => $parent));
+                            Dever::db(Dever::config('base')->invite)->update(array('where_id' => $v['id'], 'uid' => $parent));
                         }
                     }
                 }*/
@@ -80,13 +88,13 @@ class Relation
     # 清理邀请关系
     public function dropParent($uid, $parent)
     {
-        return Dever::db('invite/relation')->delete(array('to_uid' => $uid, 'uid' => $parent));
+        return Dever::db(Dever::config('base')->invite)->delete(array('to_uid' => $uid, 'uid' => $parent));
     }
 
     # 获取某个用户的上级数据
     public function getParent($uid, $level = 1)
     {
-        return Dever::db('invite/relation')->one(array('to_uid' => $uid, 'level' => $level));
+        return Dever::db(Dever::config('base')->invite)->one(array('to_uid' => $uid, 'level' => $level));
     }
 
     # 获取某个用户的所有上级数据
@@ -97,9 +105,9 @@ class Relation
             $where['level'] = $level;
         }
         if ($order == 1) {
-            return Dever::db('invite/relation')->getParent($where);
+            return Dever::db(Dever::config('base')->invite)->getParent($where);
         } else {
-            return Dever::db('invite/relation')->getParentLevel($where);
+            return Dever::db(Dever::config('base')->invite)->getParentLevel($where);
         }
     }
 
@@ -111,9 +119,9 @@ class Relation
             $where['level'] = $level;
         }
         if ($page) {
-            return Dever::db('invite/relation')->getChildByPage($where);
+            return Dever::db(Dever::config('base')->invite)->getChildByPage($where);
         } else {
-            return Dever::db('invite/relation')->getChild($where);
+            return Dever::db(Dever::config('base')->invite)->getChild($where);
         }
         
     }
@@ -152,7 +160,7 @@ class Relation
             $method = 'getChild';
         }
         
-        return Dever::db('invite/relation')->$method($where);
+        return Dever::db(Dever::config('base')->invite)->$method($where);
     }
 
     # 插入数据
@@ -163,10 +171,10 @@ class Relation
         $data['level'] = $level;
         $data['clear'] = true;
 
-        $info = Dever::db('invite/relation')->one($data);
+        $info = Dever::db(Dever::config('base')->invite)->one($data);
 
         if (!$info) {
-            return Dever::db('invite/relation')->insert($data);
+            return Dever::db(Dever::config('base')->invite)->insert($data);
         }
         
         return false;