123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589 |
- <?php
- // +----------------------------------------------------------------------
- // | ThinkPHP [ WE CAN DO IT JUST THINK ]
- // +----------------------------------------------------------------------
- // | Copyright (c) 2006~2017 http://thinkphp.cn All rights reserved.
- // +----------------------------------------------------------------------
- // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
- // +----------------------------------------------------------------------
- // | Author: liu21st <liu21st@gmail.com>
- // +----------------------------------------------------------------------
- //------------------------
- // ThinkPHP 助手函数
- //-------------------------
- use think\Cache;
- use think\Config;
- use think\Cookie;
- use think\Db;
- use think\Debug;
- use think\exception\HttpException;
- use think\exception\HttpResponseException;
- use think\Lang;
- use think\Loader;
- use think\Log;
- use think\Model;
- use think\Request;
- use think\Response;
- use think\Session;
- use think\Url;
- use think\View;
- if (!function_exists('load_trait')) {
- /**
- * 快速导入Traits PHP5.5以上无需调用
- * @param string $class trait库
- * @param string $ext 类库后缀
- * @return boolean
- */
- function load_trait($class, $ext = EXT)
- {
- return Loader::import($class, TRAIT_PATH, $ext);
- }
- }
- if (!function_exists('exception')) {
- /**
- * 抛出异常处理
- *
- * @param string $msg 异常消息
- * @param integer $code 异常代码 默认为0
- * @param string $exception 异常类
- *
- * @throws Exception
- */
- function exception($msg, $code = 0, $exception = '')
- {
- $e = $exception ?: '\think\Exception';
- throw new $e($msg, $code);
- }
- }
- if (!function_exists('debug')) {
- /**
- * 记录时间(微秒)和内存使用情况
- * @param string $start 开始标签
- * @param string $end 结束标签
- * @param integer|string $dec 小数位 如果是m 表示统计内存占用
- * @return mixed
- */
- function debug($start, $end = '', $dec = 6)
- {
- if ('' == $end) {
- Debug::remark($start);
- } else {
- return 'm' == $dec ? Debug::getRangeMem($start, $end) : Debug::getRangeTime($start, $end, $dec);
- }
- }
- }
- if (!function_exists('lang')) {
- /**
- * 获取语言变量值
- * @param string $name 语言变量名
- * @param array $vars 动态变量值
- * @param string $lang 语言
- * @return mixed
- */
- function lang($name, $vars = [], $lang = '')
- {
- return Lang::get($name, $vars, $lang);
- }
- }
- if (!function_exists('config')) {
- /**
- * 获取和设置配置参数
- * @param string|array $name 参数名
- * @param mixed $value 参数值
- * @param string $range 作用域
- * @return mixed
- */
- function config($name = '', $value = null, $range = '')
- {
- if (is_null($value) && is_string($name)) {
- return 0 === strpos($name, '?') ? Config::has(substr($name, 1), $range) : Config::get($name, $range);
- } else {
- return Config::set($name, $value, $range);
- }
- }
- }
- if (!function_exists('input')) {
- /**
- * 获取输入数据 支持默认值和过滤
- * @param string $key 获取的变量名
- * @param mixed $default 默认值
- * @param string $filter 过滤方法
- * @return mixed
- */
- function input($key = '', $default = null, $filter = '')
- {
- if (0 === strpos($key, '?')) {
- $key = substr($key, 1);
- $has = true;
- }
- if ($pos = strpos($key, '.')) {
- // 指定参数来源
- list($method, $key) = explode('.', $key, 2);
- if (!in_array($method, ['get', 'post', 'put', 'patch', 'delete', 'route', 'param', 'request', 'session', 'cookie', 'server', 'env', 'path', 'file'])) {
- $key = $method . '.' . $key;
- $method = 'param';
- }
- } else {
- // 默认为自动判断
- $method = 'param';
- }
- if (isset($has)) {
- return request()->has($key, $method, $default);
- } else {
- return request()->$method($key, $default, $filter);
- }
- }
- }
- if (!function_exists('widget')) {
- /**
- * 渲染输出Widget
- * @param string $name Widget名称
- * @param array $data 传入的参数
- * @return mixed
- */
- function widget($name, $data = [])
- {
- return Loader::action($name, $data, 'widget');
- }
- }
- if (!function_exists('model')) {
- /**
- * 实例化Model
- * @param string $name Model名称
- * @param string $layer 业务层名称
- * @param bool $appendSuffix 是否添加类名后缀
- * @return \think\Model
- */
- function model($name = '', $layer = 'model', $appendSuffix = false)
- {
- return Loader::model($name, $layer, $appendSuffix);
- }
- }
- if (!function_exists('validate')) {
- /**
- * 实例化验证器
- * @param string $name 验证器名称
- * @param string $layer 业务层名称
- * @param bool $appendSuffix 是否添加类名后缀
- * @return \think\Validate
- */
- function validate($name = '', $layer = 'validate', $appendSuffix = false)
- {
- return Loader::validate($name, $layer, $appendSuffix);
- }
- }
- if (!function_exists('db')) {
- /**
- * 实例化数据库类
- * @param string $name 操作的数据表名称(不含前缀)
- * @param array|string $config 数据库配置参数
- * @param bool $force 是否强制重新连接
- * @return \think\db\Query
- */
- function db($name = '', $config = [], $force = false)
- {
- return Db::connect($config, $force)->name($name);
- }
- }
- if (!function_exists('controller')) {
- /**
- * 实例化控制器 格式:[模块/]控制器
- * @param string $name 资源地址
- * @param string $layer 控制层名称
- * @param bool $appendSuffix 是否添加类名后缀
- * @return \think\Controller
- */
- function controller($name, $layer = 'controller', $appendSuffix = false)
- {
- return Loader::controller($name, $layer, $appendSuffix);
- }
- }
- if (!function_exists('action')) {
- /**
- * 调用模块的操作方法 参数格式 [模块/控制器/]操作
- * @param string $url 调用地址
- * @param string|array $vars 调用参数 支持字符串和数组
- * @param string $layer 要调用的控制层名称
- * @param bool $appendSuffix 是否添加类名后缀
- * @return mixed
- */
- function action($url, $vars = [], $layer = 'controller', $appendSuffix = false)
- {
- return Loader::action($url, $vars, $layer, $appendSuffix);
- }
- }
- if (!function_exists('import')) {
- /**
- * 导入所需的类库 同java的Import 本函数有缓存功能
- * @param string $class 类库命名空间字符串
- * @param string $baseUrl 起始路径
- * @param string $ext 导入的文件扩展名
- * @return boolean
- */
- function import($class, $baseUrl = '', $ext = EXT)
- {
- return Loader::import($class, $baseUrl, $ext);
- }
- }
- if (!function_exists('vendor')) {
- /**
- * 快速导入第三方框架类库 所有第三方框架的类库文件统一放到 系统的Vendor目录下面
- * @param string $class 类库
- * @param string $ext 类库后缀
- * @return boolean
- */
- function vendor($class, $ext = EXT)
- {
- return Loader::import($class, VENDOR_PATH, $ext);
- }
- }
- if (!function_exists('dump')) {
- /**
- * 浏览器友好的变量输出
- * @param mixed $var 变量
- * @param boolean $echo 是否输出 默认为true 如果为false 则返回输出字符串
- * @param string $label 标签 默认为空
- * @return void|string
- */
- function dump($var, $echo = true, $label = null)
- {
- return Debug::dump($var, $echo, $label);
- }
- }
- if (!function_exists('url')) {
- /**
- * Url生成
- * @param string $url 路由地址
- * @param string|array $vars 变量
- * @param bool|string $suffix 生成的URL后缀
- * @param bool|string $domain 域名
- * @return string
- */
- function url($url = '', $vars = '', $suffix = true, $domain = false)
- {
- return Url::build($url, $vars, $suffix, $domain);
- }
- }
- if (!function_exists('session')) {
- /**
- * Session管理
- * @param string|array $name session名称,如果为数组表示进行session设置
- * @param mixed $value session值
- * @param string $prefix 前缀
- * @return mixed
- */
- function session($name, $value = '', $prefix = null)
- {
- if (is_array($name)) {
- // 初始化
- Session::init($name);
- } elseif (is_null($name)) {
- // 清除
- Session::clear('' === $value ? null : $value);
- } elseif ('' === $value) {
- // 判断或获取
- return 0 === strpos($name, '?') ? Session::has(substr($name, 1), $prefix) : Session::get($name, $prefix);
- } elseif (is_null($value)) {
- // 删除
- return Session::delete($name, $prefix);
- } else {
- // 设置
- return Session::set($name, $value, $prefix);
- }
- }
- }
- if (!function_exists('cookie')) {
- /**
- * Cookie管理
- * @param string|array $name cookie名称,如果为数组表示进行cookie设置
- * @param mixed $value cookie值
- * @param mixed $option 参数
- * @return mixed
- */
- function cookie($name, $value = '', $option = null)
- {
- if (is_array($name)) {
- // 初始化
- Cookie::init($name);
- } elseif (is_null($name)) {
- // 清除
- Cookie::clear($value);
- } elseif ('' === $value) {
- // 获取
- return 0 === strpos($name, '?') ? Cookie::has(substr($name, 1), $option) : Cookie::get($name, $option);
- } elseif (is_null($value)) {
- // 删除
- return Cookie::delete($name);
- } else {
- // 设置
- return Cookie::set($name, $value, $option);
- }
- }
- }
- if (!function_exists('cache')) {
- /**
- * 缓存管理
- * @param mixed $name 缓存名称,如果为数组表示进行缓存设置
- * @param mixed $value 缓存值
- * @param mixed $options 缓存参数
- * @param string $tag 缓存标签
- * @return mixed
- */
- function cache($name, $value = '', $options = null, $tag = null)
- {
- if (is_array($options)) {
- // 缓存操作的同时初始化
- $cache = Cache::connect($options);
- } elseif (is_array($name)) {
- // 缓存初始化
- return Cache::connect($name);
- } else {
- $cache = Cache::init();
- }
- if (is_null($name)) {
- return $cache->clear($value);
- } elseif ('' === $value) {
- // 获取缓存
- return 0 === strpos($name, '?') ? $cache->has(substr($name, 1)) : $cache->get($name);
- } elseif (is_null($value)) {
- // 删除缓存
- return $cache->rm($name);
- } elseif (0 === strpos($name, '?') && '' !== $value) {
- $expire = is_numeric($options) ? $options : null;
- return $cache->remember(substr($name, 1), $value, $expire);
- } else {
- // 缓存数据
- if (is_array($options)) {
- $expire = isset($options['expire']) ? $options['expire'] : null; //修复查询缓存无法设置过期时间
- } else {
- $expire = is_numeric($options) ? $options : null; //默认快捷缓存设置过期时间
- }
- if (is_null($tag)) {
- return $cache->set($name, $value, $expire);
- } else {
- return $cache->tag($tag)->set($name, $value, $expire);
- }
- }
- }
- }
- if (!function_exists('trace')) {
- /**
- * 记录日志信息
- * @param mixed $log log信息 支持字符串和数组
- * @param string $level 日志级别
- * @return void|array
- */
- function trace($log = '[think]', $level = 'log')
- {
- if ('[think]' === $log) {
- return Log::getLog();
- } else {
- Log::record($log, $level);
- }
- }
- }
- if (!function_exists('request')) {
- /**
- * 获取当前Request对象实例
- * @return Request
- */
- function request()
- {
- return Request::instance();
- }
- }
- if (!function_exists('response')) {
- /**
- * 创建普通 Response 对象实例
- * @param mixed $data 输出数据
- * @param int|string $code 状态码
- * @param array $header 头信息
- * @param string $type
- * @return Response
- */
- function response($data = [], $code = 200, $header = [], $type = 'html')
- {
- return Response::create($data, $type, $code, $header);
- }
- }
- if (!function_exists('view')) {
- /**
- * 渲染模板输出
- * @param string $template 模板文件
- * @param array $vars 模板变量
- * @param array $replace 模板替换
- * @param integer $code 状态码
- * @return \think\response\View
- */
- function view($template = '', $vars = [], $replace = [], $code = 200)
- {
- return Response::create($template, 'view', $code)->replace($replace)->assign($vars);
- }
- }
- if (!function_exists('json')) {
- /**
- * 获取\think\response\Json对象实例
- * @param mixed $data 返回的数据
- * @param integer $code 状态码
- * @param array $header 头部
- * @param array $options 参数
- * @return \think\response\Json
- */
- function json($data = [], $code = 200, $header = [], $options = [])
- {
- return Response::create($data, 'json', $code, $header, $options);
- }
- }
- if (!function_exists('jsonp')) {
- /**
- * 获取\think\response\Jsonp对象实例
- * @param mixed $data 返回的数据
- * @param integer $code 状态码
- * @param array $header 头部
- * @param array $options 参数
- * @return \think\response\Jsonp
- */
- function jsonp($data = [], $code = 200, $header = [], $options = [])
- {
- return Response::create($data, 'jsonp', $code, $header, $options);
- }
- }
- if (!function_exists('xml')) {
- /**
- * 获取\think\response\Xml对象实例
- * @param mixed $data 返回的数据
- * @param integer $code 状态码
- * @param array $header 头部
- * @param array $options 参数
- * @return \think\response\Xml
- */
- function xml($data = [], $code = 200, $header = [], $options = [])
- {
- return Response::create($data, 'xml', $code, $header, $options);
- }
- }
- if (!function_exists('redirect')) {
- /**
- * 获取\think\response\Redirect对象实例
- * @param mixed $url 重定向地址 支持Url::build方法的地址
- * @param array|integer $params 额外参数
- * @param integer $code 状态码
- * @param array $with 隐式传参
- * @return \think\response\Redirect
- */
- function redirect($url = [], $params = [], $code = 302, $with = [])
- {
- if (is_integer($params)) {
- $code = $params;
- $params = [];
- }
- return Response::create($url, 'redirect', $code)->params($params)->with($with);
- }
- }
- if (!function_exists('abort')) {
- /**
- * 抛出HTTP异常
- * @param integer|Response $code 状态码 或者 Response对象实例
- * @param string $message 错误信息
- * @param array $header 参数
- */
- function abort($code, $message = null, $header = [])
- {
- if ($code instanceof Response) {
- throw new HttpResponseException($code);
- } else {
- throw new HttpException($code, $message, null, $header);
- }
- }
- }
- if (!function_exists('halt')) {
- /**
- * 调试变量并且中断输出
- * @param mixed $var 调试变量或者信息
- */
- function halt($var)
- {
- dump($var);
- throw new HttpResponseException(new Response);
- }
- }
- if (!function_exists('token')) {
- /**
- * 生成表单令牌
- * @param string $name 令牌名称
- * @param mixed $type 令牌生成方法
- * @return string
- */
- function token($name = '__token__', $type = 'md5')
- {
- $token = Request::instance()->token($name, $type);
- return '<input type="hidden" name="' . $name . '" value="' . $token . '" />';
- }
- }
- if (!function_exists('load_relation')) {
- /**
- * 延迟预载入关联查询
- * @param mixed $resultSet 数据集
- * @param mixed $relation 关联
- * @return array
- */
- function load_relation($resultSet, $relation)
- {
- $item = current($resultSet);
- if ($item instanceof Model) {
- $item->eagerlyResultSet($resultSet, $relation);
- }
- return $resultSet;
- }
- }
- if (!function_exists('collection')) {
- /**
- * 数组转换为数据集对象
- * @param array $resultSet 数据集数组
- * @return \think\model\Collection|\think\Collection
- */
- function collection($resultSet)
- {
- $item = current($resultSet);
- if ($item instanceof Model) {
- return \think\model\Collection::make($resultSet);
- } else {
- return \think\Collection::make($resultSet);
- }
- }
- }
|