如何高效解析.thrift文件提取服务名称及对应函数名称

方案1：Python thriftpy2库实现（推荐） #

这是成本最低的方案，thriftpy2是纯Python实现的Thrift解析库，原生支持直接加载.thrift文件，直接读取结构化的服务、函数元数据，适配绝大多数复杂Thrift语法（包括嵌套引入、自定义结构体、异常定义等场景）。
示例代码如下：

import thriftpy2

# 替换为你的.thrift文件路径
thrift_obj = thriftpy2.load("post_storage.thrift", module_name="post_thrift")

for attr_name in dir(thrift_obj):
    service = getattr(thrift_obj, attr_name)
    # 筛选服务对象
    if hasattr(service, "thrift_services"):
        print(f"服务名：{attr_name}")
        # 遍历服务下所有函数
        for func_name in service.thrift_services:
            print(f"  函数名：{func_name}")

运行后输出结果完全匹配需求：

服务名：PostStorageService
  函数名：StorePost
  函数名：ReadPost
  函数名：ReadPosts

方案2：正则快速提取（无依赖，仅适合格式规整的文件） #

如果你的Thrift文件格式规范、无复杂注释和特殊格式，可以用正则脚本快速提取，不需要安装额外依赖：

# 提取所有服务名
grep "^service" your_file.thrift | awk '{print $2}'

# 提取指定服务下的所有函数名，以PostStorageService为例
sed -n '/service PostStorageService {/,/}/p' your_file.thrift | grep "(" | grep -v "throws" | awk '{print $2}'

注意该方案对格式不规范、有跨行注释的Thrift文件会出现提取误差。

方案3：基于官方Thrift编译器AST解析（适合高度定制化场景） #

你之前用的--gen json参数是生成代码相关的配置描述，无法拿到完整的IDL结构，换用官方编译器的--gen ast参数可以生成完整的语法树JSON文件，只需要写简单脚本遍历JSON结构就能提取需要的服务、函数信息，不需要自己处理Thrift语法解析。

内容的提问来源于stack exchange，提问作者Azad Md Abul Kalam