GraphQL với Spring for GraphQL: Hướng Dẫn Toàn Diện

<!-- pom.xml -->
<dependencies>
    <!-- Spring for GraphQL -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-graphql</artifactId>
    </dependency>
 
    <!-- Web layer -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
 
    <!-- JPA cho database -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-jpa</artifactId>
    </dependency>
 
    <!-- Security -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-security</artifactId>
    </dependency>
 
    <!-- Driver PostgreSQL -->
    <dependency>
        <groupId>org.postgresql</groupId>
        <artifactId>postgresql</artifactId>
        <scope>runtime</scope>
    </dependency>
 
    <!-- GraphQL testing -->
    <dependency>
        <groupId>org.springframework.graphql</groupId>
        <artifactId>spring-graphql-test</artifactId>
        <scope>test</scope>
    </dependency>
 
    <!-- WebSocket cho subscriptions (tùy chọn) -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-websocket</artifactId>
    </dependency>
</dependencies>

# application.yml
spring:
  graphql:
    graphiql:
      enabled: true           # Bật GraphiQL browser IDE
      path: /graphiql
    path: /graphql            # Đường dẫn GraphQL endpoint
    schema:
      locations: classpath:graphql/  # Vị trí schema files
      file-extensions: .graphqls, .gqls
    websocket:
      path: /graphql-ws       # WebSocket path cho subscriptions
 
  datasource:
    url: jdbc:postgresql://localhost:5432/blogdb
    username: postgres
    password: postgres
 
  jpa:
    hibernate:
      ddl-auto: validate
    show-sql: false

# src/main/resources/graphql/schema.graphqls
 
# Scalar types
scalar DateTime
 
# ── Types ──────────────────────────────────
 
type Author {
    id: ID!
    name: String!
    email: String!
    bio: String
    posts: [Post!]!
    postCount: Int!
    createdAt: DateTime!
}
 
type Post {
    id: ID!
    title: String!
    content: String!
    published: Boolean!
    author: Author!
    comments: [Comment!]!
    tags: [String!]!
    viewCount: Int!
    createdAt: DateTime!
    updatedAt: DateTime!
}
 
type Comment {
    id: ID!
    content: String!
    author: Author!
    post: Post!
    createdAt: DateTime!
}
 
# ── Input Types ────────────────────────────
 
input CreatePostInput {
    title: String!
    content: String!
    tags: [String!]
}
 
input UpdatePostInput {
    title: String
    content: String
    tags: [String!]
    published: Boolean
}
 
input CreateCommentInput {
    postId: ID!
    content: String!
}
 
# ── Pagination ─────────────────────────────
 
type PostConnection {
    edges: [PostEdge!]!
    pageInfo: PageInfo!
    totalCount: Int!
}
 
type PostEdge {
    node: Post!
    cursor: String!
}
 
type PageInfo {
    hasNextPage: Boolean!
    hasPreviousPage: Boolean!
    startCursor: String
    endCursor: String
}
 
# ── Query ──────────────────────────────────
 
type Query {
    # Records đơn lẻ
    author(id: ID!): Author
    post(id: ID!): Post
    me: Author
 
    # Collections
    authors: [Author!]!
    posts(
        published: Boolean
        authorId: ID
        tag: String
        first: Int = 10
        after: String
    ): PostConnection!
    searchPosts(query: String!): [Post!]!
}
 
# ── Mutation ───────────────────────────────
 
type Mutation {
    createPost(input: CreatePostInput!): Post!
    updatePost(id: ID!, input: UpdatePostInput!): Post!
    deletePost(id: ID!): Boolean!
    publishPost(id: ID!): Post!
 
    createComment(input: CreateCommentInput!): Comment!
    deleteComment(id: ID!): Boolean!
}
 
# ── Subscription ───────────────────────────
 
type Subscription {
    commentAdded(postId: ID!): Comment!
    postPublished: Post!
}

// Author.java
@Entity
@Table(name = "authors")
public class Author {
 
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
 
    @Column(nullable = false)
    private String name;
 
    @Column(nullable = false, unique = true)
    private String email;
 
    private String bio;
 
