时间戳

首页 > 解决方案 > 如何使用 nom 解析 Redis RESP 批量字符串？

rust - 如何使用 nom 解析 Redis RESP 批量字符串？

问题描述

我需要使用 nom 来解析RESP 请求/回复。当我来到批量字符串时，例如

"$6\r\nfoobar\r\n"

或者

$-1\r\n

首先，我编写函数从数据中提取len。

named!(signed_digits<&str, (Option<&str>, &str)>,
   pair!(
      opt!(alt!(tag!("+") | tag!("-"))),
      nom::digit
   )
);

named!(signed_integer<&str, i64>,
   map_res!(recognize!(signed_digits), FromStr::from_str)
);

named!(get_len_in_bulk_string<&str, i64>,
   do_parse!(
      tag!("$") >>
      len: signed_integer >>
      tag!("\r\n") >>
      (len)
   )
);

然后我根据len得到原始字符串：

named!(parse_bulk_string<&str, Record>,
  map_res!(gen_len_in_bulk_string, |n|{
    if n < 0 {
        Record::BulkString(None)
    } else {
        Record::BulkString(Some(take!(n)))
    }
  })
);

但我得到一个编译错误：

Record::BulkString(Some(take!(n)))
                             ^ missing tokens in macro arguments

如何根据之前从文本中提取的len获取原始字符串？好像我不能take!在自己的闭包中使用。

标签： rustnom

解决方案

像这样的宏take!在第一个位置需要一个“隐式”参数：要解析的字符串。通常你看不到它，因为它是在嵌套在另一个 nom 宏中时隐式传递的。

但是，在这里，您是直接“调用”它，因此它明确需要此参数。

相反，您可以执行以下操作：

named!(get_bulk_string<&str, &str>,
   do_parse!(
      tag!("$") >>
      len: signed_integer >>
      string: take!(len) >>
      tag!("\r\n") >>
      (string)
   )
);

当然，这忽略了这种-1情况，你可以用一个开关来处理：

named!(get_bulk_string<&str, Option<&str>>,
   do_parse!(
      tag!("$") >>
      string: switch!(signed_integer,
        -1 => map!(take!(0), |_| None) |
        _ => map!(take!(42), |s| Some(s))
      ) >>
      tag!("\r\n") >>
      (string)
   )
);

推荐阅读