How about append codes about modify 'self.application_state'

[osangu]

Hello!
How about adding self.application_state = WebSocketState.DISCONNECTED to under the if statement?

In my opinion, it can make more specify to handling closing websocket.

starlette/starlette/websockets.py

Lines 111 to 114 in c2e3a39

	def _raise_on_disconnect(self, message: Message) -> None:
	if message["type"] == "websocket.disconnect":
	raise WebSocketDisconnect(message["code"], message.get("reason"))

This change will be suitable for below example.

from fastapi import FastAPI
from fastapi.websockets import  WebSocket, WebSoketState, WebSocketDis

app = FastAPI()

@app.websocket('/foo')
async def bar(
        websocket: WebSocket
):
    await websocket.accepted()
   
    try:
        service_functinos(websocket) # WebSocketDisconnect raised by client, while doing this one.

    except WebSocketDisconnect:
        pass

    finally:
        # True, because there is no code about chaing state in `_raise_on_disconnect`
        still_connected = websocket.application_state == WebSocketState.CONNECTED 

        if still_connected:        
            await websocket.close()  # make second error, about closing twice

May be, you can think about there is no problem replace return to pass and add close function to try scope.
But In that case we can't close it when unexpected error occured : (

Thanks!