add ReadRawMessage

Author: Tao Wen

gen/main.go

@@ -264,6 +264,13 @@ func genDecodeStmt(node ast.Node, ptr string) {
 		_f("    "+genAnonymousStruct(x), ptr)
 	case *ast.MapType:
 		_f("    "+genAnonymousMap(x), ptr)
+	case *ast.InterfaceType:
+		if nodeToString(node) == "interface{}" {
+			_f("    iter.ReadInterface(%s)", ptr)
+		} else {
+			reportError(fmt.Errorf("unknown type: %s", nodeToString(node)))
+			return
+		}
 	default:
 		reportError(fmt.Errorf("unknown type: %s", nodeToString(node)))
 		return

iter_skip.go

@@ -93,6 +93,16 @@ func (iter *Iterator) skipString() error {
 	return iter.ReportError("skipString", "incomplete string")
 }
 
+func (iter *Iterator) ReadRawMessage(out *RawMessage) (ret error) {
+	before := iter.head
+	ret = iter.Skip()
+	if ret == nil {
+		after := iter.head
+		*out = iter.buf[before:after]
+	}
+	return
+}
+
 // Skip skips a json object and positions to relatively the next json object
 func (iter *Iterator) Skip() error {
 	return iter.skip(iter.nextToken())

iter_tests/skip_test.go

@@ -34,6 +34,15 @@ func Test_skip(t *testing.T) {
 	}
 }
 
+func Test_read_raw_message(t *testing.T) {
+	var msg jsoniter.RawMessage
+	iter := jsoniter.ParseBytes([]byte(`"hello",100`))
+	iter.ReadRawMessage(&msg)
+	if string(msg) != `"hello"` {
+		t.Fatal()
+	}
+}
+
 func Test_skip_null(t *testing.T) {
 	parseNull := func() *jsoniter.Iterator {
 		return jsoniter.ParseBytes([]byte(`null`))

value_tests/NamedStruct.go

@@ -5,6 +5,6 @@ import "bytes"
 //go:generate go run github.com/json-iterator/tinygo/gen
 type NamedStruct struct {
 	Name         string
-	Price        int
+	Price        interface{}
 	privateField *bytes.Buffer
 }

value_tests/NamedStruct_json.go

@@ -10,7 +10,7 @@ func NamedStruct_json_unmarshal(iter *jsoniter.Iterator, out *NamedStruct) {
     case field == `Name`:
     iter.ReadString(&(*out).Name)
     case field == `Price`:
-    iter.ReadInt(&(*out).Price)
+    iter.ReadInterface(&(*out).Price)
     default:
       iter.Skip()
     }