class HRDB{ protected $pdo; protected $res; protected $config;
/*构造函数*/ function __construct($config){ $this->Config = $config; $this->connect(); }
/*数据库连接*/ public function connect(){ $this->pdo = new PDO($this->Config["dsn"], $this->Config["name"], $this->Config["password"]); $this->pdo->query("set names utf8;"); //把结果序列化成stdClass //$this->pdo->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_OBJ); //自己写代码捕获Exception $this->pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION); }
/*数据库关闭*/ public function close(){ $this->pdo = null; }
public function query($sql){ $res = $this->pdo->query($sql); if($res){ $this->res = $res; } } public function exec($sql){ $res = $this->pdo->exec($sql); if($res){ $this->res = $res; } } public function fetchAll(){ return $this->res->fetchAll(); } public function fetch(){ return $this->res->fetch(); } public function fetchColumn(){ return $this->res->fetchColumn(); } public function lastInsertId(){ return $this->res->lastInsertId(); }
/** * 参数说明 * int $debug 是否开启调试,开启则输出sql语句 * 0 不开启 * 1 开启 * 2 开启并终止程序 * int $mode 返回类型 * 0 返回多条记录 * 1 返回单条记录 * 2 返回行数 * string/array $table 数据库表,两种传值模式 * 普通模式: * "tb_member, tb_money" * 数组模式: * array("tb_member", "tb_money") * string/array $fields 需要查询的数据库字段,允许为空,默认为查找全部,两种传值模式 * 普通模式: * "username, password" * 数组模式: * array("username", "password") * string/array $sqlwhere 查询条件,允许为空,两种传值模式 * 普通模式: * "and type = 1 and username like "%os%"" * 数组模式: * array("type = 1", "username like "%os%"") * string $orderby 排序,默认为id倒序 */ public function select($debug, $mode, $table, $fields="*", $sqlwhere="", $orderby="tbid desc"){ //参数处理 if(is_array($table)){ $table = implode(", ", $table); } if(is_array($fields)){ $fields = implode(", ", $fields); } if(is_array($sqlwhere)){ $sqlwhere = " and ".implode(" and ", $sqlwhere); } //数据库操作 if($debug === 0){ if($mode === 2){ $this->query("select count(tbid) from $table where 1=1 $sqlwhere"); $return = $this->fetchColumn(); }else if($mode === 1){ $this->query("select $fields from $table where 1=1 $sqlwhere order by $orderby"); $return = $this->fetch(); }else{ $this->query("select $fields from $table where 1=1 $sqlwhere order by $orderby"); $return = $this->fetchAll(); } return $return; }else{ if($mode === 2){ echo "select count(tbid) from $table where 1=1 $sqlwhere"; }else if($mode === 1){ echo "select $fields from $table where 1=1 $sqlwhere order by $orderby"; } else{ echo "select $fields from $table where 1=1 $sqlwhere order by $orderby"; } if($debug === 2){ exit; } } }