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

Question

我需要使用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!。

Answer 1

像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)
   )
);