Function handle_connection 

pub async fn handle_connection(
    ws: WebSocket,
    state: Arc<RwLock<ServerState>>,
    broadcast_tx: Arc<Sender<String>>,
)

Handle a WebSocket connection