def createDAXClient(region, endpoint): if environ.get('CODEBUILD_BUILD_ID') is not None: return amazondax.AmazonDaxClient(region_name=region, endpoints=[endpoint]) else: logging.info('using profile {}'.format(PROFILE)) session = botocore.session.Session(profile=PROFILE) return amazondax.AmazonDaxClient(session, region_name=region, endpoints=[endpoint])
import os, sys, time import amazondax import botocore.session region = os.environ.get('AWS_DEFAULT_REGION', 'us-west-2') session = botocore.session.get_session() dynamodb = session.create_client('dynamodb', region_name=region) # low-level client table_name = "TryDaxTable" if len(sys.argv) > 1: endpoint = sys.argv[1] dax = amazondax.AmazonDaxClient(session, region_name=region, endpoints=[endpoint]) client = dax else: client = dynamodb pk = 5 sk1 = 2 sk2 = 9 iterations = 5 params = { 'TableName': table_name, 'KeyConditionExpression': 'pk = :pkval and sk between :skval1 and :skval2', 'ExpressionAttributeValues': { ":pkval": {