    @OneToMany(mappedBy = "author",
        fetch = FetchType.LAZY)
    private List<Post> posts = new ArrayList<>();
 
    @Column(name = "created_at")
    private Instant createdAt = Instant.now();
 
    // Getters/setters
}
 
// Post.java
@Entity
@Table(name = "posts")
public class Post {
 
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
 
    @Column(nullable = false)
    private String title;
 
    @Column(nullable = false, columnDefinition = "TEXT")
    private String content;
 
    private boolean published = false;
 
    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "author_id", nullable = false)
    private Author author;
 
    @OneToMany(mappedBy = "post",
        cascade = CascadeType.ALL,
        fetch = FetchType.LAZY)
    private List<Comment> comments = new ArrayList<>();
 
    @ElementCollection
    @CollectionTable(name = "post_tags",
        joinColumns = @JoinColumn(name = "post_id"))
    @Column(name = "tag")
    private List<String> tags = new ArrayList<>();
 
    @Column(name = "view_count")
    private int viewCount = 0;
 
    @Column(name = "created_at")
    private Instant createdAt = Instant.now();
 
    @Column(name = "updated_at")
    private Instant updatedAt = Instant.now();
 
    // Getters/setters
}
 
// Comment.java
@Entity
@Table(name = "comments")
public class Comment {
 
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
 
    @Column(nullable = false, columnDefinition = "TEXT")
    private String content;
 
    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "author_id", nullable = false)
    private Author author;
 
    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "post_id", nullable = false)
    private Post post;
 
    @Column(name = "created_at")
    private Instant createdAt = Instant.now();
 
    // Getters/setters
}

// AuthorRepository.java
public interface AuthorRepository
        extends JpaRepository<Author, Long> {
 
    Optional<Author> findByEmail(String email);
 
    @Query("""
        SELECT a FROM Author a
        WHERE a.id IN :ids
        """)
    List<Author> findAllByIds(
        @Param("ids") Collection<Long> ids);
}
 
// PostRepository.java
public interface PostRepository
        extends JpaRepository<Post, Long> {
 
    Page<Post> findByPublished(
        boolean published, Pageable pageable);
 
    @Query("""
        SELECT p FROM Post p
        WHERE LOWER(p.title) LIKE LOWER(:query)
           OR LOWER(p.content) LIKE LOWER(:query)
        """)
    List<Post> searchByQuery(
        @Param("query") String query);
 
    @Query("""
        SELECT p FROM Post p
        WHERE :tag MEMBER OF p.tags
        """)
    List<Post> findByTag(
        @Param("tag") String tag);
 
    List<Post> findByAuthorId(Long authorId);
}
 
// CommentRepository.java
public interface CommentRepository
        extends JpaRepository<Comment, Long> {
 
    List<Comment> findByPostId(Long postId);
 
    @Query("""
        SELECT c FROM Comment c
        WHERE c.post.id IN :postIds
        """)
    List<Comment> findByPostIds(
        @Param("postIds") Collection<Long> postIds);
}

@Controller
public class AuthorController {
 
    @Autowired
    private AuthorService authorService;
 
    // Resolve Query.author
    @QueryMapping
    public Optional<Author> author(
            @Argument Long id) {
        return authorService.findById(id);
    }
 
    // Resolve Query.authors
    @QueryMapping
    public List<Author> authors() {
        return authorService.findAll();
    }
 
    // Resolve Query.me (yêu cầu xác thực)
    @QueryMapping
    @PreAuthorize("isAuthenticated()")
    public Author me(
            @AuthenticationPrincipal
            UserDetails userDetails) {
        return authorService
            .findByEmail(userDetails.getUsername())
            .orElseThrow(() ->
                new GraphQLException("Author not found"));
    }
 
    // Resolve Author.postCount — computed field
    @SchemaMapping(typeName = "Author",
        field = "postCount")
    public int postCount(Author author) {
        return author.getPosts().size();
    }
}

@Controller
public class PostController {
 
    @Autowired
    private PostService postService;
 
    @QueryMapping
    public Optional<Post> post(@Argument Long id) {
        return postService.findById(id);
    }
 
