diff options
author | Adam Scarr <adam@vektah.net> | 2017-08-07 21:57:12 +1000 |
---|---|---|
committer | Adam Scarr <adam@vektah.net> | 2017-08-07 21:57:12 +1000 |
commit | b4f5fb423e5dec43ec702987dc3ddcb9df317d75 (patch) | |
tree | db505058e801547137a053bb0fb8c5d404f04295 | |
parent | 88aaf567a51e9a0b11322db8a967f54057a7e340 (diff) |
Fast single byte matcher
-rw-r--r-- | parser.go | 15 | ||||
-rw-r--r-- | parser_test.go | 14 |
2 files changed, 28 insertions, 1 deletions
@@ -81,6 +81,21 @@ func ParseString(parser Parserish, input string) (result interface{}, remaining } func Exact(match string) Parser { + if len(match) == 1 { + matchByte := match[0] + return NewParser(match, func(ps *State) Node { + ps.AutoWS() + if ps.Input[ps.Pos] != matchByte { + ps.ErrorHere(match) + return Node{} + } + + ps.Advance(1) + + return Node{Token: match} + }) + } + return NewParser(match, func(ps *State) Node { ps.AutoWS() if !strings.HasPrefix(ps.Get(), match) { diff --git a/parser_test.go b/parser_test.go index c61dcda..594fcae 100644 --- a/parser_test.go +++ b/parser_test.go @@ -49,17 +49,29 @@ func TestParsifyAll(t *testing.T) { } func TestExact(t *testing.T) { - t.Run("success", func(t *testing.T) { + t.Run("success string", func(t *testing.T) { node, ps := runParser("foobar", Exact("fo")) require.Equal(t, "fo", node.Token) require.Equal(t, "obar", ps.Get()) }) + t.Run("success char", func(t *testing.T) { + node, ps := runParser("foobar", Exact("f")) + require.Equal(t, "f", node.Token) + require.Equal(t, "oobar", ps.Get()) + }) + t.Run("error", func(t *testing.T) { _, ps := runParser("foobar", Exact("bar")) require.Equal(t, "bar", ps.Error.Expected) require.Equal(t, 0, ps.Pos) }) + + t.Run("error char", func(t *testing.T) { + _, ps := runParser("foobar", Exact("o")) + require.Equal(t, "o", ps.Error.Expected) + require.Equal(t, 0, ps.Pos) + }) } func TestChars(t *testing.T) { |