| 22 | } |
| 23 | |
| 24 | async fn on_connect(socket: SocketRef) { |
| 25 | info!("socket connected: {}", socket.id); |
| 26 | |
| 27 | socket.on( |
| 28 | "join", |
| 29 | |socket: SocketRef, Data::<String>(room), store: State<state::MessageStore>| async move { |
| 30 | info!("Received join: {:?}", room); |
| 31 | let _ = socket.leave_all(); |
| 32 | let _ = socket.join(room.clone()); |
| 33 | let messages = store.get(&room).await; |
| 34 | let _ = socket.emit("messages", Messages { messages }); |
| 35 | }, |
| 36 | ); |
| 37 | |
| 38 | socket.on( |
| 39 | "message", |
| 40 | |socket: SocketRef, Data::<MessageIn>(data), store: State<state::MessageStore>| async move { |
| 41 | info!("Received message: {:?}", data); |
| 42 | |
| 43 | let response = state::Message { |
| 44 | text: data.text, |
| 45 | user: format!("anon-{}", socket.id), |
| 46 | date: chrono::Utc::now(), |
| 47 | }; |
| 48 | |
| 49 | store.insert(&data.room, response.clone()).await; |
| 50 | |
| 51 | let _ = socket.within(data.room).emit("message", response); |
| 52 | }, |
| 53 | ) |
| 54 | } |
| 55 | |
| 56 | async fn handler(axum::extract::State(io): axum::extract::State<SocketIo>) { |
| 57 | info!("handler called"); |