Amazon DAX Client for JavaScript

const AmazonDaxClient = require('amazon-dax-client');
const AWS = require('aws-sdk'); // Make sure aws-sdk is installed

// Replace with your actual DAX cluster discovery endpoint
const endpoint = process.env.DAX_CLUSTER_ENDPOINT ?? 'dax://my-cluster.abc123.dax-clusters.us-east-1.amazonaws.com';
const region = process.env.AWS_REGION ?? 'us-east-1';

console.log(`Connecting to DAX endpoint: ${endpoint} in region: ${region}`);

try {
  // Instantiate the DAX client
  const dax = new AmazonDaxClient({ endpoints: [endpoint], region: region });

  // Integrate DAX with DynamoDB DocumentClient
  const docClient = new AWS.DynamoDB.DocumentClient({ service: dax });

  // Example: Put an item using DocumentClient with DAX
  const tableName = 'MyTestTable'; // Replace with your table name
  const item = { pk: '123', sk: 'data', attribute: 'cached value' };

  console.log('Putting item with DAX-enabled DocumentClient...');
  docClient.put({ TableName: tableName, Item: item }, (err, data) => {
    if (err) {
      console.error('Error putting item:', JSON.stringify(err, null, 2));
    } else {
      console.log('Successfully put item:', data);

      // Example: Get the item using DocumentClient with DAX
      console.log('Getting item with DAX-enabled DocumentClient...');
      docClient.get({ TableName: tableName, Key: { pk: '123', sk: 'data' } }, (getErr, getData) => {
        if (getErr) {
          console.error('Error getting item:', JSON.stringify(getErr, null, 2));
        } else {
          console.log('Successfully got item:', getData);
        }
      });
    }
  });
} catch (e) {
  console.error('Failed to initialize DAX client:', e);
}