在Java中實現WebSocket狀態管理，通常需要以下幾個步驟：

1. 引入依賴庫：首先，你需要引入一個支持WebSocket的庫，例如Java API for WebSocket (JSR 356) 或者 Spring Boot的WebSocket支持。這里以Spring Boot為例。

2. 創建WebSocket配置類：創建一個配置類，用于初始化WebSocket相關的設置。在這個類中，你需要定義一個WebSocketMessageBrokerConfigurer的實例，并配置WebSocket的端點和消息代理。

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/topic");
        config.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/websocket").withSockJS();
    }
}

3. 創建WebSocket控制器：創建一個控制器，用于處理客戶端發送的消息和廣播消息給所有連接的客戶端。在這個類中，你需要定義一個@MessageMapping的方法來處理客戶端發送的消息，并使用SimpMessagingTemplate來廣播消息。

@Controller
public class WebSocketController {

    @MessageMapping("/send")
    @SendTo("/topic/messages")
    public String handleMessage(@Payload String message) {
        return "Received message: " + message;
    }
}

4. 狀態管理：在WebSocket通信中，可以使用Session對象來管理客戶端的狀態。你可以在控制器中注入SimpMessagingTemplate，然后使用它來發送消息給特定的客戶端。例如，你可以創建一個方法來處理客戶端的連接和斷開連接，并在這些方法中更新客戶端的狀態。

@Autowired
private SimpMessagingTemplate messagingTemplate;

public void handleConnected(String clientId) {
    // 更新客戶端狀態為已連接
    updateClientStatus(clientId, "connected");
}

public void handleDisconnected(String clientId) {
    // 更新客戶端狀態為已斷開連接
    updateClientStatus(clientId, "disconnected");
}

private void updateClientStatus(String clientId, String status) {
    // 這里可以實現你的狀態管理邏輯，例如將客戶端狀態存儲在數據庫中
    // 然后使用 messagingTemplate 發送狀態更新消息給客戶端
    messagingTemplate.convertAndSend("/topic/status/" + clientId, status);
}

5. 客戶端實現：在客戶端，你需要使用JavaScript和SockJS庫來實現WebSocket連接。當客戶端連接到WebSocket服務器時，可以監聽connect和disconnect事件，以便在連接和斷開連接時更新客戶端狀態。

const socket = new SockJS('/websocket');
const stompClient = Stomp.over(socket);

stompClient.connect({}, () => {
    console.log('Connected to WebSocket server');
    // 發送客戶端ID到服務器，以便服務器更新客戶端狀態
    stompClient.send("/app/send", {}, JSON.stringify({ clientId: 'your-client-id' }));
});

socket.onclose = () => {
    console.log('Disconnected from WebSocket server');
    // 發送客戶端ID到服務器，以便服務器更新客戶端狀態
    stompClient.send("/app/send", {}, JSON.stringify({ clientId: 'your-client-id' }));
};

通過以上步驟，你可以在Java中實現WebSocket的狀態管理。請注意，這只是一個簡單的示例，實際應用中可能需要根據具體需求進行更多的定制和優化。