一种解决方法是使用Amazon API Gateway的WebSocket API来管理Websocket连接，AWS API Gateway可以保留WebSocket连接，并且可以配置Callback URL来支持WebSocket连接的连接管理和消息转发。在这种情况下，使用AWS Lambda将处理传入连接请求，将连接转发到正确的节点，并转发来自客户端的所有消息。使用AWS API Gateway代替自建的WebSocket服务器也意味着可以轻松实现扩展和高可用性。

示例代码如下：

import json
import boto3
import decimal

dynamodb = boto3.resource('dynamodb')

def lambda_handler(event, context):
    connection_id = event['requestContext']['connectionId']
    message = json.loads(event['body'])

    # Save the message to DynamoDB
    table = dynamodb.Table('MyWebSocketMessagesTable')
    table.put_item(
        Item={
            'connectionId': connection_id,
            'message': message
        }
    )

    # Send a response back to the client
    response = {
        'statusCode': 200
    }

    return response

此lambda函数将传入的消息写入DynamoDB，并简单地返回200响应以表示成功。您可以根据自己的需求自定义此代码。