how to close websocket after it has been moved into select?

[hp8wvvvgnj6asjm7]

#[tokio::main]
async fn main(){

    let base = String::from("wss://");

    let (stream, _) = connect_async(&base).await.unwrap();

    let (tx, rx) = mpsc::unbounded_channel();

    let (write, read) = stream.split();

    let ws_to_stdout  = { 
        
        read.for_each(|message| async {

            // dbg!(&message);

            let thread_tx = tx.clone();

            if !message.as_ref().unwrap().is_ping() {

                let data:X= serde_json::from_str(&message.unwrap().into_text().unwrap()).unwrap(); 
                
                // dbg!(&data);
                
                tx.send(data).unwrap();
            }
        })
    };

    pin!(rx, ws_to_stdout);

    let (a, b, c) = tokio::join!(async {

        loop {

            let x = rx.recv().await;

            // dbg!(&x);
        }

    }, ws_to_stdout, async {

        tokio::time::timeout(Duration::from_secs(5), async {

            

            // let x = ws_to_stdout.as_mut();

            

            write.close(Some(CloseFrame { code: CloseCode::Normal, reason: String::from("").into() })).await.unwrap();
            
        }).await

    });

    println!("end.");
    
}

[sdroege]

I'm not sure what the exact question here is. You'd close it by calling close() on it, or by dropping the websocket (i.e. in your case the read and write side of the socket).

What's the problem you're running into?

[hp8wvvvgnj6asjm7]

I don't know on which object I have to call the close method into which i put Some(CloseFrame { code: CloseCode::Normal, reason: String::from("").into() })

[hp8wvvvgnj6asjm7]

I can't use the websocket because it is moved into the join! macro

[sdroege]

You can write a close frame.

[hp8wvvvgnj6asjm7]

   |
107 |             stream.send(Some(CloseFrame { code: CloseCode::Normal, reason: String::from("").into() }));
    |             ^^^^^^^^^^^^-----------------------------------------------------------------------------^
    |                         |
    |                         this argument influences the return type of `send

[hp8wvvvgnj6asjm7]

 write.send(Some(CloseFrame { code: CloseCode::Normal, reason: String::from("").into() }));
    |             ^^^^^^^^^^^-----------------------------------------------------------------------------^
    |                        |
    |                        this argument influences the return type of `send

[hp8wvvvgnj6asjm7]

109 |             write.close(CloseFrame { code: CloseCode::Normal, reason: String::from("").into() });
    |                   ^^^^^ -----------------------------------------------------------------------
    |                         |
    |                         unexpected argument of type `CloseFrame<'_>`
    |                         help: remove the extra argument

[hp8wvvvgnj6asjm7]

I guess this is the correct way ?

    write.send(
                
                Message::Close(Some(CloseFrame { code: CloseCode::Normal, reason: String::from("").into() }))
            ).await;

[hp8wvvvgnj6asjm7]

It just causes a panic.

[sdroege]

What panic?

[hp8wvvvgnj6asjm7]

Thanks a lot, I think I got the close frame right, but because I'm not parsing the incoming data correctly the panic is caused inside the for_each.

[sdroege]

Great :) So I think this can be closed? Please re-open if there's still some kind of problem