github.com/dubbogo/gost@v1.14.0/encoding/json/parser.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package json

import (
	"io/ioutil"
	"log"
	"reflect"
	"strconv"
	"strings"
	"time"
)

import (
	"github.com/buger/jsonparser"

	perrors "github.com/pkg/errors"
)

// HessianRegisterPair define the pair to register to hessian
type HessianRegisterPair struct {
	JavaClassName string
	Obj           interface{}
}

// jsonStructParser can use reflect to create arbitrary interface{} of go, from user defined json file.
type jsonStructParser struct {
	structFields        []reflect.StructField
	hessianRegisterPair []HessianRegisterPair
	valueMap            map[string]string
	subObjValueMap      map[string]reflect.Value
}

// newJSONStructParser create a new json struct parser
func newJSONStructParser() *jsonStructParser {
	return &jsonStructParser{
		structFields:        make([]reflect.StructField, 0, 16),
		valueMap:            make(map[string]string, 8),
		hessianRegisterPair: make([]HessianRegisterPair, 0, 16),
		subObjValueMap:      make(map[string]reflect.Value, 8),
	}
}

// File2Interface first read json byte from @filePath, and parse it to interface
func File2Interface(filePath string) ([]HessianRegisterPair, interface{}, error) {
	defer func() {
		defaultJSONStructParser = newJSONStructParser()
	}()
	return defaultJSONStructParser.jsonFilePath2Struct(filePath)
}

func init() {
	defaultJSONStructParser = newJSONStructParser()
}

var defaultJSONStructParser *jsonStructParser

// RemoveTargetNameField remove target file in @v
func RemoveTargetNameField(v interface{}, targetName string) interface{} {
	defer func() {
		defaultJSONStructParser = newJSONStructParser()
	}()
	return defaultJSONStructParser.removeTargetNameField(v, targetName)
}

func (jsp *jsonStructParser) cb(key []byte, value []byte, dataType jsonparser.ValueType, offset int) error {
	switch dataType {
	case jsonparser.Object:
		// parse sub interface, use a new parser to deal with it the same way
		newParser := newJSONStructParser()
		subObj := newParser.json2Struct(value)
		javaClassName, err := getJavaClassName(subObj)
		if err != nil {
			return err
		}
		jsp.hessianRegisterPair = append(jsp.hessianRegisterPair, HessianRegisterPair{
			JavaClassName: javaClassName,
			Obj:           subObj,
		})
		jsp.structFields = append(jsp.structFields, reflect.StructField{
			Name: string(key),
			Type: reflect.TypeOf(subObj),
		})
		jsp.subObjValueMap[string(key)] = reflect.ValueOf(subObj)

	case jsonparser.Array: // TODO slice parse
	case jsonparser.String: // normal struct parse
		// "type@value"
		arr := strings.Split(string(value), "@")
		var userDefinedType reflect.Type
		switch arr[0] {
		case "int":
			userDefinedType = reflect.TypeOf(0)
		case "string":
			userDefinedType = reflect.TypeOf("")
		case "uint64":
			userDefinedType = reflect.TypeOf(uint64(0))
		case "time.Time":
			userDefinedType = reflect.TypeOf(time.Time{})
		case "float32":
			userDefinedType = reflect.TypeOf(float32(0))
		case "float64":
			userDefinedType = reflect.TypeOf(float64(0))
		case "bool":
			userDefinedType = reflect.TypeOf(false)
		default:
			log.Printf("error: dataType %s in json is not supported\n", string(value))
			return perrors.Errorf("dataType %s in json is not supported", string(value))
		}
		if len(arr) > 1 {
			jsp.valueMap[string(key)] = arr[1]
		}
		jsp.structFields = append(jsp.structFields, reflect.StructField{
			Name: string(key),
			Type: userDefinedType,
		})
	default:
		log.Printf("error: dataType %s in json is not supported\n", string(value))
		return perrors.Errorf("dataType %s in json is not supported", string(value))
	}
	return nil
}

