You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

181 lines
6.7 KiB

  1. <?php
  2. // +----------------------------------------------------------------------
  3. // | ThinkPHP [ WE CAN DO IT JUST THINK ]
  4. // +----------------------------------------------------------------------
  5. // | Copyright (c) 2006~2018 http://thinkphp.cn All rights reserved.
  6. // +----------------------------------------------------------------------
  7. // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
  8. // +----------------------------------------------------------------------
  9. // | Author: liu21st <liu21st@gmail.com>
  10. // +----------------------------------------------------------------------
  11. namespace think;
  12. use think\db\Connection;
  13. use think\db\Query;
  14. /**
  15. * Class Db
  16. * @package think
  17. * @method Query table(string $table) static 指定数据表(含前缀)
  18. * @method Query name(string $name) static 指定数据表(不含前缀)
  19. * @method Query where(mixed $field, string $op = null, mixed $condition = null) static 查询条件
  20. * @method Query join(mixed $join, mixed $condition = null, string $type = 'INNER') static JOIN查询
  21. * @method Query union(mixed $union, boolean $all = false) static UNION查询
  22. * @method Query limit(mixed $offset, integer $length = null) static 查询LIMIT
  23. * @method Query order(mixed $field, string $order = null) static 查询ORDER
  24. * @method Query cache(mixed $key = null , integer $expire = null) static 设置查询缓存
  25. * @method mixed value(string $field) static 获取某个字段的值
  26. * @method array column(string $field, string $key = '') static 获取某个列的值
  27. * @method Query view(mixed $join, mixed $field = null, mixed $on = null, string $type = 'INNER') static 视图查询
  28. * @method mixed find(mixed $data = null) static 查询单个记录
  29. * @method mixed select(mixed $data = null) static 查询多个记录
  30. * @method integer insert(array $data, boolean $replace = false, boolean $getLastInsID = false, string $sequence = null) static 插入一条记录
  31. * @method integer insertGetId(array $data, boolean $replace = false, string $sequence = null) static 插入一条记录并返回自增ID
  32. * @method integer insertAll(array $dataSet) static 插入多条记录
  33. * @method integer update(array $data) static 更新记录
  34. * @method integer delete(mixed $data = null) static 删除记录
  35. * @method boolean chunk(integer $count, callable $callback, string $column = null) static 分块获取数据
  36. * @method mixed query(string $sql, array $bind = [], boolean $master = false, bool $pdo = false) static SQL查询
  37. * @method integer execute(string $sql, array $bind = [], boolean $fetch = false, boolean $getLastInsID = false, string $sequence = null) static SQL执行
  38. * @method Paginator paginate(integer $listRows = 15, mixed $simple = null, array $config = []) static 分页查询
  39. * @method mixed transaction(callable $callback) static 执行数据库事务
  40. * @method void startTrans() static 启动事务
  41. * @method void commit() static 用于非自动提交状态下面的查询提交
  42. * @method void rollback() static 事务回滚
  43. * @method boolean batchQuery(array $sqlArray) static 批处理执行SQL语句
  44. * @method string quote(string $str) static SQL指令安全过滤
  45. * @method string getLastInsID($sequence = null) static 获取最近插入的ID
  46. */
  47. class Db
  48. {
  49. /**
  50. * @var Connection[] 数据库连接实例
  51. */
  52. private static $instance = [];
  53. /**
  54. * @var int 查询次数
  55. */
  56. public static $queryTimes = 0;
  57. /**
  58. * @var int 执行次数
  59. */
  60. public static $executeTimes = 0;
  61. /**
  62. * 数据库初始化,并取得数据库类实例
  63. * @access public
  64. * @param mixed $config 连接配置
  65. * @param bool|string $name 连接标识 true 强制重新连接
  66. * @return Connection
  67. * @throws Exception
  68. */
  69. public static function connect($config = [], $name = false)
  70. {
  71. if (false === $name) {
  72. $name = md5(serialize($config));
  73. }
  74. if (true === $name || !isset(self::$instance[$name])) {
  75. // 解析连接参数 支持数组和字符串
  76. $options = self::parseConfig($config);
  77. if (empty($options['type'])) {
  78. throw new \InvalidArgumentException('Undefined db type');
  79. }
  80. $class = false !== strpos($options['type'], '\\') ?
  81. $options['type'] :
  82. '\\think\\db\\connector\\' . ucwords($options['type']);
  83. // 记录初始化信息
  84. if (App::$debug) {
  85. Log::record('[ DB ] INIT ' . $options['type'], 'info');
  86. }
  87. if (true === $name) {
  88. $name = md5(serialize($config));
  89. }
  90. self::$instance[$name] = new $class($options);
  91. }
  92. return self::$instance[$name];
  93. }
  94. /**
  95. * 清除连接实例
  96. * @access public
  97. * @return void
  98. */
  99. public static function clear()
  100. {
  101. self::$instance = [];
  102. }
  103. /**
  104. * 数据库连接参数解析
  105. * @access private
  106. * @param mixed $config 连接参数
  107. * @return array
  108. */
  109. private static function parseConfig($config)
  110. {
  111. if (empty($config)) {
  112. $config = Config::get('database');
  113. } elseif (is_string($config) && false === strpos($config, '/')) {
  114. $config = Config::get($config); // 支持读取配置参数
  115. }
  116. return is_string($config) ? self::parseDsn($config) : $config;
  117. }
  118. /**
  119. * DSN 解析
  120. * 格式: mysql://username:passwd@localhost:3306/DbName?param1=val1&param2=val2#utf8
  121. * @access private
  122. * @param string $dsnStr 数据库 DSN 字符串解析
  123. * @return array
  124. */
  125. private static function parseDsn($dsnStr)
  126. {
  127. $info = parse_url($dsnStr);
  128. if (!$info) {
  129. return [];
  130. }
  131. $dsn = [
  132. 'type' => $info['scheme'],
  133. 'username' => isset($info['user']) ? $info['user'] : '',
  134. 'password' => isset($info['pass']) ? $info['pass'] : '',
  135. 'hostname' => isset($info['host']) ? $info['host'] : '',
  136. 'hostport' => isset($info['port']) ? $info['port'] : '',
  137. 'database' => !empty($info['path']) ? ltrim($info['path'], '/') : '',
  138. 'charset' => isset($info['fragment']) ? $info['fragment'] : 'utf8',
  139. ];
  140. if (isset($info['query'])) {
  141. parse_str($info['query'], $dsn['params']);
  142. } else {
  143. $dsn['params'] = [];
  144. }
  145. return $dsn;
  146. }
  147. /**
  148. * 调用驱动类的方法
  149. * @access public
  150. * @param string $method 方法名
  151. * @param array $params 参数
  152. * @return mixed
  153. */
  154. public static function __callStatic($method, $params)
  155. {
  156. return call_user_func_array([self::connect(), $method], $params);
  157. }
  158. }