    // Cursor-based pagination
    @QueryMapping
    public PostConnection posts(
            @Argument Boolean published,
            @Argument Long authorId,
            @Argument String tag,
            @Argument Integer first,
            @Argument String after) {
 
        return postService.findPosts(
            published, authorId, tag, first, after);
    }
 
    @QueryMapping
    public List<Post> searchPosts(
            @Argument String query) {
        return postService.search(query);
    }
}

@Controller
public class PostMutationController {
 
    @Autowired
    private PostService postService;
 
    @MutationMapping
    @PreAuthorize("isAuthenticated()")
    public Post createPost(
            @Argument CreatePostInput input,
            @AuthenticationPrincipal
            UserDetails userDetails) {
 
        return postService.createPost(
            input, userDetails.getUsername());
    }
 
    @MutationMapping
    @PreAuthorize("isAuthenticated()")
    public Post updatePost(
            @Argument Long id,
            @Argument UpdatePostInput input,
            @AuthenticationPrincipal
            UserDetails userDetails) {
 
        return postService.updatePost(
            id, input, userDetails.getUsername());
    }
 
    @MutationMapping
    @PreAuthorize("isAuthenticated()")
    public boolean deletePost(
            @Argument Long id,
            @AuthenticationPrincipal
            UserDetails userDetails) {
 
        postService.deletePost(
            id, userDetails.getUsername());
        return true;
    }
 
    @MutationMapping
    @PreAuthorize("isAuthenticated()")
    public Post publishPost(
            @Argument Long id,
            @AuthenticationPrincipal
            UserDetails userDetails) {
 
        return postService.publishPost(
            id, userDetails.getUsername());
    }
}
 
@Controller
public class CommentMutationController {
 
    @Autowired
    private CommentService commentService;
 
    @Autowired
    private ApplicationEventPublisher eventPublisher;
 
    @MutationMapping
    @PreAuthorize("isAuthenticated()")
    public Comment createComment(
            @Argument CreateCommentInput input,
            @AuthenticationPrincipal
            UserDetails userDetails) {
 
        Comment comment = commentService.createComment(
            input, userDetails.getUsername());
 
        // Publish event cho subscription
        eventPublisher.publishEvent(
            new CommentAddedEvent(comment));
 
        return comment;
    }
 
    @MutationMapping
    @PreAuthorize("isAuthenticated()")
    public boolean deleteComment(
            @Argument Long id,
            @AuthenticationPrincipal
            UserDetails userDetails) {
 
        commentService.deleteComment(
            id, userDetails.getUsername());
        return true;
    }
}

@Service
@Transactional
public class PostService {
 
    @Autowired
    private PostRepository postRepository;
 
    @Autowired
    private AuthorRepository authorRepository;
 
    @Transactional(readOnly = true)
    public PostConnection findPosts(
            Boolean published, Long authorId,
            String tag, int first, String after) {
 
        // Decode cursor (base64-encoded ID)
        Long afterId = after != null
            ? decodeCursor(after) : null;
 
        List<Post> posts;
        if (authorId != null) {
            posts = postRepository
                .findByAuthorId(authorId);
        } else if (tag != null) {
            posts = postRepository.findByTag(tag);
        } else if (published != null) {
            posts = postRepository
                .findByPublished(published,
                    PageRequest.of(0, first + 1,
                        Sort.by("createdAt").descending()))
                .getContent();
        } else {
            posts = postRepository
                .findAll(PageRequest.of(0, first + 1,
                    Sort.by("createdAt").descending()))
                .getContent();
        }
 
        // Áp dụng cursor filtering
        if (afterId != null) {
            final Long id = afterId;
            posts = posts.stream()
                .filter(p -> p.getId() > id)
                .collect(Collectors.toList());
        }
 
        boolean hasNextPage = posts.size() > first;
        List<Post> pageItems = hasNextPage
            ? posts.subList(0, first) : posts;
 
        List<PostEdge> edges = pageItems.stream()
            .map(p -> new PostEdge(p, encodeCursor(p)))
            .collect(Collectors.toList());
 
        PageInfo pageInfo = new PageInfo(
            hasNextPage,
            afterId != null,
            edges.isEmpty() ? null
                : edges.get(0).cursor(),
            edges.isEmpty() ? null
                : edges.get(edges.size() - 1).cursor()
        );
 
        return new PostConnection(
            edges, pageInfo, posts.size());
    }
 
