* Add sql check and modify zentao sql.

2023-06-20 10:16:26 +08:00
parent e76f50a3f0
commit fa3a7dc79e
8 changed files with 65 additions and 8 deletions
@@ -168,6 +168,37 @@ class dataviewModel extends model
        return $columnTypes;
    }

+    /**
+     * Check that the column of an sql query is unique.
+     *
+     * @param  string $sql
+     * @param  bool   $repeat
+     * @access public
+     * @return bool
+     */
+    public function checkUniColumn($sql, $repeat = false)
+    {
+        $columns = $this->dao->getColumns($sql);
+
+        $isUnique     = true;
+        $columnList   = array();
+        $repeatColumn = array();
+        foreach($columns as $column)
+        {
+            $field = $column['name'];
+            if(isset($columnTypes[$field]))
+            {
+                $isUnique = false;
+                $repeatColumn[$field] = $field;
+            }
+
+            $columnTypes[$field] = $field;
+        }
+
+        if($repeat) return array($isUnique, $repeatColumn);
+        return $isUnique;
+    }
+
    /**
     * Get type options
     *