2

如何更改我的 PDO 包装类,以便如果我期望查询的单行结果它使用 fetch(),如果它期望多个结果它使用 fetchAll()。

现在,如果我只有一个结果,我仍然必须遍历结果数组,这对我来说似乎很不切实际。

在模型中查询:

public function doccEdit() {

    $id = mysql_real_escape_string($_GET['id']);

    $this->result = $GLOBALS['db']->select("creditcards", "id = ?", $id);

    print_r($this->result);

}

在包装类中:

public function run($sql, $bind="") {
    $this->sql = trim($sql);
    $this->bind = $this->cleanup($bind);
    $this->error = "";

    try {
        $pdostmt = $this->prepare($this->sql);
        if($pdostmt->execute($this->bind) !== false) {
            if(preg_match("/^(" . implode("|", array("select", "describe", "pragma")) . ") /i", $this->sql))
                return $pdostmt->fetchall(PDO::FETCH_OBJ);
            elseif(preg_match("/^(" . implode("|", array("delete", "insert", "update")) . ") /i", $this->sql))
                return $pdostmt->rowCount();
        }   
    } catch (PDOException $e) {
        $this->error = $e->getMessage();    
        $this->debug();
        return false;
    }
}
4

2 回答 2

2

不要试图自动化一切

代码中的魔力越少,支持就越容易,麻烦就越少。
不要试图把所有的逻辑都塞进一个单一的方法中。这是一堂课!您可以根据需要创建任意数量的方法。

当您需要时rowCount()- 明确选择它!这并不难。
但是当你在几个月后偶然发现这段代码时,你就会知道这个值是什么意思。

当您需要单行时 - 使用一种方法来获取单行。当您需要很多行时 - 使用一种方法来获取很多行。
这很简单,而且非常明确!

当你在 2 个月后回到你的代码时,你会完全不知道,你期望什么。所以 -总是明确地写出来。

这是我的mysqli 包装类的摘录,可以给你一个想法:

public function query()
{   
    return $this->rawQuery($this->prepareQuery(func_get_args()));
}
/**
 * Helper function to get scalar value right out of query and optional arguments
 * 
 * Examples:
 * $name = $db->getOne("SELECT name FROM table WHERE id=1");
 * $name = $db->getOne("SELECT name FROM table WHERE id=?i", $id);
 *
 * @param string $query - an SQL query with placeholders
 * @param mixed  $arg,... unlimited number of arguments to match placeholders in the query
 * @return string|FALSE either first column of the first row of resultset or FALSE if none found
 */
public function getOne()
{
    $query = $this->prepareQuery(func_get_args());
    if ($res = $this->rawQuery($query))
    {
        $row = $this->fetch($res);
        if (is_array($row)) {
            return reset($row);
        }
        $this->free($res);
    }
    return FALSE;
}

/**
 * Helper function to get single row right out of query and optional arguments
 * 
 * Examples:
 * $data = $db->getRow("SELECT * FROM table WHERE id=1");
 * $data = $db->getOne("SELECT * FROM table WHERE id=?i", $id);
 *
 * @param string $query - an SQL query with placeholders
 * @param mixed  $arg,... unlimited number of arguments to match placeholders in the query
 * @return array|FALSE either associative array contains first row of resultset or FALSE if none found
 */
public function getRow()
{
    $query = $this->prepareQuery(func_get_args());
    if ($res = $this->rawQuery($query)) {
        $ret = $this->fetch($res);
        $this->free($res);
        return $ret;
    }
    return FALSE;
}

/**
 * Helper function to get single column right out of query and optional arguments
 * 
 * Examples:
 * $ids = $db->getCol("SELECT id FROM table WHERE cat=1");
 * $ids = $db->getCol("SELECT id FROM tags WHERE tagname = ?s", $tag);
 *
 * @param string $query - an SQL query with placeholders
 * @param mixed  $arg,... unlimited number of arguments to match placeholders in the query
 * @return array|FALSE either enumerated array of first fields of all rows of resultset or FALSE if none found
 */
public function getCol()
{
    $ret   = array();
    $query = $this->prepareQuery(func_get_args());
    if ( $res = $this->rawQuery($query) )
    {
        while($row = $this->fetch($res))
        {
            $ret[] = reset($row);
        }
        $this->free($res);
    }
    return $ret;
}