    public Post createPost(
            CreatePostInput input,
            String authorEmail) {
 
        Author author = authorRepository
            .findByEmail(authorEmail)
            .orElseThrow(() ->
                new IllegalArgumentException(
                    "Author not found"));
 
        Post post = new Post();
        post.setTitle(input.title());
        post.setContent(input.content());
        post.setAuthor(author);
        post.setTags(input.tags() != null
            ? input.tags() : List.of());
 
        return postRepository.save(post);
    }
 
    public Post publishPost(
            Long id, String authorEmail) {
 
        Post post = postRepository.findById(id)
            .orElseThrow(() ->
                new IllegalArgumentException(
                    "Post not found: " + id));
 
        if (!post.getAuthor().getEmail()
                .equals(authorEmail)) {
            throw new AccessDeniedException(
                "You don't own this post");
        }
 
        post.setPublished(true);
        post.setUpdatedAt(Instant.now());
        return postRepository.save(post);
    }
 
    private String encodeCursor(Post post) {
        return Base64.getEncoder().encodeToString(
            post.getId().toString().getBytes());
    }
 
    private Long decodeCursor(String cursor) {
        return Long.parseLong(new String(
            Base64.getDecoder().decode(cursor)));
    }
}

# Query này kích hoạt N+1 nếu không có DataLoader:
query {
    posts(first: 20) {
        edges {
            node {
                title
                author {      # ← Một DB query cho mỗi post!
                    name
                }
            }
        }
    }
}

@Controller
public class PostBatchController {
 
    @Autowired
    private AuthorRepository authorRepository;
 
    @Autowired
    private CommentRepository commentRepository;
 
    // Batch load Author cho tất cả posts cùng lúc
    @BatchMapping(typeName = "Post", field = "author")
    public Map<Post, Author> postAuthors(
            List<Post> posts) {
 
        // Thu thập tất cả author IDs
        Set<Long> authorIds = posts.stream()
            .map(p -> p.getAuthor().getId())
            .collect(Collectors.toSet());
 
        // Một query duy nhất cho tất cả authors
        Map<Long, Author> authorsById =
            authorRepository.findAllByIds(authorIds)
                .stream()
                .collect(Collectors.toMap(
                    Author::getId,
                    Function.identity()));
 
        // Map mỗi post với author của nó
        return posts.stream()
            .collect(Collectors.toMap(
                Function.identity(),
                p -> authorsById.get(
                    p.getAuthor().getId())));
    }
 
    // Batch load Comments cho tất cả posts cùng lúc
    @BatchMapping(typeName = "Post", field = "comments")
    public Map<Post, List<Comment>> postComments(
            List<Post> posts) {
 
        List<Long> postIds = posts.stream()
            .map(Post::getId)
            .collect(Collectors.toList());
 
        // Một query duy nhất cho tất cả comments
        Map<Long, List<Comment>> commentsByPostId =
            commentRepository.findByPostIds(postIds)
                .stream()
                .collect(Collectors.groupingBy(
                    c -> c.getPost().getId()));
 
        return posts.stream()
            .collect(Collectors.toMap(
                Function.identity(),
                p -> commentsByPostId.getOrDefault(
                    p.getId(), List.of())));
    }
}

@Configuration
public class DataLoaderConfig {
 
    @Bean
    public BatchLoaderRegistry batchLoaderRegistry(
            AuthorRepository authorRepository) {
 
        return registrar -> {
            registrar.forTypePair(
                Long.class, Author.class)
                .registerBatchLoader(
                    (authorIds, environment) ->
                        Mono.fromCallable(() ->
                            authorRepository
                                .findAllByIds(authorIds))
                            .map(authors -> {
                                Map<Long, Author> byId =
                                    authors.stream()
                                        .collect(
                                            Collectors.toMap(
                                                Author::getId,
                                                Function.identity()));
                                return authorIds.stream()
                                    .map(id -> byId.get(id))
                                    .collect(
                                        Collectors.toList());
                            }));
        };
    }
}
 
