Postgres psql扩展：如何在C语言中区分Cypher与SQL子句？

你当前通过前缀字符串匹配区分Cypher和SQL语句的方案，确实存在关键字冲突（如CREATE）、拼写错误无法识别的问题，以下是几种更优的C语言实现思路：

方法1：引入显式语句前缀标记 #

让用户通过固定前缀明确指定语句类型，比如要求Cypher语句以CYPHER 开头（类似psql原生的\copy元命令逻辑）。

实现代码 #

/* 检查是否为显式标记的Cypher语句 */
bool is_cypher = pg_strncasecmp(query_buf->data, "CYPHER ", 7) == 0;
if (is_cypher) {
    // 跳过前缀，获取实际Cypher语句内容
    char *cypher_query = query_buf->data + 7;
    cypherCmdStatus = HandleCypherCmds(scan_state, cond_stack, cypher_query, previous_buf);
    success = cypherCmdStatus != PSQL_CMD_ERROR;
    
    if (cypherCmdStatus == PSQL_CMD_SEND) {
        success = SendQuery(convert_to_psql_command(cypher_query));
    }
} else {
    success = SendQuery(query_buf->data);
}

优缺点 #

• 优点：彻底解决关键字冲突问题，逻辑简单易懂，用户意图完全明确
• 缺点：需要用户额外输入前缀，增加了学习和使用成本

方法2：基于词法扫描的语法特征判断 #

利用Cypher和SQL的语法结构差异，先做轻量词法分析，而非仅匹配前缀：

1. 跳过语句开头的空格、注释，提取第一个有效关键字
2. 对冲突关键字（如CREATE），进一步扫描后续token判断语法特征：比如Cypher的CREATE后通常紧跟(（节点定义），而SQL的CREATE后是TABLE/INDEX等对象类型

实现思路示例 #

// 复用psql现有词法扫描逻辑，提取前两个有效token
ScanToken first_token = scan_first_valid_token(query_buf->data);
ScanToken second_token = scan_next_valid_token();

bool is_cypher = false;
// 无冲突的Cypher关键字直接判定
if (pg_strcasecmp(first_token.val, "MATCH") == 0 || 
    pg_strcasecmp(first_token.val, "OPTIONAL") == 0 ||
    pg_strcasecmp(first_token.val, "EXPLAIN") == 0) {
    is_cypher = true;
} 
// 处理冲突关键字CREATE
else if (pg_strcasecmp(first_token.val, "CREATE") == 0) {
    // 判断后续token是否符合Cypher CREATE的特征
    if (second_token.type == TOKEN_PAREN_OPEN || 
        pg_strcasecmp(second_token.val, "NODE") == 0) {
        is_cypher = true;
    }
}

if (is_cypher) {
    cypherCmdStatus = HandleCypherCmds(scan_state, cond_stack, query_buf, previous_buf);
    success = cypherCmdStatus != PSQL_CMD_ERROR;
    
    if (cypherCmdStatus == PSQL_CMD_SEND) {
        success = SendQuery(convert_to_psql_command(query_buf->data));
    }
} else {
    success = SendQuery(query_buf->data);
}

优缺点 #

• 优点：无需用户额外操作，能精准区分大部分冲突场景
• 缺点：实现复杂度高，需要处理多种语法分支；对拼写错误的识别仍有限

方法3：解析失败自动回退机制 #

保留现有前缀匹配逻辑，但增加错误回退：先尝试Cypher解析，失败后自动转用SQL解析。

实现代码 #

bool tried_cypher = false;
cypherCmdStatus = PSQL_CMD_ERROR;

// 先尝试匹配Cypher关键字并解析
if (pg_strncasecmp(query_buf->data, "MATCH", 5) == 0 ||
    pg_strncasecmp(query_buf->data, "OPTIONAL", 8) == 0 ||
    pg_strncasecmp(query_buf->data, "EXPLAIN", 7) == 0 ||
    pg_strncasecmp(query_buf->data, "CREATE", 6) == 0)
{
    tried_cypher = true;
    cypherCmdStatus = HandleCypherCmds(scan_state, cond_stack, query_buf, previous_buf);
    
    if (cypherCmdStatus != PSQL_CMD_ERROR) {
        success = true;
        if (cypherCmdStatus == PSQL_CMD_SEND) {
            success = SendQuery(convert_to_psql_command(query_buf->data));
        }
    }
}

// 如果Cypher解析失败，尝试SQL解析
if (!success || (tried_cypher && cypherCmdStatus == PSQL_CMD_ERROR)) {
    success = SendQuery(query_buf->data);
}

优缺点 #

• 优点：兼容现有逻辑，无需用户改动使用习惯；能自动处理关键字冲突（比如SQL的CREATE语句会在Cypher解析失败后走SQL解析）
• 缺点：存在两次解析的性能开销；无法区分用户确实想写Cypher但拼写错误的情况（会直接走SQL解析并报错）

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