How to solved the share state in WebSocket?

[callmeitachi]

Summary

I have the following code:

use axum::{
Extension,
extract::{ws::{Message, WebSocket, WebSocketUpgrade}, State}, response::{IntoResponse, Response}, routing::get, Router
};
use std::sync::{Arc,Mutex};
pub struct AppState {
pub data:Arc<Mutex>,
}

pub async fn handler1(ws: WebSocketUpgrade,State(state):State) -> Response {
ws.on_upgrade(|socket| handle_socket(socket,state))
}

pub async fn handle_socket(mut socket: WebSocket,state:AppState) {
while let Some(msg) = socket.recv().await {
println!("got it");
let msg = if let Ok(msg) = msg {
msg
} else {
// client disconnected
return;
};
let mut state=state.data.lock().unwrap();

    *state+=1;
   let res_s=format!("{state}");
    let res_msg=Message::Text(res_s);
    if socket.send(res_msg).await.is_err() {
        return;
    }
}

}
but it can't compile?why?

axum version

axum = {version="0.7.7",features=["multipart","ws"]}