基于AST改写C++变量声明：Query构造匹配与代码转换问题

我正在将旧的Query构造代码：

Query q(
      "SELECT %LC FROM %T WHERE %W",
      getColumnArray(),
      "table_name",
      buildConstraints(id, "abc")
);

重构为链式调用风格：

auto q = QueryBuilder.from("table_name").select(getColumnArray()).where(id,"abc").build();

已通过以下匹配器定位目标代码块：

varDecl(isExpansionInMainFile(),hasType(cxxRecordDecl(hasName("Query"))))

针对三个技术问题解答如下：

1. 是否可以匹配参数中包含"SELECT"的Query构造？ #

可以。在现有匹配器基础上，关联构造函数初始化表达式并检查第一个参数的字符串内容，示例匹配器：

varDecl(
  isExpansionInMainFile(),
  hasType(cxxRecordDecl(hasName("Query"))),
  hasInitializer(
    cxxConstructExpr(
      hasArgument(
        0,
        stringLiteral(contains("SELECT"))
      )
    )
  )
)

通过hasInitializer定位变量的构造调用，hasArgument(0)指向第一个SQL模板参数，stringLiteral(contains("SELECT"))匹配包含目标关键字的字符串字面量。

2. 定位到目标代码块后，如何提取构造函数参数复用至新代码？ #

通过Clang AST节点遍历提取对应参数：

1. 从匹配到的VarDecl节点调用getInit()，获取构造表达式CXXConstructExpr；
2. 调用getArg(i)获取对应索引的参数节点：
   • 索引1：getColumnArray()，用Expr::printPretty()输出代码文本直接复用；
   • 索引2：表名字符串，从StringLiteral提取字符串值用于from()；
   • 索引3：buildConstraints(id, "abc")，可拆解为CallExpr后提取内部参数，或直接输出表达式文本用于where()。

示例伪代码：

if (auto *constructExpr = dyn_cast<CXXConstructExpr>(varDecl->getInit())) {
  // 提取select参数
  Expr *selectArg = constructExpr->getArg(1);
  std::string selectCode;
  llvm::raw_string_ostream oss(selectCode);
  selectArg->printPretty(oss, nullptr, PrintingPolicy(LangOptions()));
  
  // 提取表名
  if (auto *strLit = dyn_cast<StringLiteral>(constructExpr->getArg(2))) {
    std::string tableName = strLit->getString().str();
  }
  
  // 提取where参数
  if (auto *callExpr = dyn_cast<CallExpr>(constructExpr->getArg(3))) {
    Expr *idArg = callExpr->getArg(0);
    Expr *abcArg = callExpr->getArg(1);
    // 输出参数代码文本
    std::string idCode, abcCode;
    llvm::raw_string_ostream idOss(idCode), abcOss(abcCode);
    idArg->printPretty(idOss, nullptr, PrintingPolicy(LangOptions()));
    abcArg->printPretty(abcOss, nullptr, PrintingPolicy(LangOptions()));
  }
}

3. 是否能够获取匹配代码块的包装函数名称？ #

可以。通过getDeclContext()向上遍历AST上下文，找到包含当前VarDecl的外层FunctionDecl：

DeclContext *ctx = varDecl->getDeclContext();
while (ctx && !isa<FunctionDecl>(ctx)) {
  ctx = ctx->getParent();
}
if (auto *funcDecl = dyn_cast<FunctionDecl>(ctx)) {
  std::string funcName = funcDecl->getNameAsString();
}

这段代码会逐层向上查找，直到找到定义该Query变量的函数节点，从而获取函数名称。

内容的提问来源于stack exchange，提问作者LunaticJape