// Sử dụng DataLoader trong controller
@SchemaMapping(typeName = "Post", field = "author")
public CompletableFuture<Author> author(
        Post post,
        DataLoader<Long, Author> dataLoader) {
    return dataLoader.load(
        post.getAuthor().getId());
}

@Controller
public class SubscriptionController {
 
    // Publisher cho comment events
    private final Sinks.Many<Comment> commentSink =
        Sinks.many().multicast()
            .onBackpressureBuffer();
 
    // Publisher cho post events
    private final Sinks.Many<Post> postSink =
        Sinks.many().multicast()
            .onBackpressureBuffer();
 
    // Được gọi khi comment được tạo
    public void onCommentAdded(Comment comment) {
        commentSink.tryEmitNext(comment);
    }
 
    // Được gọi khi post được publish
    public void onPostPublished(Post post) {
        postSink.tryEmitNext(post);
    }
 
    @SubscriptionMapping
    public Flux<Comment> commentAdded(
            @Argument Long postId) {
        return commentSink.asFlux()
            .filter(c -> c.getPost().getId()
                .equals(postId));
    }
 
    @SubscriptionMapping
    public Flux<Post> postPublished() {
        return postSink.asFlux()
            .filter(Post::isPublished);
    }
}

// Cách tiếp cận dựa trên events sử dụng Spring events
@Component
public class GraphQLEventBridge
        implements ApplicationListener<
            CommentAddedEvent> {
 
    @Autowired
    private SubscriptionController subscriptionController;
 
    @Override
    public void onApplicationEvent(
            CommentAddedEvent event) {
        subscriptionController
            .onCommentAdded(event.getComment());
    }
}

// Frontend WebSocket subscription (JavaScript)
import { createClient } from 'graphql-ws';
 
const client = createClient({
    url: 'ws://localhost:8080/graphql-ws',
});
 
// Subscribe vào comments mới trên post 1
const unsubscribe = client.subscribe(
    {
        query: `
            subscription OnCommentAdded($postId: ID!) {
                commentAdded(postId: $postId) {
                    id
                    content
                    author {
                        name
                    }
                    createdAt
                }
            }
        `,
        variables: { postId: '1' },
    },
    {
        next: (data) =>
            console.log('Comment mới:', data),
        error: (err) => console.error(err),
        complete: () =>
            console.log('Subscription kết thúc'),
    }
);

@Component
public class GraphQLExceptionHandler
        implements DataFetcherExceptionResolver {
 
    @Override
    public Mono<List<GraphQLError>> resolveException(
            Throwable exception,
            DataFetchingEnvironment environment) {
 
        GraphQLError error = resolveError(exception);
        if (error != null) {
            return Mono.just(List.of(error));
        }
        return Mono.empty(); // Để handler mặc định xử lý
    }
 
    private GraphQLError resolveError(
            Throwable exception) {
 
        if (exception instanceof IllegalArgumentException) {
            return GraphqlErrorBuilder.newError()
                .errorType(ErrorType.BAD_REQUEST)
                .message(exception.getMessage())
                .build();
        }
 
        if (exception instanceof AccessDeniedException) {
            return GraphqlErrorBuilder.newError()
                .errorType(ErrorType.FORBIDDEN)
                .message("Access denied")
                .build();
        }
 
        if (exception instanceof EntityNotFoundException) {
            return GraphqlErrorBuilder.newError()
                .errorType(ErrorType.NOT_FOUND)
                .message(exception.getMessage())
                .build();
        }
 
        return null; // Không xử lý — để mặc định xử lý
    }
}

@Configuration
public class ScalarConfig
        implements RuntimeWiringConfigurer {
 
    @Override
    public void configure(
            RuntimeWiring.Builder builder) {
        builder.scalar(ExtendedScalars.DateTime);
    }
}

# Query
query {
    post(id: 1) {
        title
        author {
            name
        }
    }
}
 
