tp5.1 权限模块 -- 方法采集

我们在制作站点用户权限auth的过程中，通常需要统计站点的方法action。如果人为的去统计action的话，效率会非常低，所以我们需要写一个自动采集action的方法。

例：

action表设计如下：

image.png

我们在页面中点击 刷新 按钮，便可以将站点的方法action 进行统计，更新、插入action数据表

image.png

代码举例：

（点击刷新按钮，触发事件 --- 后端执行refreshActionList()，此处做忽略。）

刷新采集action方法refreshActionList()如下：

/**
     * 刷新功能列表
     * @return \think\response\Json
     */
    public function refreshActionList()
    {
        //获取action数据表中现有节点
        $nodeUrl = Db::name('action')->column('action_url');
        $node    = new NodeService();   // 实例化采集服务---之后会详细说明

        $modules = $node->getModule();  //采集站点所有模块

        $i      = 0;
        $result = array();
        $moArr  = array('common','Common','services','Services','behavior','Behavior','facade','Facade','push','Push');//待过滤模块组

        foreach ($modules as $module) {
            if(!in_array($module, $moArr)){  //过滤模块
                $all_controller = $node->getController($module);  // 采集当前模块所有控制器
                if(!$all_controller) continue;

                foreach ($all_controller as $controller) {
                    $controller_name = $module.'/'.$controller;
                    $all_action = $node->getAction($controller_name);  // 采集当前控制器下所有 action

                    foreach ($all_action as $action) {
                        $str = $module.'/'.$controller.'/'.$action;
                        //如果当前节点url不在数据库中,说明是一个新节点需要添加到数据库
                        if(!in_array($str, $nodeUrl)){
                            $result[$i]['action_url'] = $str;
                            $result[$i]['module']     = $module;
                            $i++;
                        }
                    }
                }
            }
        }

        $tran_result = true;
        Db::startTrans();//开启事务
        try {// 异常处理
            //将新节点批量插入到 action数据表 中
            Db::name('action')->insertAll($result);
        } catch (Exception $ex) {
            $tran_result = false;
            // 记录日志
            Log::record($ex->getMessage(), 'DEBUG');
        }

        if ($tran_result === false) {
            Db::rollback();
            $data['code'] = 0; //更新失败
            $data['msg']    = '刷新失败';
            Log::record("refresh:刷新失败", 'operate');
        } else {
            Db::commit();
            $data['code'] = 1; //更新成功
            $data['msg']    = '刷新成功';
            Log::record("refresh:刷新成功", 'operate');
        }

        return json($data);
    }

在此我们将采集 model、controller、action 的方法封装成服务 NodeService.php

image.png
NodeService.php代码如下：

<?php
namespace app\services;
use think\facade\Env;

class NodeService
{
    public function __construct(){
        //echo 'action';
    }

    //获取应用下的模块名称
    public function getModule(){
        $dir = array();
        define('APP_PATH', Env::get('app_path'));  // 定义应用目录
        $path = APP_PATH . '/*/';
        $dir_name = glob($path);  // glob() 函数返回匹配指定模式的文件名或目录
        foreach ($dir_name as $value){
            array_push($dir, basename($value));  // basename() 函数返回路径中的文件名部分。
        }
        return $dir;
    }

    //获取所有控制器名称
    public function getController($module){
        if(empty($module)) return false;
        $module_path = APP_PATH . '/' . $module . '/controller/';  //控制器路径
        if(!is_dir($module_path)) return false;
        $module_path .= '/*.php';
        
        $ary_files = glob($module_path);
        $files     = array();
        foreach ($ary_files as $file) {
            if (is_dir($file)) {
                continue;
            }else {
                $files[] = basename($file, '.php');
            }
        }
        return $files;
    }

    //获取所有方法名称
    public function getAction($controller){
        if(empty($controller)) return false;

        $con = controller($controller);  // 实例化 controller

        $functions = get_class_methods($con);  //get_class_methods()返回由类的方法名组成的数组
        $customer_functions = [];
        //定义需排除的方法
        $inherents_functions = array('_initialize','__construct','getActionName','isAjax','display','show','fetch','buildHtml','assign','theme','__set','get','__get','__isset','__call','error','success','ajaxReturn','redirect','__destruct', '_empty');

        foreach ($functions as $func){
            if(!in_array($func, $inherents_functions)){
                $customer_functions[] = $func;
            }
        }
        return $customer_functions;
    }
}

至此：当我们在页面上点击刷新按钮，调用refreshActionList()方法，便可以更新站点的所有 action

注：转载请注明出处，尊重原创。欢迎大家来简书关注笔者。也更加感谢大家的打赏与支持。谢谢！