加强版phplib的DB类

复制代码 代码如下:

  <?php

  /****************************************************************************************************************

  为了便于自己的开发,又不想使用ADODB、PEAR::DB这样的庞然大物,

  就用在PHPLib DB类的基础上、参考PEAR::DB类,封装的DB类,简单好使,非常方便。

  MySQL有效

  [ 连接数据库 ]

  //包含数据库处理类文件

  include_once("database.inc.php");

  //本地数据库配置

  define("DB_HOST",    "localhost"); //数据库服务器

  define("DB_USER_NAME",   "root");  //数据库用户名

  define("DB_USER_PASS",   "");   //密码

  define("DB_DATABASE",   "test");  //数据库

  //连接本地数据库

  $db = new DB_Sql();

  $db->connect(DB_DATABASE, DB_HOST, DB_USER_NAME, DB_USER_PASS);

  [ 使用方法 ]

  //获取所有记录

  $sql = "SELECT * FROM table1";

  $all_record = $db->get_all($sql);

  //获取一条

  $sql = "SELECT * FROM table1 WHERE id = '1'";

  $one_row = $db->get_one($sql);

  //分页查询,提取20条记录

  $sql = "SELECT * FROM table1";

  $page_record = $db->limit_query($sql, $start=0, $offset=20, $order="ORDER BY id DESC");

  //提取指定数目的记录

  $sql = "SELECT * FROM table1";

  $limit_record = $db->get_limit($sql,10);

  //统计记录数,统计所有类型为学生的

  $count = $db->count("table1", "id", "type = 'student'");

  //插入一条记录

  $info_array = array(

  "name"  => "heiyeluren",

  "type"  => "student",

  "age"  => "22",

  "gender" => "boy"

  );

  $db->insert("table1", $info_array);

  //更新一条记录

  $info_array = array(

  "name"  => "heiyeluren",

  "type"  => "teacher",

  "age"  => "22",

  "gender" => "boy"

  );

  $db->update("table1", $info_array, "name = 'heiyeluren'");

  //删除记录

  $db->delete("table1", "name = 'heiyeluren'");

  //执行一条无结果集的SQL

  $db->execute("DELETE FROM table1 WHERE name = 'heiyeluren'");

  ********************************************************************************************************/

  /**

  * 文件: database.inc.php

  * 描述: 数据库操作类

  * 说明: 本库使用PHPLib DB库作为核心, 同时增加一些实用方法, 详细参考注释

  */

  class DB_Sql

  {

  /* public: connection parameters */

  var $Host     = "";

  var $Database = "";

  var $User     = "";

  var $Password = "";

  /* public: configuration parameters */

  var $Auto_Free     = 1;     ## Set to 1 for automatic mysql_free_result()

  var $Debug         = 0;     ## Set to 1 for debugging messages.

  var $Halt_On_Error = "yes"; ## "yes" (halt with message), "no" (ignore errors quietly), "report" (ignore errror, but spit a warning)

  var $PConnect      = 0;     ## Set to 1 to use persistent database connections

  var $Seq_Table     = "db_sequence";

  /* public: result array and current row number */

  var $Record   = array();

  var $Row;

  /* public: current error number and error text */

  var $Errno    = 0;

  var $Error    = "";

  /* public: this is an api revision, not a CVS revision. */

  var $type     = "mysql";

  //var $revision = "1.2";

  /* private: link and query handles */

  var $Link_ID  = 0;

  var $Query_ID = 0;

  var $locked   = false;      ## set to true while we have a lock

  /* public: constructor */

  function DB_Sql() {

  $this->query($query);

  }

  /* public: some trivial reporting */

  function link_id() {

  return $this->Link_ID;

  }

  function query_id() {

  return $this->Query_ID;

  }

  /* public: connection management */

  function connect($Database = "", $Host = "", $User = "", $Password = "") {

  /* Handle defaults */

  if ("" == $Database)

  $Database = $this->Database;

  if ("" == $Host)

  $Host     = $this->Host;

  if ("" == $User)

  $User     = $this->User;

  if ("" == $Password)

  $Password = $this->Password;

  /* establish connection, select database */

  if ( 0 == $this->Link_ID ) {

  if(!$this->PConnect) {

  $this->Link_ID = mysql_connect($Host, $User, $Password);

  } else {

  $this->Link_ID = mysql_pconnect($Host, $User, $Password);

  }

  if (!$this->Link_ID) {

  $this->halt("connect($Host, $User, \$Password) failed.");

  return 0;

  }

  if (!@mysql_select_db($Database,$this->Link_ID)) {

  $this->halt("cannot use database ".$Database);

  return 0;

  }

  }

  return $this->Link_ID;

  }

  /* public: discard the query result */

  function free() {

  @mysql_free_result($this->Query_ID);

  $this->Query_ID = 0;

  }

  /* public: perform a query */

  function query($Query_String) {

  /* No empty queries, please, since PHP4 chokes on them. */

  if ($Query_String == "")

  /* The empty query string is passed on from the constructor,

  * when calling the class without a query, e.g. in situations

  * like these: '$db = new DB_Sql_Subclass;'

  */

  return 0;

  if (!$this->connect()) {

  return 0; /* we already complained in connect() about that. */

  };

  # New query, discard previous result.

  if ($this->Query_ID) {

  $this->free();

  }

  if ($this->Debug)

  printf("Debug: query = %s<br>\n", $Query_String);

  $this->Query_ID = @mysql_query($Query_String,$this->Link_ID);

  $this->Row   = 0;

  $this->Errno = mysql_errno();

  $this->Error = mysql_error();

  if (!$this->Query_ID) {

  $this->halt("Invalid SQL: ".$Query_String);

  }

  # Will return nada if it fails. That's fine.

  return $this->Query_ID;

  }

  /* public: walk result set */

  function next_record() {

  if (!$this->Query_ID) {

  $this->halt("next_record called with no query pending.");

  return 0;

  }

  $this->Record = @mysql_fetch_array($this->Query_ID);

  $this->Row   += 1;

  $this->Errno  = mysql_errno();

  $this->Error  = mysql_error();

  $stat = is_array($this->Record);

  if (!$stat && $this->Auto_Free) {

  $this->free();

  }

  return $stat;

  }

  /* public: position in result set */

  function seek($pos = 0) {

  $status = @mysql_data_seek($this->Query_ID, $pos);

  if ($status)

  $this->Row = $pos;

  else {

  $this->halt("seek($pos) failed: result has ".$this->num_rows()." rows.");

  /* half assed attempt to save the day,

  * but do not consider this documented or even

  * desireable behaviour.

  */

  @mysql_data_seek($this->Query_ID, $this->num_rows());

  $this->Row = $this->num_rows();

  return 0;

  }

  return 1;

  }

  /* public: table locking */

  function lock($table, $mode = "write") {

  $query = "lock tables ";

  if(is_array($table)) {

  while(list($key,$value) = each($table)) {

  // text keys are "read", "read local", "write", "low priority write"

  if(is_int($key)) $key = $mode;

  if(strpos($value, ",")) {

  $query .= str_replace(",", " $key, ", $value) . " $key, ";

  } else {

  $query .= "$value $key, ";

  }

  }

  $query = substr($query, 0, -2);

  } elseif(strpos($table, ",")) {

  $query .= str_replace(",", " $mode, ", $table) . " $mode";

  } else {

  $query .= "$table $mode";

  }

  if(!$this->query($query)) {

  $this->halt("lock() failed.");

  return false;

  }

  $this->locked = true;

  return true;

  }

  function unlock() {

  // set before unlock to avoid potential loop

  $this->locked = false;

  if(!$this->query("unlock tables")) {

  $this->halt("unlock() failed.");

  return false;

  }

  return true;

  }

  /* public: evaluate the result (size, width) */

  function affected_rows() {

  return @mysql_affected_rows($this->Link_ID);

  }

  function num_rows() {

  return @mysql_num_rows($this->Query_ID);

  }

  function num_fields() {

  return @mysql_num_fields($this->Query_ID);

  }

  /* public: shorthand notation */

  function nf() {

  return $this->num_rows();

  }

  function np() {

  print $this->num_rows();

  }

  function f($Name) {

  if (isset($this->Record[$Name])) {

  return $this->Record[$Name];

  }

  }

  function p($Name) {

  if (isset($this->Record[$Name])) {

  print $this->Record[$Name];

  }

  }

  /* public: sequence numbers */

  function nextid($seq_name) {

  /* if no current lock, lock sequence table */

  if(!$this->locked) {

  if($this->lock($this->Seq_Table)) {

  $locked = true;

  } else {

  $this->halt("cannot lock ".$this->Seq_Table." - has it been created?");

  return 0;

  }

  }

  /* get sequence number and increment */

  $q = sprintf("select nextid from %s where seq_name = '%s'",

  $this->Seq_Table,

  $seq_name);

  if(!$this->query($q)) {

  $this->halt('query failed in nextid: '.$q);

  return 0;

  }

  /* No current value, make one */

  if(!$this->next_record()) {

  $currentid = 0;

  $q = sprintf("insert into %s values('%s', %s)",

  $this->Seq_Table,

  $seq_name,

  $currentid);

  if(!$this->query($q)) {

  $this->halt('query failed in nextid: '.$q);

  return 0;

  }

  } else {

  $currentid = $this->f("nextid");

  }

  $nextid = $currentid + 1;

  $q = sprintf("update %s set nextid = '%s' where seq_name = '%s'",

  $this->Seq_Table,

  $nextid,

  $seq_name);

  if(!$this->query($q)) {

  $this->halt('query failed in nextid: '.$q);

  return 0;

  }

  /* if nextid() locked the sequence table, unlock it */

  if($locked) {

  $this->unlock();

  }

  return $nextid;

  }

  /* public: return table metadata */

  function metadata($table = "", $full = false) {

  $count = 0;

  $id    = 0;

  $res   = array();

  /*

  * Due to compatibility problems with Table we changed the behavior

  * of metadata();

  * depending on $full, metadata returns the following values:

  *

  * - full is false (default):

  * $result[]:

  *   [0]["table"]  table name

  *   [0]["name"]   field name

  *   [0]["type"]   field type

  *   [0]["len"]    field length

  *   [0]["flags"]  field flags

  *

  * - full is true

  * $result[]:

  *   ["num_fields"] number of metadata records

  *   [0]["table"]  table name

  *   [0]["name"]   field name

  *   [0]["type"]   field type

  *   [0]["len"]    field length

  *   [0]["flags"]  field flags

  *   ["meta"][field name]  index of field named "field name"

  *   This last one could be used if you have a field name, but no index.

  *   Test:  if (isset($result['meta']['myfield'])) { ...

  */

  // if no $table specified, assume that we are working with a query

  // result

  if ($table) {

  $this->connect();

  $id = @mysql_list_fields($this->Database, $table);

  if (!$id) {

  $this->halt("Metadata query failed.");

  return false;

  }

  } else {

  $id = $this->Query_ID;

  if (!$id) {

  $this->halt("No query specified.");

  return false;

  }

  }

  $count = @mysql_num_fields($id);

  // made this IF due to performance (one if is faster than $count if's)

  if (!$full) {

  for ($i=0; $i<$count; $i++) {

  $res[$i]["table"] = @mysql_field_table ($id, $i);

  $res[$i]["name"]  = @mysql_field_name  ($id, $i);

  $res[$i]["type"]  = @mysql_field_type  ($id, $i);

  $res[$i]["len"]   = @mysql_field_len   ($id, $i);

  $res[$i]["flags"] = @mysql_field_flags ($id, $i);

  }

  } else { // full

  $res["num_fields"]= $count;

  for ($i=0; $i<$count; $i++) {

  $res[$i]["table"] = @mysql_field_table ($id, $i);

  $res[$i]["name"]  = @mysql_field_name  ($id, $i);

  $res[$i]["type"]  = @mysql_field_type  ($id, $i);

  $res[$i]["len"]   = @mysql_field_len   ($id, $i);

  $res[$i]["flags"] = @mysql_field_flags ($id, $i);

  $res["meta"][$res[$i]["name"]] = $i;

  }

  }

  // free the result only if we were called on a table

  if ($table) {

  @mysql_free_result($id);

  }

  return $res;

  }

  /* public: find available table names */

  function table_names() {

  $this->connect();

  $h = @mysql_query("show tables", $this->Link_ID);

  $i = 0;

  while ($info = @mysql_fetch_row($h)) {

  $return[$i]["table_name"]      = $info[0];

  $return[$i]["tablespace_name"] = $this->Database;

  $return[$i]["database"]        = $this->Database;

  $i++;

  }

  @mysql_free_result($h);

  return $return;

  }

  /* private: error handling */

  function halt($msg) {

  $this->Error = @mysql_error($this->Link_ID);

  $this->Errno = @mysql_errno($this->Link_ID);

  if ($this->locked) {

  $this->unlock();

  }

  if ($this->Halt_On_Error == "no")

  return;

  $this->haltmsg($msg);

  if ($this->Halt_On_Error != "report")

  die("Session halted.");

  }

  function haltmsg($msg) {

  printf("</td></tr></table><b>Database error:</b> %s<br>\n", $msg);

  printf("<b>MySQL Error</b>: %s (%s)<br>\n",

  $this->Errno,

  $this->Error);

  }

  //----------------------------------

  // 模块: 自定义函数

  // 功能: 部分实用的数据库处理方法

  // 作者: heiyeluren

  // 时间: 2005-12-26

  //----------------------------------

  /**

  * 方法: execute($sql)

  * 功能: 执行一条SQL语句,主要针对没有结果集返回的SQL

  * 参数: $sql 需要执行的SQL语句,例如:execute("DELETE FROM table1 WHERE id = '1'")

  * 返回: 更新成功返回True,失败返回False

  */

  function execute($sql)

  {

  if (empty($sql))

  {

  $this->error("Invalid parameter");

  }

  if (!$this->query($sql))

  {

  return false;

  }

  return true;

  }

  /**

  * 方法: get_all($sql)

  * 功能: 获取SQL执行的所有记录

  * 参数: $sql  需要执行的SQL,例如: get_all("SELECT * FROM Table1")

  * 返回: 返回包含所有查询结果的二维数组

  */

  function get_all($sql)

  {

  $this->query($sql);

  $result_array = array();

  while($this->next_record())

  {

  $result_array[] = $this->Record;

  }

  if (count($result_array)<=0)

  {

  return 0;

  }

  return $result_array;

  }

  /**

  * 方法: get_one($sql)

  * 功能: 获取SQL执行的一条记录

  * 参数: $sql 需要执行的SQL,例如: get_one("SELECT * FROM Table1 WHERE id = '1'")

  * 返回: 返回包含一条查询结果的一维数组

  */

  function get_one($sql)

  {

  $this->query($sql);

  if (!$this->next_record())

  {

  return 0;

  }

  return $this->Record;

  }

  /**

  * 方法: get_limit($sql, $limit)

  * 功能: 获取SQL执行的指定数量的记录

  * 参数:

  * $sql  需要执行的SQL,例如: SELECT * FROM Table1

  * $limit 需要限制的记录数

  * 例如  需要获取10条记录, get_limit("SELECT * FROM Table1", 10);

  *

  * 返回: 返回包含所有查询结果的二维数组

  */

  function get_limit($sql, $limit)

  {

  $this->query($sql);

  $result_array = array();

  for ($i=0; $i<$limit&&$this->next_record(); $i++)

  {

  $result_array[] = $this->Record;

  }

  if (count($result_array) <= 0)

  {

  return 0;

  }

  return $result_array;

  }

  /**

  * 方法: limit_query($sql, $start=0, $offset=20, $order="")

  * 功能: 为分页的获取SQL执行的指定数量的记录

  * 参数:

  * $sql  需要执行的SQL,例如: SELECT * FROM Table1

  * $start 记录的开始数, 缺省为0

  * $offset 记录的偏移量,缺省为20

  * $order 排序方式,缺省为空,例如:ORDER BY id DESC

  * 例如  需要获取从0到10的记录并且按照ID号倒排, get_limit("SELECT * FROM Table1", 0, 10, "ORDER BY id DESC");

  *

  * 返回: 返回包含所有查询结果的二维数组

  */

  function limit_query($sql, $start=0, $offset=20, $order="")

  {

  $sql = $sql ." $order  LIMIT $start,$offset";

  $this->query($sql);

  $result = array();

  while($this->next_record())

  {

  $result[] = $this->Record;

  }

  if (count($result) <=0 )

  {

  return 0;

  }

  return $result;

  }

  /**

  * 方法: count($table,$field="*", $where="")

  * 功能: 统计表中数据总数

  * 参数:

  * $table 需要统计的表名

  * $field 需要统计的字段,默认为*

  * $where 条件语句,缺省为空

  * 例如  按照ID统计所有年龄小于20岁的用户, count("user_table", "id", "user_age < 20")

  *

  * 返回: 返回统计结果的数字

  */

  function count($table,$field="*", $where="")

  {

  $sql = (empty($where) ? "SELECT COUNT($field) FROM $table" : "SELECT COUNT($field) FROM $table WHERE $where");

  $result = $this->get_one($sql);

  if (!is_array($result))

  {

  return 0;

  }

  return $result[0];

  }

  /**

  * 方法: insert($table,$dataArray)

  * 功能: 插入一条记录到表里

  * 参数:

  * $table  需要插入的表名

  * $dataArray 需要插入字段和值的数组,键为字段名,值为字段值,例如:array("user_name"=>"张三", "user_age"=>"20岁");

  * 例如   比如插入用户张三,年龄为20, insert("users", array("user_name"=>"张三", "user_age"=>"20岁"))

  *

  * 返回: 插入记录成功返回True,失败返回False

  */

  function insert($table,$dataArray)

  {

  if (!is_array($dataArray) || count($dataArray)<=0)

  {

  $this->error("Invalid parameter");

  }

  while(list($key,$val) = each($dataArray))

  {

  $field .= "$key,";

  $value .= "'$val',";

  }

  $field = substr($field, 0, -1);

  $value = substr($value, 0, -1);

  $sql = "INSERT INTO $table ($field) VALUES ($value)";

  if (!$this->query($sql))

  {

  return false;

  }

  return true;

  }

  /**

  * 方法: update($talbe, $dataArray, $where)

  * 功能: 更新一条记录

  * 参数:

  * $table  需要更新的表名

  * $dataArray 需要更新字段和值的数组,键为字段名,值为字段值,例如:array("user_name"=>"张三", "user_age"=>"20岁");

  * $where  条件语句

  * 例如   比如更新姓名为张三的用户为李四,年龄为21

  *    update("users",  array("user_name"=>"张三", "user_age"=>"20岁"),  "user_name='张三'")

  *

  * 返回: 更新成功返回True,失败返回False

  */

  function update($talbe, $dataArray, $where)

  {

  if (!is_array($dataArray) || count($dataArray)<=0)

  {

  $this->error("Invalid parameter");

  }

  while(list($key,$val) = each($dataArray))

  {

  $value .= "$key = '$val',";

  }

  $value = substr($value, 0, -1);

  $sql = "UPDATE $talbe SET $value WHERE $where";

  if (!$this->query($sql))

  {

  return false;

  }

  return true;

  }

  /**

  * 方法: delete($table, $where)

  * 功能: 删除一条记录

  * 参数:

  * $table  需要删除记录的表名

  * $where  需要删除记录的条件语句

  * 例如   比如要删除用户名为张三的用户,delete("users",  "user_name='张三'")

  *

  * 返回: 更新成功返回True,失败返回False

  */

  function delete($table, $where)

  {

  if (empty($where))

  {

  $this->error("Invalid parameter");

  }

  $sql = "DELETE FROM $table WHERE $where";

  if (!$this->query($sql))

  {

  return false;

  }

  return true;

  }

  /**

  * 方法: error($msg="")

  * 功能: 显示错误信息后中止脚本

  * 参数: $msg 需要显示的错误信息

  * 返回: 无返回

  */

  function error($msg="")

  {

  echo "<strong>Error</strong>: $msg\n<br>\n";

  exit();

  }

  /**

  * 方法:get_insert_id()

  * 功能:获取最后插入的ID

  * 参数: 无参数

  * 返回:关闭成功返回ID,失败返回0

  */

  function get_insert_id()

  {

  return mysql_insert_id($this->Link_ID);

  }

  /**

  * 方法:close()

  * 功能:关闭当前数据库连接

  * 参数: 无参数

  * 返回:关闭成功返回true,失败返回false

  */

  function close()

  {

  return mysql_close($this->Link_ID);

  }

  }

  ?>