// json2Struct parse data from json file to user defined interface
func (jsp *jsonStructParser) json2Struct(jsonData []byte) interface{} {
	// first: call ObjectEach to parse jsonData to reflect.StructField item
	if err := jsonparser.ObjectEach(jsonData, jsp.cb); err != nil {
		log.Println("jsonparser.ObjectEach error = ", err)
	}

	// second: parse structField to reflectType
	typ := reflect.StructOf(jsp.structFields)
	v := reflect.New(typ).Elem()
	newty := reflect.TypeOf(v.Addr().Interface()).Elem()

	// finally: traverse each json field, and set user defined value
	for i := 0; i < typ.NumField(); i++ {
		valStr, ok1 := jsp.valueMap[newty.Field(i).Name]
		subObj, ok2 := jsp.subObjValueMap[newty.Field(i).Name]
		if !ok1 && !ok2 {
			continue
		}

		if newty.Field(i).Type.Kind() == reflect.Ptr {
			v.Field(i).Set(subObj)
			continue
		}
		switch newty.Field(i).Type {
		case reflect.TypeOf(0), reflect.TypeOf(uint64(0)):
			if parsedInt, err := strconv.Atoi(valStr); err == nil {
				v.Field(i).SetInt(int64(parsedInt))
				break
			}
			v.Field(i).SetInt(0)
		case reflect.TypeOf(""):
			v.Field(i).SetString(valStr)
		case reflect.TypeOf(time.Time{}):
			// todo time support v.Field(i).
		case reflect.TypeOf(float64(0)), reflect.TypeOf(float32(0)):
			if parsedFloat, err := strconv.ParseFloat(valStr, 64); err == nil {
				v.Field(i).SetFloat(parsedFloat)
				break
			}
			v.Field(i).SetFloat(0)
		case reflect.TypeOf(false):
			if valStr == "true" || valStr == "1" {
				v.Field(i).SetBool(true)
			}
		default:
			log.Printf("error: val %s in value is not supported\n", valStr)
			return perrors.Errorf("val %s in value is not supported", valStr)
		}
	}
	s := v.Addr().Interface()
	return s
}

// jsonFilePath2Struct read file from @filePath and parse data to interface
func (jsp *jsonStructParser) jsonFilePath2Struct(filePath string) ([]HessianRegisterPair, interface{}, error) {
	jsonData, err := ioutil.ReadFile(filePath)
	if err != nil {
		return []HessianRegisterPair{}, nil, err
	}
	return jsp.hessianRegisterPair, jsp.json2Struct(jsonData), nil
}

// removeTargetNameField remove origin interface @v's target field by @targetName
func (jsp *jsonStructParser) removeTargetNameField(v interface{}, targetName string) interface{} {
	typ := reflect.TypeOf(v).Elem()
	val := reflect.ValueOf(v).Elem()
	nums := val.NumField()
	structFields := make([]reflect.StructField, 0)
	fieldMap := make(map[string]reflect.Value)
	for i := 0; i < nums; i++ {
		if typ.Field(i).Name != targetName {
			structFields = append(structFields, reflect.StructField{
				Name: typ.Field(i).Name,
				Type: typ.Field(i).Type,
			})
			fieldMap[typ.Field(i).Name] = val.Field(i)
		}
	}
	newtyp := reflect.StructOf(structFields)
	newi := reflect.New(newtyp).Elem()
	newty := reflect.TypeOf(newi.Addr().Interface()).Elem()
	for i := 0; i < nums-1; i++ {
		newi.Field(i).Set(fieldMap[newty.Field(i).Name])
	}
	return newi.Addr().Interface()
}

// getJavaClassName can read field JavaClassName of interface{}, used in cli-tool to do hessian registry
func getJavaClassName(pkg interface{}) (string, error) {
	val := reflect.ValueOf(pkg).Elem()
	typ := reflect.TypeOf(pkg).Elem()
	nums := val.NumField()
	for i := 0; i < nums; i++ {
		if typ.Field(i).Name == "JavaClassName" {
			return val.Field(i).String(), nil
		}
	}
	return "", perrors.Errorf("JavaClassName field not found error")
}