# Response với partial error
{
  "data": {
    "post": {
      "title": "My Post",
      "author": null        ← field thất bại
    }
  },
  "errors": [
    {
      "message": "Author not found",
      "path": ["post", "author"],
      "extensions": {
        "classification": "NOT_FOUND"
      }
    }
  ]
}

@Configuration
@EnableWebSecurity
@EnableMethodSecurity
public class SecurityConfig {
 
    @Autowired
    private JwtAuthenticationFilter jwtFilter;
 
    @Bean
    public SecurityFilterChain filterChain(
            HttpSecurity http) throws Exception {
        return http
            .csrf(csrf -> csrf.disable())
            .sessionManagement(session -> session
                .sessionCreationPolicy(
                    SessionCreationPolicy.STATELESS))
            .authorizeHttpRequests(auth -> auth
                // GraphQL endpoint — cho phép tất cả
                // (auth theo field qua @PreAuthorize)
                .requestMatchers("/graphql").permitAll()
                .requestMatchers("/graphiql").permitAll()
                .requestMatchers("/graphql-ws").permitAll()
                .anyRequest().authenticated())
            .addFilterBefore(jwtFilter,
                UsernamePasswordAuthenticationFilter.class)
            .build();
    }
}

@Controller
public class AuthorController {
 
    // Ai cũng có thể query public posts
    @QueryMapping
    public List<Post> posts() {
        return postService.findPublished();
    }
 
    // Chỉ người dùng đã xác thực mới thấy drafts của mình
    @QueryMapping
    @PreAuthorize("isAuthenticated()")
    public List<Post> myDrafts(
            @AuthenticationPrincipal
            UserDetails userDetails) {
        return postService.findDraftsByAuthor(
            userDetails.getUsername());
    }
 
    // Chỉ admins mới có thể xóa bất kỳ post nào
    @MutationMapping
    @PreAuthorize("hasRole('ADMIN')")
    public boolean adminDeletePost(
            @Argument Long id) {
        postService.forceDelete(id);
        return true;
    }
}

@Controller
public class AuthorController {
 
    // Email chỉ hiển thị cho chính author đó
    @SchemaMapping(typeName = "Author", field = "email")
    public String email(
            Author author,
            @AuthenticationPrincipal
            UserDetails userDetails) {
 
        if (userDetails != null &&
                author.getEmail().equals(
                    userDetails.getUsername())) {
            return author.getEmail();
        }
        return "***@***.***"; // Ẩn với người khác
    }
}

@SpringBootTest
@AutoConfigureGraphQlTester
class PostControllerTest {
 
    @Autowired
    private GraphQlTester graphQlTester;
 
    @Autowired
    private PostRepository postRepository;
 
    @Autowired
    private AuthorRepository authorRepository;
 
    @BeforeEach
    void setUp() {
        Author author = new Author();
        author.setName("Alice");
        author.setEmail("[email protected]");
        authorRepository.save(author);
 
        Post post = new Post();
        post.setTitle("GraphQL trong Spring Boot");
        post.setContent("Nội dung hướng dẫn...");
        post.setAuthor(author);
        post.setPublished(true);
        postRepository.save(post);
    }
}

@Test
void shouldFetchPublishedPosts() {
    graphQlTester.document("""
            query {
                posts(published: true, first: 10) {
                    edges {
                        node {
                            title
                            published
                            author {
                                name
                            }
                        }
                    }
                    totalCount
                }
            }
            """)
        .execute()
        .path("posts.edges[0].node.title")
        .entity(String.class)
        .isEqualTo("GraphQL trong Spring Boot")
        .path("posts.edges[0].node.author.name")
        .entity(String.class)
        .isEqualTo("Alice")
        .path("posts.totalCount")
        .entity(Integer.class)
        .satisfies(count -> assertThat(count)
            .isGreaterThan(0));
}
 
@Test
void shouldFetchSinglePost() {
    Long postId = postRepository.findAll()
        .get(0).getId();
 
    graphQlTester.document("""
            query GetPost($id: ID!) {
                post(id: $id) {
                    title
                    content
                }
            }
            """)
        .variable("id", postId)
        .execute()
        .path("post.title")
        .entity(String.class)
        .isEqualTo("GraphQL trong Spring Boot");
}
 
