添加简单的test

gjson.go

@@ -6,6 +6,7 @@ import (
 	"encoding/json"
 	"errors"
 	"reflect"
+	"regexp"
 	"strconv"
 	"strings"
 	"sync"
@@ -706,6 +707,8 @@ type arrayPathResult struct {
 		value string
 		all   bool
 	}
+
+	re *regexp.Regexp
 }
 
 func parseArrayPath(path string) (r arrayPathResult) {
@@ -741,6 +744,7 @@ func parseArrayPath(path string) (r arrayPathResult) {
 							path[i] == '<' ||
 							path[i] == '>' ||
 							path[i] == '%' ||
+							path[i] == '~' ||
 							path[i] == ']' {
 							break
 						}
@@ -776,39 +780,32 @@ func parseArrayPath(path string) (r arrayPathResult) {
 								break
 							}
 						}
-						s = i
+
+						s = i + 1
+						pair := path[i]
+						i++
+					GETREGEXPSTR:
 						for ; i < len(path); i++ {
-							if path[i] == '"' {
-								i++
-								s2 := i
-								for ; i < len(path); i++ {
-									if path[i] > '\\' {
+							if path[i] == pair {
+								for iend := i + 1; iend < len(path); iend++ {
+
+									if path[iend] == ' ' {
 										continue
 									}
-									if path[i] == '"' {
-										// look for an escaped slash
-										if path[i-1] == '\\' {
-											n := 0
-											for j := i - 2; j > s2-1; j-- {
-												if path[j] != '\\' {
-													break
-												}
-												n++
-											}
-											if n%2 == 0 {
-												continue
-											}
+
+									if path[iend] == ']' {
+										if iend+1 < len(path) && path[iend+1] == '#' {
+											r.query.all = true
 										}
+										break GETREGEXPSTR
+									} else {
 										break
 									}
+
 								}
-							} else if path[i] == ']' {
-								if i+1 < len(path) && path[i+1] == '#' {
-									r.query.all = true
-								}
-								break
 							}
 						}
+
 						if i > len(path) {
 							i = len(path)
 						}
@@ -1077,6 +1074,22 @@ func parseObject(c *parseContext, i int, path string) (int, bool) {
 	}
 	return i, false
 }
+
+func escapeRegexpString(rpv string) string {
+	var content []byte
+	lrpv := len(rpv)
+	for i := 0; i < lrpv; i++ {
+		rpvChar := rpv[i]
+		if rpvChar == '\\' {
+			content = append(content, rpv[i+1])
+			i++
+		} else {
+			content = append(content, rpvChar)
+		}
+	}
+	return string(content)
+}
+
 func queryMatches(rp *arrayPathResult, value Result) bool {
 	rpv := rp.query.value
 	if len(rpv) > 2 && rpv[0] == '"' && rpv[len(rpv)-1] == '"' {
@@ -1101,6 +1114,11 @@ func queryMatches(rp *arrayPathResult, value Result) bool {
 			return match.Match(value.Str, rpv)
 		case "!%":
 			return !match.Match(value.Str, rpv)
+		case "~":
+			if rp.re == nil {
+				rp.re = regexp.MustCompile(rpv)
+			}
+			return rp.re.MatchString(value.Str)
 		}
 	case Number:
 		rpvn, _ := strconv.ParseFloat(rpv, 64)

gjson_test.go

@@ -390,6 +390,7 @@ func TestBasic1(t *testing.T) {
 		t.Fatalf("expected %v, got %v", 3, count)
 	}
 }
+
 func TestBasic2(t *testing.T) {
 	mtok := get(basicJSON, `loggy.programmers.#[age=101].firstName`)
 	if mtok.String() != "1002.3" {
@@ -399,14 +400,15 @@ func TestBasic2(t *testing.T) {
 	if mtok.String() != "Jason" {
 		t.Fatalf("expected %v, got %v", "Jason", mtok.String())
 	}
-	mtok = get(basicJSON, `loggy.programmers.#[firstName % "Bre*"].email`)
+	mtok = get(basicJSON, `loggy.programmers.#[firstName % "Bre.*"].email`)
 	if mtok.String() != "aaaa" {
 		t.Fatalf("expected %v, got %v", "aaaa", mtok.String())
 	}
-	mtok = get(basicJSON, `loggy.programmers.#[firstName !% "Bre*"].email`)
+	mtok = get(basicJSON, `loggy.programmers.#[firstName !% "Bre.*"].email`)
 	if mtok.String() != "bbbb" {
 		t.Fatalf("expected %v, got %v", "bbbb", mtok.String())
 	}
+
 	mtok = get(basicJSON, `loggy.programmers.#[firstName == "Brett"].email`)
 	if mtok.String() != "aaaa" {
 		t.Fatalf("expected %v, got %v", "aaaa", mtok.String())
@@ -1431,3 +1433,18 @@ func TestArrayValues(t *testing.T) {
 	}
 
 }
+
+func TestRegexp(t *testing.T) {
+	mtok := get(`{"data": [ {"dat": "123\"", "next": [{"a": "\"32"}, {"a": "32"}]}, {"dat": "234"} ] }`, `data.#[dat ~ "3\""].next.#[a ~ @\"32@]#.a`)
+	if mtok.String() != `["\"32"]` {
+		t.Fatalf("expected %v, got %v", `"32`, mtok.String())
+	}
+}
+
+func TestChinese(t *testing.T) {
+	js := `{"data": [{"f":"\"广告"}, {"f": "广告"}]}`
+	mtok := get(js, `data.#[f=""广告"]#`)
+	if mtok.Array()[0].String() != `{"f":"\"广告"}` {
+		t.Error("error", mtok.Array())
+	}
+}