通过预热 Amazon WorkSpaces 提升用户操作体验

{    "Version": "2012-10-17",    "Statement": [        {            "Sid": "state1",            "Effect": "Allow",            "Action": [                "dynamodb:DeleteItem",                "dynamodb:GetItem",                "dynamodb:PutItem",                "dynamodb:Scan",                "dynamodb:UpdateItem"            ],            "Resource": "arn:<partition>:dynamodb:<region>:<account-id>:table/Jobs"        },        {            "Sid": "state2",            "Effect": "Allow",            "Action": [                "logs:CreateLogStream",                "logs:PutLogEvents"            ],            "Resource": "arn:<partition>:logs:<region>:<account-id>:log-group:/aws/lambda/OnDemandRequest:*"        },        {            "Sid": "state3",            "Effect": "Allow",            "Action": "logs:CreateLogGroup",            "Resource": "arn:<partition>:logs:<region>:<account-id>:*"        }    ]}

from datetime import datetime
import jsonimport uuid
import boto3
def lambda_handler(event, context):    """    Lambda function export to API Gateway to recieve a json formatted post job.    JSON parameters description:        'username': The username of WorkSpace. Required.        'spaceid': The WorkSpace ID request to warm-up.                    Optional. If not assign, will warm up all WorkSpace of the user.        'on_at': The will to be used time of the WorkSpace. Required.
        For example:        {            "username": "xxxxx",            "spaceid": "ws-1234abcd",            "on_at": "2020-03-03 12:12:12"        }    """    print('Recieved on demand warm-up request - [%s].' % json.dumps(event))
    jobID = str(uuid.uuid1())    on_at = datetime.strptime(event['on_at'], '%Y-%m-%d %H:%M:%S').isoformat()
    try:        dynamodb = boto3.resource('dynamodb', region_name='cn-northwest-1')        table = dynamodb.Table('Jobs')
        table.put_item(            Item={                'id': jobID,                'username': event['username'],                'spaceid': event.get('spaceid', '-'),                'on_at': on_at            }        )
        print('On demand warm-up request has created with jobID [%s].' % jobID)
        return {            'statusCode': 200,            'body': {                'id': jobID            }        }    except Exception as ex:        return {            'statusCode': 500,            'body': {                'error': ex            }        }

pip install awscurlexport AWS_ACCESS_KEY_ID="<your-account-access-key-id>"export AWS_SECRET_ACCESS_KEY="<your-account-secret-access-key>"awscurl --region <your-api-region> "<your-api-endpoint>" -X POST -d "{\"username\": \"<workspace-username>\", \"spaceid\": \"<workspace-id> \", \"on_at\": \"<request-on-datetime, format as: 2020-03-03 12:12:12>\"}"

{  "username": "xxxxx",  "spaceid": "ws-1234abcd",  "on_at": "2020-03-03 12:12:12"}

{    "Version": "2012-10-17",    "Statement": [        {            "Sid": "state1",            "Effect": "Allow",            "Action": [                "workspaces:StartWorkspaces",                "workspaces:DescribeWorkspaces"            ],            "Resource": "arn:<aws-partitionname>:workspaces:<region>:<account-id>:workspace/*"        },        {            "Sid": "state2",            "Effect": "Allow",            "Action": [                "workspaces:DescribeWorkspaceDirectories"            ],            "Resource": " arn:<aws-partitionname>:workspaces:<region>:<account-id>:*"        }    ]}

from datetime import datetime
import boto3
def lambda_handler(event, context):    """    Lambda function used with CloudWatch Rule Event to execute scheduled WorkSpaces warm-up job.    Can pass in a DirectoryId as parameter, for example:        event['directory'] = 'd-123456'    if no DirectoryId specified, the job will scan all available Directoies and warm-up WorkSpaces.    """
    print('Start scheduled warm-up at %s.' % datetime.now().strftime('%Y-%m-%d %H:%M:%S'))    directory = event.get('directory', None)
    try:        directories = []        workspaces = []
        client = boto3.client('workspaces')
        check = {} if directory is None else {'DirectoryIds': [directory]}
        resp = client.describe_workspace_directories(**check)
        while resp:            directories += resp['Directories']
            resp = client.describe_workspace_directories(                **check,                NextToken=resp['NextToken']            ) if 'NextToken' in resp else None
        for dir in directories:            dir_id = dir['DirectoryId']            resp = client.describe_workspaces(DirectoryId=dir_id)
            while resp:                workspaces += resp['Workspaces']
                resp = client.describe_workspaces(                    DirectoryId=dir_id, NextToken=resp['NextToken']                ) if 'NextToken' in resp else None
        for workspace in workspaces:            if workspace['State'] == 'STOPPED':                client.start_workspaces(                    StartWorkspaceRequests=[{                        'WorkspaceId': workspace['WorkspaceId']                    }]                )
                print('Starting WorkSpace for id - [%s].' % workspace['WorkspaceId'])    except Exception as ex:        print('Error to run scheduled warm-up job - [%s].' % ex)