add regexp.Match for compare

flier · flier · commit bcdeaa84ff82 · 2021-11-23T14:21:57.000+08:00
diff --git a/tests/scan_test.go b/tests/scan_test.go
@@ -2,6 +2,7 @@ package tests
 
 import (
 	"os"
+	"regexp"
 	"strings"
 	"testing"
 
@@ -60,7 +61,7 @@ func BenchmarkBlockScan(b *testing.B) {
 	isRaceBuilder := strings.HasSuffix(testenv(), "-race")
 
 	for _, data := range benchData {
-		p := hyperscan.NewPattern(data.re)
+		p := hyperscan.NewPattern(data.re, hyperscan.MultiLine)
 		db, err := hyperscan.NewBlockDatabase(p)
 		if err != nil {
 			b.Fatalf("compile pattern %s: `%s`, %s", data.name, data.re, err)
@@ -72,7 +73,7 @@ func BenchmarkBlockScan(b *testing.B) {
 		}
 
 		m := func(id uint, from, to uint64, flags uint, context interface{}) error {
-			return nil
+			return hyperscan.ErrUnexpected
 		}
 
 		for _, size := range benchSizes {
@@ -81,7 +82,7 @@ func BenchmarkBlockScan(b *testing.B) {
 			}
 			t := makeText(size.n)
 			b.Run(data.name+"/"+size.name, func(b *testing.B) {
-				b.SetBytes(int64(size.n))
+				b.SetBytes(int64(len(t)))
 				for i := 0; i < b.N; i++ {
 					if err = db.Scan(t, s, m, nil); err != nil {
 						b.Fatalf("match, %s", err)
@@ -91,3 +92,25 @@ func BenchmarkBlockScan(b *testing.B) {
 		}
 	}
 }
+
+func BenchmarkMatch(b *testing.B) {
+	isRaceBuilder := strings.HasSuffix(testenv(), "-race")
+
+	for _, data := range benchData {
+		r := regexp.MustCompile(data.re)
+		for _, size := range benchSizes {
+			if (isRaceBuilder || testing.Short()) && size.n > 1<<10 {
+				continue
+			}
+			t := makeText(size.n)
+			b.Run(data.name+"/"+size.name, func(b *testing.B) {
+				b.SetBytes(int64(size.n))
+				for i := 0; i < b.N; i++ {
+					if r.Match(t) {
+						b.Fatal("match!")
+					}
+				}
+			})
+		}
+	}
+}
