Add extensions to result

Author: Fontinalis

types.go

@@ -6,11 +6,23 @@ import (
 
 // type Schema interface{}
 
+// Result has the response, errors and extensions from the resolved schema
 type Result struct {
-	Data   interface{}                `json:"data"`
-	Errors []gqlerrors.FormattedError `json:"errors,omitempty"`
+	Data       interface{}                `json:"data"`
+	Errors     []gqlerrors.FormattedError `json:"errors,omitempty"`
+	Extensions map[string]interface{}     `json:"extensions,omitempty"`
 }
 
+// HasErrors just a simple function to help you decide if the result has errors or not
 func (r *Result) HasErrors() bool {
 	return len(r.Errors) > 0
 }
+
+func (r *Result) addExtensionResults(p *ExecuteParams) {
+	r.Extensions = map[string]interface{}{}
+	for _, ext := range p.Schema.extensions {
+		if ext.HasResult() {
+			r.Extensions[ext.Name()] = ext.GetResult(p.Context)
+		}
+	}
+}