cache.ts

// Copyright 2016-2019, Pulumi Corporation.  All rights reserved.

import * as awsx from "@pulumi/awsx";
import * as pulumi from "@pulumi/pulumi";
import * as config from "./config";

// A simple cache abstraction that wraps Redis.
export class Cache {
    private readonly redis: awsx.ecs.FargateService;
    private readonly endpoint: pulumi.Output<awsx.elasticloadbalancingv2.ListenerEndpoint>;

    public get: (key: string) => Promise<string>;
    public set: (key: string, value: string) => Promise<void>;

    constructor(name: string, memory: number = 128) {
        const pw = config.redisPassword;
        const listener = new awsx.elasticloadbalancingv2.NetworkListener(name, { port: 6379 });
        this.redis = new awsx.ecs.FargateService(name, {
            taskDefinitionArgs: {
                containers: {
                    redis: {
                        image: "redis:alpine",
                        memory: memory,
                        portMappings: [listener],
                        command: ["redis-server", "--requirepass", pw],
                    },
                },
            },
        });

        this.endpoint = listener.endpoint;

        // Expose get/set as member fields that don't capture 'this'.  That way we don't try to
        // serialize pulumi resources unnecessarily into our lambdas.
        this.get = key => {
            const ep = listener.endpoint.get();
            console.log(`Getting key '${key}' on Redis@${ep.hostname}:${ep.port}`);

            const client = require("redis").createClient(ep.port, ep.hostname, { password: config.redisPassword });
            return new Promise<string>((resolve, reject) => {
                client.get(key, (err: any, v: any) => {
                    if (err) {
                        reject(err);
                    } else {
                        resolve(v);
                    }
                });
            });
        };

        this.set = (key, value) => {
            const ep = listener.endpoint.get();
            console.log(`Setting key '${key}' to '${value}' on Redis@${ep.hostname}:${ep.port}`);

            const client = require("redis").createClient(ep.port, ep.hostname, { password: config.redisPassword });
            return new Promise<void>((resolve, reject) => {
                client.set(key, value, (err: any, v: any) => {
                    if (err) {
                        reject(err);
                    } else {
                        console.log("Set succeed: " + JSON.stringify(v));
                        resolve();
                    }
                });
            });
        };
    }
}