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

时间:2019-02-20 16:00:09

标签: rust nom

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

1 个答案:

答案 0 :(得分: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)
   )
);