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": {