@Test
void shouldReturnNullForMissingPost() {
    graphQlTester.document("""
            query {
                post(id: 99999) {
                    title
                }
            }
            """)
        .execute()
        .path("post")
        .valueIsNull();
}

@Test
@WithMockUser(username = "[email protected]")
void shouldCreatePost() {
    graphQlTester.document("""
            mutation CreatePost($input: CreatePostInput!) {
                createPost(input: $input) {
                    id
                    title
                    published
                    author {
                        name
                    }
                }
            }
            """)
        .variable("input", Map.of(
            "title", "Bài Viết Mới",
            "content", "Nội dung bài viết",
            "tags", List.of("java", "graphql")))
        .execute()
        .path("createPost.title")
        .entity(String.class)
        .isEqualTo("Bài Viết Mới")
        .path("createPost.published")
        .entity(Boolean.class)
        .isEqualTo(false)
        .path("createPost.author.name")
        .entity(String.class)
        .isEqualTo("Alice");
}
 
@Test
void shouldRejectUnauthenticatedMutation() {
    graphQlTester.document("""
            mutation {
                createPost(input: {
                    title: "Hack",
                    content: "Attempt"
                }) {
                    id
                }
            }
            """)
        .execute()
        .errors()
        .satisfy(errors ->
            assertThat(errors).isNotEmpty());
}

@Test
void shouldReturnErrorForInvalidInput() {
    graphQlTester.document("""
            mutation {
                createPost(input: {
                    title: "",
                    content: ""
                }) {
                    id
                }
            }
            """)
        .execute()
        .errors()
        .satisfy(errors ->
            assertThat(errors)
                .anyMatch(e -> e.getErrorType()
                    == ErrorType.BAD_REQUEST));
}

# Query độc hại — lồng nhau sâu
query {
    posts {
        edges { node {
            comments {
                author {
                    posts {
                        edges { node {
                            comments {
                                author { name }
                            }
                        }}
                    }
                }
            }
        }}
    }
}

spring:
  graphql:
    schema:
      introspection:
        enabled: false      # Tắt trong production

@Configuration
public class GraphQLConfig implements
        GraphQlSourceBuilderCustomizer {
 
    @Override
    public void customize(
            GraphQlSource.SchemaResourceBuilder builder) {
        builder.configureRuntimeWiring(wiringBuilder ->
            wiringBuilder
                .instrumentations(List.of(
                    new MaxQueryDepthInstrumentation(10),
                    new MaxQueryComplexityInstrumentation(
                        200)))
        );
    }
}

@Configuration
public class PersistedQueryConfig {
 
    @Bean
    public PersistedQueryCache persistedQueryCache() {
        // Cache trong bộ nhớ của các queries được phép
        Map<String, String> allowedQueries = Map.of(
            "getPosts",
                "query { posts(first: 10) { ... } }",
            "getPost",
                "query GetPost($id: ID!) { post(id: $id) { ... } }"
        );
        return new InMemoryPersistedQueryCache(
            allowedQueries);
    }
}

@Component
public class GraphQLLoggingInstrumentation
        extends SimpleInstrumentation {
 
    private static final Logger log =
        LoggerFactory.getLogger(
            GraphQLLoggingInstrumentation.class);
 
    @Override
    public InstrumentationContext<ExecutionResult>
            beginExecution(
                InstrumentationExecutionParameters params,
                InstrumentationState state) {
 
        long startTime = System.currentTimeMillis();
        String operationName =
            params.getOperation();
 
        return SimpleInstrumentationContext
            .whenCompleted((result, throwable) -> {
                long duration =
                    System.currentTimeMillis()
                        - startTime;
 
                if (throwable != null || !result
                        .getErrors().isEmpty()) {
                    log.warn("GraphQL {} thất bại trong {}ms: {}",
                        operationName, duration,
                        result.getErrors());
                } else {
                    log.debug("GraphQL {} hoàn thành trong {}ms",
                        operationName, duration);
                }
            });
    }
}