add unmarshal for exportertype

ava-labs · Nov 22, 2024 · 62583c1 · 62583c1
1 parent 5f70e7b
commit 62583c1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/trace/exporter_type.go b/trace/exporter_type.go
@@ -33,6 +33,16 @@ func (t ExporterType) MarshalJSON() ([]byte, error) {
 	return []byte(`"` + t.String() + `"`), nil
 }
 
+func (t *ExporterType) UnmarshalJSON(b []byte) error {
+	exporterTypeStr := strings.Trim(string(b), `"`)
+	exporterType, err := ExporterTypeFromString(exporterTypeStr)
+	if err != nil && !errors.Is(err, errUnknownExporterType) {
+		return err
+	}
+	*t = exporterType
+	return nil
+}
+
 func (t ExporterType) String() string {
 	switch t {
 	case GRPC: