Spring Boot中集成GraphQL订阅

在Spring Boot中集成GraphQL订阅，你需要遵循以下步骤：

1. 添加依赖

在你的pom.xml文件中添加graphql-spring-boot-starter和graphql-java-tools依赖：

<dependencies>
    <!-- GraphQL Spring Boot Starter -->
    <dependency>
        <groupId>com.graphql-spring-boot-starter</groupId>
        <artifactId>graphql-spring-boot-starter</artifactId>
        <version>11.1.0</version>
    </dependency>

    <!-- GraphQL Java Tools -->
    <dependency>
        <groupId>com.graphql-java-tools</groupId>
        <artifactId>graphql-java-tools</artifactId>
        <version>5.2.4</version>
    </dependency>

    <!-- Other dependencies -->
</dependencies>

2. 配置GraphQL

在你的application.yml或application.properties文件中配置GraphQL：

graphql:
  servlet:
    mapping: /graphql
    enabled: true
  tools:
    schema-location-pattern: classpath*:/schema/*.graphqls

3. 创建GraphQL Schema

在你的项目中创建一个schema.graphqls文件，定义你的类型、查询和订阅：

type Query {
    hello: String
}

type Subscription {
    newMessage: Message
}

type Message {
    id: ID!
    content: String!
    timestamp: String!
}

4. 创建Resolver

创建一个MessageResolver类来处理查询和订阅：

import com.coxautodev.graphql.tools.GraphQLQueryResolver;
import com.coxautodev.graphql.tools.GraphQLSubscriptionResolver;
import org.springframework.stereotype.Component;

@Component
public class MessageResolver implements GraphQLQueryResolver, GraphQLSubscriptionResolver {

    public String hello() {
        return "Hello, GraphQL!";
    }

    public Message newMessage(String content) {
        // 这里可以处理消息的创建逻辑，例如保存到数据库
        Message message = new Message();
        message.setContent(content);
        message.setTimestamp(java.time.LocalDateTime.now().toString());
        return message;
    }
}

5. 配置WebSocket

为了让GraphQL订阅正常工作，你需要配置WebSocket。在你的application.yml或application.properties文件中添加WebSocket配置：

spring:
  websocket:
    server:
      port: 8081

6. 创建WebSocket配置类

创建一个WebSocketConfig类来配置WebSocket：

import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {

    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(messageWebSocketHandler(), "/message").setAllowedOrigins("*");
    }

    public WebSocketHandler messageWebSocketHandler() {
        return new MessageWebSocketHandler();
    }
}

7. 创建WebSocket处理器

创建一个MessageWebSocketHandler类来处理WebSocket连接和消息：

import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

public class MessageWebSocketHandler extends TextWebSocketHandler {

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) {
        // 这里可以处理接收到的消息，例如广播给其他客户端
    }
}

现在你已经成功地在Spring Boot中集成了GraphQL订阅。你可以通过以下方式订阅新的消息：

subscription {
    newMessage(content: "Hello, world!") {
        id
        content
        timestamp
    }
}

每当有新的消息时，订阅者都会收到通知。