com.apollographql.federation.graphqljava._Entity Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of federation-graphql-java-support Show documentation

GraphQL Java server support for Apollo Federation

There is a newer version: 5.2.0

Show newest version

package com.apollographql.federation.graphqljava;

import static graphql.schema.GraphQLArgument.newArgument;
import static graphql.schema.GraphQLFieldDefinition.newFieldDefinition;

import graphql.schema.GraphQLFieldDefinition;
import graphql.schema.GraphQLList;
import graphql.schema.GraphQLNonNull;
import graphql.schema.GraphQLTypeReference;
import graphql.schema.GraphQLUnionType;
import java.util.Set;
import org.jetbrains.annotations.NotNull;

public final class _Entity {
  public static final String argumentName = "representations";
  public static final String typeName = "_Entity";
  static final String fieldName = "_entities";

  private _Entity() {}

  // graphql-java will mutate GraphQLTypeReference in-place,
  // so we need to create a new instance every time.
  static GraphQLFieldDefinition field(@NotNull Set typeNames) {
    return newFieldDefinition()
        .name(fieldName)
        .argument(
            newArgument()
                .name(argumentName)
                .type(
                    new GraphQLNonNull(
                        new GraphQLList(
                            new GraphQLNonNull(new GraphQLTypeReference(_Any.typeName)))))
                .build())
        .type(
            new GraphQLNonNull(
                new GraphQLList(
                    GraphQLUnionType.newUnionType()
                        .name(typeName)
                        .possibleTypes(
                            typeNames.stream()
                                .map(GraphQLTypeReference::new)
                                .toArray(GraphQLTypeReference[]::new))
                        .build())))
        .build();
  }
}