/**
 * Helper function to get all the rows of resultset right out of query and optional arguments
 * 
 * Examples:
 * $data = $db->getAll("SELECT * FROM table");
 * $data = $db->getAll("SELECT * FROM table LIMIT ?i,?i", $start, $rows);
 *
 * @param string $query - an SQL query with placeholders
 * @param mixed  $arg,... unlimited number of arguments to match placeholders in the query
 * @return array enumerated 2d array contains the resultset. Empty if no rows found. 
 */
public function getAll()
{
    $ret   = array();
    $query = $this->prepareQuery(func_get_args());
    if ( $res = $this->rawQuery($query) )
    {
        while($row = $this->fetch($res))
        {
            $ret[] = $row;
        }
        $this->free($res);
    }
    return $ret;
}

看 - 从函数名称中您总能判断出预期的结果:

$name = $db->getOne('SELECT name FROM table WHERE id = ?i',$_GET['id']);
$data = $db->getAll("SELECT * FROM ?n WHERE mod=?s LIMIT ?i",$table,$mod,$limit);

不要被返回行数这样的陷阱所迷惑。
您打算填充的结果集中可能有一行fetchAll。因此,它将返回一维数组而不是多维数组,并且您的页面上将有大量的视频效果

于 2013-02-23T14:21:50.813 回答
-1

由于您没有将答案标记为已接受。我以为我会回答你的问题。我在自己寻找答案时发现了这一点。我同意“你的常识”,因为它们应该是两个独立的功能。但是,直接回答您的问题,这就是我所拥有的(PDO 示例而不是 mysqli):

function select($sql,$params=NULL,$fetchType=NULL){
    try{
        $qry = $this->db->prepare($sql);
        $qry->execute($params);
        if($qry->rowCount() > 1){
            if($fetchType == 'OBJ'){//returns object
                $results = $qry->fetchAll(PDO::FETCH_OBJ);
            }elseif($fetchType == 'NUM'){//-numerical array
                $results = $qry->fetchAll(PDO::FETCH_NUM);
            }else{//default - associative array
                $results = $qry->fetchAll(PDO::FETCH_ASSOC);
            }
        }
        else{
            if($fetchType == 'OBJ'){//returns object
                $results = $qry->fetch(PDO::FETCH_OBJ);
            }elseif($fetchType == 'NUM'){//-numerical array
                $results = $qry->fetch(PDO::FETCH_NUM);
            }else{//default - associative array
                $results = $qry->fetch(PDO::FETCH_ASSOC);
            }
        }

        if($results){
            return $results;
        }else{
            return NULL;
        }
    }
    catch(PDOException $err){
        $this->logError($err);
    }
}

但是我发现,如果我查询表中的所有行但表中只有一行,它将返回一维数组而不是二维数组。我处理结果的代码不适用于两种类型的数组。我每次都可以处理,但我发现如上所述将它们分成不同的函数更容易,所以如果我知道只有一个答案,我可以调用适当的函数。这就是我现在所拥有的:

function select($sql,$params=NULL,$fetchType=NULL){
    try{
        $qry = $this->db->prepare($sql);
        $qry->execute($params);

        if($fetchType == 'OBJ'){//returns object
            $results = $qry->fetch(PDO::FETCH_OBJ);
        }elseif($fetchType == 'NUM'){//-numerical array
            $results = $qry->fetch(PDO::FETCH_NUM);
        }else{//default - associative array
            $results = $qry->fetch(PDO::FETCH_ASSOC);
        }

        if($results){
            return $results;
        }else{
            return NULL;
        }
    }
    catch(PDOException $err){
        $this->logError($err);
    }
}

function selectAll($sql,$params=NULL,$fetchType=NULL){
    try{
        $qry = $this->db->prepare($sql);
        $qry->execute($params);

        if($fetchType == 'OBJ'){//returns object
            $results = $qry->fetchAll(PDO::FETCH_OBJ);
        }elseif($fetchType == 'NUM'){//-numerical array
            $results = $qry->fetchAll(PDO::FETCH_NUM);
        }else{//default - associative array
            $results = $qry->fetchAll(PDO::FETCH_ASSOC);
        }

        if($results){
            return $results;
        }else{
            return NULL;
        }
    }
    catch(PDOException $err){
        $this->logError($err);
    }
}
于 2013-09-06T17:26:47.783 回答