Use proper type switch in evalProjectionArray.

Author: Dragomir-Ivanov

schema/query/projection_evaluator.go

@@ -93,14 +93,15 @@ func evalProjectionArray(ctx context.Context, pf ProjectionField, payload []inte
 		name = pf.Alias
 	}
 
-	if ref, ok := validator.(*schema.Reference); ok {
+	switch fieldType := validator.(type) {
+	case *schema.Reference:
 		// Execute sub-request in batch
 		e := &In{Field: "id", Values: payload}
 		q := &Query{
 			Projection: pf.Children,
 			Predicate:  Predicate{e},
 		}
-		rbr.request(ref.Path, q, func(payloads []map[string]interface{}, validator schema.Validator) error {
+		rbr.request(fieldType.Path, q, func(payloads []map[string]interface{}, validator schema.Validator) error {
 			var v interface{}
 			var err error
 			for i := range payloads {
@@ -119,14 +120,14 @@ func evalProjectionArray(ctx context.Context, pf ProjectionField, payload []inte
 			resMu.Unlock()
 			return nil
 		})
-	} else if dict, ok := validator.(*schema.Dict); ok {
+	case schema.FieldGetter:
 		for _, val := range payload {
 			subval, ok := val.(map[string]interface{})
 			if !ok {
-				return nil, fmt.Errorf("%s: invalid value: not a dict", pf.Name)
+				return nil, fmt.Errorf("%s: invalid value: not a dict/object", pf.Name)
 			}
 			var err error
-			if subval, err = evalProjection(ctx, pf.Children, subval, dict, rbr); err != nil {
+			if subval, err = evalProjection(ctx, pf.Children, subval, fieldType, rbr); err != nil {
 				return nil, fmt.Errorf("%s.%v", pf.Name, err)
 			}
 			var v interface{}
@@ -135,12 +136,12 @@ func evalProjectionArray(ctx context.Context, pf ProjectionField, payload []inte
 			}
 			res = append(res, v)
 		}
-	} else if array, ok := validator.(*schema.Array); ok {
+	case *schema.Array:
 		for i, val := range payload {
 			if subval, ok := val.([]interface{}); ok {
 				var err error
 				var subvalp *[]interface{}
-				if subvalp, err = evalProjectionArray(ctx, pf, subval, &array.Values, rbr); err != nil {
+				if subvalp, err = evalProjectionArray(ctx, pf, subval, &fieldType.Values, rbr); err != nil {
 					return nil, fmt.Errorf("%s: error applying projection on array item #%d: %v", pf.Name, i, err)
 				}
 				var v interface{}
@@ -152,26 +153,8 @@ func evalProjectionArray(ctx context.Context, pf ProjectionField, payload []inte
 				return nil, fmt.Errorf("%s. is not an array", pf.Name)
 			}
 		}
-	} else {
-		for _, val := range payload {
-			subval, ok := val.(map[string]interface{})
-			if !ok {
-				return nil, fmt.Errorf("%s: invalid value: not a dict", pf.Name)
-			}
-			if fg, ok := validator.(schema.FieldGetter); ok {
-				var err error
-				if subval, err = evalProjection(ctx, pf.Children, subval, fg, rbr); err != nil {
-					return nil, fmt.Errorf("%s.%v", pf.Name, err)
-				}
-				var v interface{}
-				if v, err = resolveFieldHandler(ctx, pf, def, subval); err != nil {
-					return nil, err
-				}
-				res = append(res, v)
-			} else {
-				return nil, fmt.Errorf("%s. is not an object", pf.Name)
-			}
-		}
+	default:
+		return nil, fmt.Errorf("%s. unknown field type", pf.Name)
 	}
 
 	return resp, nil