SCTMES_V5/mes_in_sct/support/helpers.php

<?php

use libs\core\Config;
use libs\core\Container;
use libs\db\Db;
use Workerman\Worker;

define('BASE_PATH', dirname(__DIR__));

/**
 *  * envs环境参数获取
 *  * @param $key
 *  * @param null $default
 *  * @return mixed|null
 *  */
function envs($key, $default = null)
{
    
    static $env_config = [];
    $env = getenv('APP_ENV');
    $file_path = "";

    if ($env === 'DEV') {
        $file_path = config_path() . '/.env.development.php';
    } else if ($env) {
        $file_path = config_path() . "/.env.$env.php";
    } else {
        $file_path = config_path() . "/.env.php";
    }
    if (!file_exists($file_path)) {
        print_r("不存在env文件：$file_path");
        exit;
    }
    $env_config = include $file_path;
    return $env_config[$key] ?? $default;
}

/**
 * if the param $path equal false,will return this program current execute directory
 * @param string|false $path
 * @return string
 */
function base_path($path = ''): string
{
    if (false === $path) {
        return run_path();
    }
    return path_combine(BASE_PATH, $path);
}

/**
 * Config path
 * @param string $path
 * @return string
 */
function config_path(string $path = ''): string
{
    return path_combine(BASE_PATH . DIRECTORY_SEPARATOR . 'config', $path);
}

/**
 * app path
 * @param string $path
 * @return string
 */
function app_path(string $path = ''): string
{
    return path_combine(BASE_PATH . DIRECTORY_SEPARATOR . 'app', $path);
}

/**
 * Get config
 * @param string|null $key
 * @param $default
 * @return array|mixed|null
 */
function config(string $key = null, $default = null)
{
    return Config::get($key, $default);
}

/**
 * Generate paths based on given information
 * @param string $front
 * @param string $back
 * @return string
 */
function path_combine(string $front, string $back): string
{
    return $front . ($back ? (DIRECTORY_SEPARATOR . ltrim($back, DIRECTORY_SEPARATOR)) : $back);
}

/**
 * @param $action
 * @param int $errcode
 * @param string $errmsg
 * @throws Exception
 */
function error($action, $errcode = 4001, $errmsg = '')
{
    throw new \Exception( $errmsg , $errcode);
}

/**
 * 查看可选筛选字段是否满足
 * @param array $rule  验证规则，['tray', 'string', '托盘号']  位置1：验证的字段，位置2：验证字段类型，位置3：提示信息
 * @param array $param 原数据
 * @return array
 */
function check_valid($action, array $rule, array $param)
{
    if (!is_array($rule)) {
        error($action, 4001, '数据验证失败');
    }

    $not_param = [];  // 定义没有传入的值

    foreach ($rule as $item) {
        // 检测数据格式
        if (isset($item[1]) && !empty($item[1])) {
            //判断是否有传入该字段
            if (!isset($param[$item[0]])) {
                array_push($not_param, $item[0]);
                continue;
            }
            // 判断是否有提示参数传入
            $tip = '';
            if (isset($item[2]) && !empty($item[2])) {
                $tip = $item[2];
            }
            switch ($item[1]) {
                case 'float': //如果要判断数据是否为float
                    if ((string)floatval($param[$item[0]]) <> $param[$item[0]]) {
                        error($action, 4001, "[{$item[0]}]{$tip}参数类型不符合，系统需求为浮点型[float]类型");
                    }
                    break;
                case 'string': //如果要判断数据是否为字符串
                    if (empty($param[$item[0]])) {
                        error($action, 4001, "[{$item[0]}]{$tip}参数不为空");
                    }
                    if (!is_string($param[$item[0]]) || strlen($param[$item[0]]) == 0) {
                        error($action, 4001, "[{$item[0]}]{$tip}参数类型不符合，系统需求为字符串[string]类型");
                    }
                    break;
                case 'array': //如果要判断数据是否为数组
                    if (!is_array($param[$item[0]])) {
                        error($action, 4001, "[{$item[0]}]{$tip}参数类型不符合，系统需求为数组[array]类型");
                    }
                    if (count($param[$item[0]]) == 0) {
                        error($action, 4001, "[{$item[0]}]{$tip}参数不为空");
                    }
                    break;
                case 'int': //如果要判断数据是否为整数
                    if (!is_int($param[$item[0]])) {
                        error($action, 4001, "[{$item[0]}]{$tip}参数类型不符合，系统需求为整数[int]类型");
                    }
                    break;
            }
        }
        if (isset($param[$item[0]]) && !empty($param[$item[0]])) {
            if (!is_array($param[$item[0]])) { //如果是数组，跳过trim
                $param[$item[0]] = trim($param[$item[0]]);
            }
        }
    }

    //判断是否有未传入字段
    if (!empty($not_param)) {
        error($action, 4001, '请上传：' . implode(',', $not_param) . '参数');
    }

    unset($rule);

    return $param;
}

/**
 * Start worker
 * @param $process_name
 * @param $config
 * @return void
 */
function worker_start($process_name, $config)
{
    $worker = new Worker($config['listen'] ?? null, $config['context'] ?? []);
    $property_map = [
        'count',
        'user',
        'group',
        'reloadable',
        'reusePort',
        'transport',
        'protocol',
    ];
    $worker->name = $process_name;
    foreach ($property_map as $property) {
        if (isset($config[$property])) {
            $worker->$property = $config[$property];
        }
    }

    $worker->onWorkerStart = function ($worker) use ($config) {
        // 加载数据库实例
        Db::instance();

        if (isset($config['handler'])) {
            if (!\class_exists($config['handler'])) {
                echo "process error: class {$config['handler']} not exists\r\n";
                return;
            }

            $instance = Container::make($config['handler'], $config['constructor'] ?? []);
            \worker_bind($worker, $instance);
        }
    };
}

/**
 * Bind worker
 * @param $worker
 * @param $class
 */
function worker_bind($worker, $class)
{
    $callback_map = [
        'onConnect',
        'onMessage',
        'onClose',
        'onError',
        'onBufferFull',
        'onBufferDrain',
        'onWorkerStop',
        'onWebSocketConnect'
    ];
    foreach ($callback_map as $name) {
        if (\method_exists($class, $name)) {
            $worker->$name = [$class, $name];
        }
    }
    if (\method_exists($class, 'onWorkerStart')) {
        \call_user_func([$class, 'onWorkerStart'], $worker);
    }
}


function record_sql_log($msg,$sql){
    $time = date('Y-m-d H:i:s',time());
    $dir = '/home/mes/log/workerman/' . 'sql-' . date('Y') . '-' . date('m') . '-' . date('d') . '.log';
    $dir_name = dirname($dir);
    if(!file_exists($dir_name)){
        $res = mkdir(iconv("UTF-8", "GBK", $dir_name),0777,true);
    }
    $fp = fopen($dir,"a");
    $data = "[{$time}] sql.DEBUG：{$sql} {messge:{$msg}}";
    //fwrite($fp,var_export($data,true)."\r\n");//写入文件
    fwrite($fp,$data."\r\n");//写入文件
    fclose($fp);
}