zhamao-framework/src/ZM/Utils/ZMUtil.php

<?php


namespace ZM\Utils;


use Exception;
use Swoole\Process;
use ZM\Console\Console;
use ZM\Framework;
use ZM\Store\Lock\SpinLock;
use ZM\Store\ZMAtomic;
use ZM\Store\ZMBuf;

class ZMUtil
{
    /**
     * @throws Exception
     */
    public static function stop() {
        if (SpinLock::tryLock('_stop_signal') === false) return;
        Console::warning(Console::setColor('Stopping server...', 'red'));
        if (Console::getLevel() >= 4) Console::trace();
        ZMAtomic::get('stop_signal')->set(1);
        server()->shutdown();
    }

    /**
     * @throws Exception
     */
    public static function reload() {
        Process::kill(server()->master_pid, SIGUSR1);
    }

    public static function getModInstance($class) {
        if (!isset(ZMBuf::$instance[$class])) {
            //Console::debug('Class instance $class not exist, so I created it.');
            return ZMBuf::$instance[$class] = new $class();
        } else {
            return ZMBuf::$instance[$class];
        }
    }

    /**
     * 在工作进程中返回可以通过reload重新加载的php文件列表
     * @return string[]|string[][]
     */
    public static function getReloadableFiles() {
        return array_map(
            function ($x) {
                return str_replace(DataProvider::getSourceRootDir() . '/', '', $x);
            }, array_diff(
                get_included_files(),
                Framework::$loaded_files
            )
        );
    }

    /**
     * 使用Psr-4标准获取目录下的所有类
     * @param $dir
     * @param $base_namespace
     * @param null|mixed $rule
     * @param bool $return_kv
     * @return String[]
     */
    public static function getClassesPsr4($dir, $base_namespace, $rule = null, $return_kv = false) {
        // 预先读取下composer的file列表
        $composer = json_decode(file_get_contents(DataProvider::getSourceRootDir() . '/composer.json'), true);
        $classes = [];
        // 扫描目录，使用递归模式，相对路径模式，因为下面此路径要用作转换成namespace
        $files = DataProvider::scanDirFiles($dir, true, true);
        foreach ($files as $v) {
            $pathinfo = pathinfo($v);
            if ($pathinfo['extension'] == 'php') {
                if ($rule === null) { //规则未设置回调时候，使用默认的识别过滤规则
                    if (substr(file_get_contents($dir . '/' . $v), 6, 6) == '#plain') continue;
                    elseif (mb_substr($v, 0, 7) == 'global_' || mb_substr($v, 0, 7) == 'script_') continue;
                    foreach (($composer['autoload']['files'] ?? []) as $fi) {
                        if (md5_file(DataProvider::getSourceRootDir().'/'.$fi) == md5_file($dir.'/'.$v)) continue 2;
                    }
                } elseif (is_callable($rule) && !($rule($dir, $pathinfo))) continue;
                $dirname = $pathinfo['dirname'] == '.' ? '' : (str_replace('/', '\\', $pathinfo['dirname']) . '\\');
                $class_name = $base_namespace . '\\' . $dirname . $pathinfo['filename'];
                if ($return_kv) $classes[$class_name] = $v;
                else $classes[] = $class_name;
            }
        }
        return $classes;
    }
}