Chat WebSocket API

import asyncio
    import json
    import random
    import websockets

    async def connect_to_chat():
        client_id = f"user_{random.randint(1000, 9999)}"
        uri = f"wss://mockly.me/ws/chat/{client_id}"
        
        async with websockets.connect(uri) as websocket:
            print(f"Connected to chat as {client_id}")
            
            # Handle incoming messages in background task
            async def receive_messages():
                while True:
                    try:
                        message = await websocket.recv()
                        data = json.loads(message)
                        
                        # Handle different message types
                        if data["type"] == "system":
                            print(f"SYSTEM: {data['message']}")
                        elif data["type"] == "message":
                            print(f"{data['sender']}: {data['message']}")
                        elif data["type"] == "error":
                            print(f"ERROR: {data['message']}")
                    except websockets.exceptions.ConnectionClosed:
                        print("Connection closed")
                        break
            
            # Start receiving messages in background
            receive_task = asyncio.create_task(receive_messages())
            
            # Send messages from command line
            try:
                while True:
                    message = input("Enter message: ")
                    if message.lower() == "exit":
                        break
                    
                    await websocket.send(json.dumps({
                        "message": message
                    }))
            finally:
                receive_task.cancel()

    # Run the chat client
    asyncio.run(connect_to_chat())

import java.net.URI;
    import java.util.Random;
    import java.util.concurrent.CountDownLatch;
    import javax.websocket.*;
    import org.glassfish.tyrus.client.ClientManager;
    import org.json.JSONObject;

    @ClientEndpoint
    public class ChatClient {
        private Session session;
        
        @OnOpen
        public void onOpen(Session session) {
            this.session = session;
            System.out.println("Connected to chat server");
        }
        
        @OnMessage
        public void onMessage(String message) {
            JSONObject data = new JSONObject(message);
            
            // Handle different message types
            String type = data.getString("type");
            switch (type) {
                case "system":
                    System.out.println("SYSTEM: " + data.getString("message"));
                    break;
                case "message":
                    System.out.println(data.getString("sender") + ": " + data.getString("message"));
                    break;
                case "error":
                    System.out.println("ERROR: " + data.getString("message"));
                    break;
            }
        }
        
        @OnClose
        public void onClose(Session session, CloseReason reason) {
            System.out.println("Disconnected from chat server: " + reason);
            this.session = null;
        }
        
        public void sendMessage(String text) {
            if (session != null) {
                JSONObject message = new JSONObject();
                message.put("message", text);
                session.getAsyncRemote().sendText(message.toString());
            }
        }
        
        public static void main(String[] args) {
            try {
                final ChatClient client = new ChatClient();
                final CountDownLatch latch = new CountDownLatch(1);
                
                // Generate a random client ID
                Random random = new Random();
                String clientId = "user_" + (1000 + random.nextInt(9000));
                
                // Connect to the chat server
                ClientManager clientManager = ClientManager.createClient();
                clientManager.connectToServer(client, 
                    URI.create("wss://mockly.me/ws/chat/" + clientId));
                
                // Allow the user to send messages
                BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
                System.out.println("Enter messages (type 'exit' to quit):");
                String line;
                while ((line = reader.readLine()) != null) {
                    if ("exit".equalsIgnoreCase(line.trim())) {
                        break;
                    }
                    client.sendMessage(line);
                }
                
                latch.await();
                
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }