67 lines
2.0 KiB
PHP
67 lines
2.0 KiB
PHP
<?php
|
|
// +----------------------------------------------------------------------
|
|
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2006-2015 http://thinkphp.cn All rights reserved.
|
|
// +----------------------------------------------------------------------
|
|
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
|
// +----------------------------------------------------------------------
|
|
// | Author: yunwuxin <448901948@qq.com>
|
|
// +----------------------------------------------------------------------
|
|
namespace think\helper;
|
|
|
|
use Closure;
|
|
use think\exception\FuncNotFoundException;
|
|
|
|
trait Macroable
|
|
{
|
|
/**
|
|
* 方法注入.
|
|
*
|
|
* @var Closure[]
|
|
*/
|
|
protected static $macro = [];
|
|
|
|
/**
|
|
* 设置方法注入.
|
|
*
|
|
* @param string $method
|
|
* @param Closure $closure
|
|
*
|
|
* @return void
|
|
*/
|
|
public static function macro(string $method, Closure $closure)
|
|
{
|
|
static::$macro[$method] = $closure;
|
|
}
|
|
|
|
/**
|
|
* 检查方法是否已经有注入
|
|
*
|
|
* @param string $name
|
|
* @return bool
|
|
*/
|
|
public static function hasMacro(string $method)
|
|
{
|
|
return isset(static::$macro[$method]);
|
|
}
|
|
|
|
public function __call($method, $args)
|
|
{
|
|
if (!isset(static::$macro[$method])) {
|
|
throw new FuncNotFoundException('method not exists: ' . static::class . '::' . $method . '()', "{static::class}::{$method}");
|
|
}
|
|
|
|
return call_user_func_array(static::$macro[$method]->bindTo($this, static::class), $args);
|
|
}
|
|
|
|
public static function __callStatic($method, $args)
|
|
{
|
|
if (!isset(static::$macro[$method])) {
|
|
throw new FuncNotFoundException('method not exists: ' . static::class . '::' . $method . '()', "{static::class}::{$method}");
|
|
}
|
|
|
|
return call_user_func_array(static::$macro[$method]->bindTo(null, static::class), $args);
|
|
}
|
|
}
|