随着项目增多，thrift idl 生成代码的管理也越复杂。此工具用于生成 thrift 的代码，尽量使其脚本化，可配置，自动化。 #!/usr/bin/env php array( 'cms-exmaple' => array( 'gen' => 'php', // 代码目标语言，用于 thrift 的 --gen 选项 'out' => '/path/to/output', // 代码输出目录，用于 thrift 的 --out 选项 'idl_git_url' => 'https://git-url', // idl 是否使用了 git 管理，如果设置，则自动 pull， // 例如 https://github.com/my/thrift-idl.git 'idl_git_pre' => '/src/master', // idl 文件的 git url 前缀。在本例中： // 假设 idl_git_url 为 https://github.com/my/thrift-idl.git // idl 路径为 /path/to/cms.thrift，idl_root_path 为 /root/thrift/idl // 则 git 的全路径为 https://github.com/my/thrift-idl/src/master/path/to/cms.thrift // 对应的本地路径为 /root/thrift/idl/path/to/cms.thrift 'idl_root_path' => '/path/to/idl', // idl 根目录，与 idls 拼接，如果有 git，此目录应当设置为 git 的根目录（含 .git 的目录） 'idls' => array( // idl 所在的目录或文件 '/path/to/idl/1', '/path/to/idl/2', ), ), ),); * * author: https://github.com/heiing * date: 2015-03-06t11:06+08:00 */define(version, 0.1.0); function usage($self, $ln = php_eol) { echo usage: {$self} project-name config-file [idl-git-url]{$ln}; echo project-name project name{$ln}; echo config-file config file path{$ln}; echo idl-git-url idl git url{$ln}; echo {$ln}; echo config-file example: {$ln}; echo 'thrift --gen {gen} -out \{out}\ \{idl}\',{$ln}; echo 'projects' => array({$ln}; echo 'cms-exmaple' => array({$ln}; echo 'gen' => 'php', // 代码目标语言，用于 thrift 的 --gen 选项{$ln}; echo 'out' => '/path/to/output', // 代码输出目录，用于 thrift 的 --out 选项{$ln}; echo 'idl_git_url' => 'https://git-url', // idl 是否使用了 git 管理，如果设置，则自动 pull，{$ln}; echo // 例如 https://github.com/my/thrift-idl.git{$ln}; echo 'idl_git_pre' => '/src/master', // idl 文件的 git url 前缀。在本例中：{$ln}; echo // 假设 idl_git_url 为 https://github.com/my/thrift-idl.git{$ln}; echo // idl 路径为 /path/to/cms.thrift，idl_root_path 为 /root/thrift/idl{$ln}; echo // 则 git 的全路径为 https://github.com/my/thrift-idl/src/master/path/to/cms.thrift{$ln}; echo // 对应的本地路径为 /root/thrift/idl/path/to/cms.thrift{$ln}; echo 'idl_root_path' => '/path/to/idl', // idl 根目录，与 idls 拼接，如果有 git，此目录应当设置为 git 的根目录（含 .git 的目录）{$ln}; echo 'idls' => array( // idl 所在的目录或文件{$ln}; echo '/path/to/idl/1',{$ln}; echo '/path/to/idl/2',{$ln}; echo ),{$ln}; echo ),{$ln}; echo ),{$ln}; echo );{$ln}; echo // -------- end of config-file{$ln}; echo {$ln}; echo usage example:{$ln}; echo 1. {$self} cms-example /root/cthrift/config.php{$ln}; echo 2. {$self} cms-example /root/cthrift/config.php https://github.com/my/thrift-idl/src/master/cms.thrift{$ln}; echo {$ln}; echo good luck{$ln}; echo {$ln}; exit(1);}function error($message, $ln = php_eol) { echo error: {$message}{$ln}; exit(1);} function info($message, $ln = php_eol) { echo {$message}{$ln};}function config($name, $value = null) { static $pool = array(); if ($value === null) { return isset($pool[$name]) ? $pool[$name] : null; } $pool[$name] = $value;}function retend_config($name, $value) { if (!is_array($value) || is_numeric(implode('', array_keys($value)))) { return config($name, $value); } foreach ($value as $n => $v) { retend_config($name . '/' . $n, $v); }}function load_config() { $file = config('/config-file'); info('load config: ' . $file); if (!is_file($file)) { error('config file not exists!'); } $configs = include $file; if (!isset($configs['projects'])) { error('invalid config!'); } if (!isset($configs['projects'][config('/project-name')])) { error('project not set!'); } foreach ($configs as $name => $value) { if (!is_array($value)) { config($name, $value); } else { retend_config($name, $value); } }}function do_command($cmd, $argv, $exit_on_error = true) { foreach ($argv as $name => $value) { $cmd = str_replace('{' . $name . '}', $value, $cmd); } info($cmd); $ret = 0; passthru($cmd, $ret); if ($ret !== 0 && $exit_on_error) { error('faild!'); } return $ret;}function process_project() { $pre = 'projects/' . config('/project-name'); $out = config({$pre}/out); if (null === ($cmd = config('thrift_command'))) { $cmd = 'thrift --gen {gen} ' . ($out ? '--out {out} ' : '') . ' {idl}'; } if (null === ($gen = config({$pre}/gen))) { error('gen not set!'); } if (null === ($path = config({$pre}/idl_root_path))) { error('idl_root_path not set!'); } if (!is_dir($path)) { error('idl_root_path not exists!'); } $path = rtrim($path, '/\\'); $git = rtrim(config({$pre}/idl_git_url), '/'); if (!empty($git)) { do_command(cd {$path}; git pull;, array(), true); } if (null !== ($url = config('/idl-git-url'))) { $git = (strtolower(substr($git, -4)) === '.git' ? substr($git, 0, -4) : $git) . config({$pre}/idl_git_pre); if ($git !== substr($url, 0, strlen($git))) { error('invalid git url!'); } $idls = array(substr($url, strlen($git))); } else if (null === ($idls = config({$pre}/idls)) || empty($idls)) { error('idls not set or empty!'); } foreach ($idls as $idl) { $idl = {$path}{$idl}; if (is_dir($idl)) { $idl_files = glob({$idl}/*.thrift); } else if (is_file($idl)) { $idl_files = array($idl); } else { info(not found: {$idl}); continue; } foreach ($idl_files as $file) { do_command($cmd, array( 'gen' => $gen, 'out' => $out, 'idl' => $file, ), true); } }}function run($argv) { info(thrift creator . version); if (!isset($argv[2])) { usage($argv[0]); } config('/config-file', $argv[2]); config('/project-name', $argv[1]); if (isset($argv[3])) { config('/idl-git-url', $argv[3]); } load_config(); process_project(); info('done.');}run($argv);