sringboot 使用websocket 当用户离开则修改数据库

在Spring Boot中使用WebSocket可以通过集成Spring的WebSocket模块来实现。下面是一个简单的示例，演示如何在用户离开时从数据库中删除数据：

1. 添加依赖

在pom.xml文件中添加以下依赖：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2. 创建WebSocket处理器

创建一个WebSocket处理器，处理用户连接和消息发送。以下是一个示例处理器：

@Component
public class MyWebSocketHandler extends TextWebSocketHandler {
    
    @Autowired
    private UserRepository userRepository;
    
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        // 新用户连接，向数据库插入新数据
        User user = new User();
        user.setName("New User");
        userRepository.save(user);
        
        // 发送欢迎消息
        session.sendMessage(new TextMessage("Welcome!"));
    }
    
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        // 用户离开，从数据库删除数据
        List<User> users = userRepository.findByName("New User");
        if (!users.isEmpty()) {
            User user = users.get(0);
            userRepository.delete(user);
        }
    }
    
    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 处理收到的消息
        session.sendMessage(new TextMessage("Received: " + message.getPayload()));
    }
    
}

在这个处理器中，我们在 `afterConnectionEstablished` 方法中处理新用户的链接，并向数据库中插入新数据。在 `afterConnectionClosed` 方法中处理用户离开，并从数据库中删除数据。在 `handleTextMessage` 方法中处理收到的消息。

3. 配置WebSocket

在Spring Boot中，使用WebSocket需要进行一些配置。在 `WebSocketConfig` 类中添加以下配置：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    
    @Autowired
    private MyWebSocketHandler myWebSocketHandler;
    
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myWebSocketHandler, "/websocket").setAllowedOrigins("*");
    }
    
}

在这个配置类中，我们启用WebSocket，并将我们的处理器添加到WebSocket处理器注册表中。

4. 创建实体类和Repository

为了演示如何从数据库中删除数据，我们需要创建一个简单的实体类和Repository。以下是一个示例实体类和Repository：

@Entity
@Table(name = "users")
public class User {
    
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    
    private String name;
    
    // getter 和 setter 省略...
    
}

@Repository
public interface UserRepository extends JpaRepository<User, Long> {
    
    List<User> findByName(String name);
    
}

在这个示例中，我们创建了一个简单的 `User` 实体类，并将其映射到名为 `users` 的数据库表中。我们还创建了一个简单的 `UserRepository` 接口，它扩展了 `JpaRepository` 接口，并添加了一个根据名称查找用户的方法。

5. 启动应用程序

最后，我们可以启动应用程序，并使用WebSocket客户端连接到 `/websocket` 路径。当用户离开时，我们将从数据库中删除相应的数据。

注意：这只是一个示例演示如何在用户离开时从数据库中删除数据。在实际应用中，你需要根据你的实际需求修改代码。