unified config loader

common/loader/json_conf.go

@@ -0,0 +1,61 @@
+// +build json
+
+package loader
+
+import (
+	"encoding/json"
+
+	"github.com/v2ray/v2ray-core/common/log"
+)
+
+type JSONConfigLoader struct {
+	*BaseConfigLoader
+	idKey     string
+	configKey string
+}
+
+func NewJSONConfigLoader(idKey string, configKey string) *JSONConfigLoader {
+	return &JSONConfigLoader{
+		idKey:            idKey,
+		configKey:        configKey,
+		BaseConfigLoader: NewBaseConfigLoader(),
+	}
+}
+
+func (this *JSONConfigLoader) LoadWithID(raw []byte, id string) (interface{}, error) {
+	config, err := this.CreateConfig(id)
+	if err != nil {
+		return nil, err
+	}
+
+	if err := json.Unmarshal(raw, config); err != nil {
+		return nil, err
+	}
+	return config, nil
+}
+
+func (this *JSONConfigLoader) Load(raw []byte) (interface{}, error) {
+	obj := make(map[string]json.RawMessage)
+	if err := json.Unmarshal(raw, obj); err != nil {
+		return nil, err
+	}
+	rawID, found := obj[this.idKey]
+	if !found {
+		log.Error(this.idKey, " not found in JSON content.")
+		return nil, ErrConfigIDKeyNotFound
+	}
+	var id string
+	if err := json.Unmarshal(rawID, id); err != nil {
+		return nil, err
+	}
+	rawConfig := json.RawMessage(raw)
+	if len(this.configKey) > 0 {
+		configValue, found := obj[this.configKey]
+		if !found {
+			log.Error(this.configKey, " not found in JSON content.")
+			return nil, ErrConfigIDKeyNotFound
+		}
+		rawConfig = configValue
+	}
+	return this.LoadWithID([]byte(rawConfig), id)
+}