PHP用PDO如何封装简单易用的DB类详解

__setup([ 'dsn'=>'mysql:dbname=db_test;host=localhost','username'=>'root','password'=>'******','charset'=>'utf8' ]);

$user = $db->fetch('SELECT * FROM user where id = :id',['id' => 1]);
echo $user['name'];
echo "n";

$insertId = $db->insert('user',['name' => 'salamander','created_at' => time()]);
echo "insert user {$insertId}n";
$users = $db->fetchAll('SELECT * FROM user');
foreach ($users as $item) {
echo "user {$item['id']} is {$item['name']} n";
}

运行结果

DB工具类

class DB
{
private $dsn;
private $sth;
private $dbh;
private $user;
private $charset;
private $password;

public $lastSQL = '';

public function __setup($config = array())
{
$this->dsn = $config['dsn'];
$this->user = $config['username'];
$this->password = $config['password'];
$this->charset = $config['charset'];
$this->connect();
}

private function connect()
{
if(!$this->dbh){
$options = array(
PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES ' . $this->charset,);
$this->dbh = new PDO($this->dsn,$this->user,$this->password,$options);
}
}

public function beginTransaction()
{
return $this->dbh->beginTransaction();
}

public function inTransaction()
{
return $this->dbh->inTransaction();
}

public function rollBack()
{
return $this->dbh->rollBack();
}

public function commit()
{
return $this->dbh->commit();
}

function watchException($execute_state)
{
if(!$execute_state){
throw new MySQLException("SQL: {$this->lastSQL}n".$this->sth->errorInfo()[2],intval($this->sth->errorCode()));
}
}

public function fetchAll($sql,$parameters=[])
{
$result = [];
$this->lastSQL = $sql;
$this->sth = $this->dbh->prepare($sql);
$this->watchException($this->sth->execute($parameters));
while($result[] = $this->sth->fetch(PDO::FETCH_ASSOC)){ }
array_pop($result);
return $result;
}

public function fetchColumnAll($sql,$parameters=[],$position=0)
{
$result = [];
$this->lastSQL = $sql;
$this->sth = $this->dbh->prepare($sql);
$this->watchException($this->sth->execute($parameters));
while($result[] = $this->sth->fetch(PDO::FETCH_COLUMN,$position)){ }
array_pop($result);
return $result;
}

public function exists($sql,$parameters=[])
{
$this->lastSQL = $sql;
$data = $this->fetch($sql,$parameters);
return !empty($data);
}

public function query($sql,$parameters=[])
{
$this->lastSQL = $sql;
$this->sth = $this->dbh->prepare($sql);
$this->watchException($this->sth->execute($parameters));
return $this->sth->rowCount();
}

public function fetch($sql,$type=PDO::FETCH_ASSOC)
{
$this->lastSQL = $sql;
$this->sth = $this->dbh->prepare($sql);
$this->watchException($this->sth->execute($parameters));
return $this->sth->fetch($type);
}

public function fetchColumn($sql,$position=0)
{
$this->lastSQL = $sql;
$this->sth = $this->dbh->prepare($sql);
$this->watchException($this->sth->execute($parameters));
return $this->sth->fetch(PDO::FETCH_COLUMN,$position);
}

public function update($table,$condition=[])
{
$table = $this->format_table_name($table);
$sql = "UPDATE $table SET ";
$fields = [];
$pdoparameters = [];
foreach ( $parameters as $field=>$value){
$fields[] = ''.$field.'=:field'.$field;
$pdoparameters['field'.$field] = $value;
}
$sql .= implode(',',$fields);
$fields = [];
$where = '';
if(is_string($condition)) {
$where = $condition;
} else if(isarray($condition)) {
foreach($condition as $field=>$value){
$parameters[$field] = $value;
$fields[] = ''.$field.'=:condition'.$field;
$pdoparameters['condition'.$field] = $value;
}
$where = implode(' AND ',$fields);
}
if(!empty($where)) {
$sql .= ' WHERE '.$where;
}
return $this->query($sql,$pdo_parameters);
}

（编辑：源码网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!