fix: add gql marshal functions for wrappedmap

Signed-off-by: David van der Spek <[email protected]>
pluralsh · Jul 14, 2023 · 1991a1b · 1991a1b
1 parent d7dbdaa
commit 1991a1b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/api/observability/v1alpha1/tempo_limit_types.go b/api/observability/v1alpha1/tempo_limit_types.go
@@ -183,6 +183,21 @@ type WrappedMap struct {
 	Object map[string]interface{} `yaml:",inline" json:",inline"`
 }
 
+func (e *WrappedMap) UnmarshalGQL(v interface{}) error {
+	m, ok := v.(map[string]interface{})
+	if !ok {
+		return fmt.Errorf("%T is not a map", v)
+	}
+
+	*e = WrappedMap{Object: m}
+
+	return nil
+}
+
+func (e WrappedMap) MarshalGQL(w io.Writer) {
+	json.NewEncoder(w).Encode(e)
+}
+
 // MarshalJSON defers JSON encoding to the wrapped map
 func (w *WrappedMap) MarshalJSON() ([]byte, error) {
 	return json.Marshal(w.Object)