Update an AWS ECS Service by updating it's Task Definition. Code works with Environment Variables. Forked from @dennis-tra

aws-lambda-update-ecs-service

//Original Gist published by dennis-tra
//https://gist.github.com/dennis-tra/ae0dacbab1576d956bf62b4ab2209f5b

exports.handler = (event, context, callback) => {
    var aws = require('aws-sdk');
    aws.config.update({region: process.env.ECS_REGION});
    var ecs = new aws.ECS();
    
    //Fetching last Task Definition
    ecs.describeTaskDefinition({ taskDefinition: process.env.ECS_TASK_NAME }, function(err, data) {
        if (err) {
            console.log(err, err.stack);
        }
        else {
            //Fetched last Task Definition
            delete data.taskDefinition["taskDefinitionArn"];
            delete data.taskDefinition["revision"];
            delete data.taskDefinition["status"];
            delete data.taskDefinition["requiresAttributes"];
            
            ecs.registerTaskDefinition(data.taskDefinition, function(err, data) {
                if (err) {
                    console.log(err, err.stack); // an error occurred
                } else {
                    console.log("Successfully registered new task definition");
                    
                    var params = {
                        service: process.env.ECS_SERVICE,
                        cluster: process.env.ECS_CLUSTER,
                        deploymentConfiguration: {
                            maximumPercent: 200,
                            minimumHealthyPercent: 0
                        },
                        desiredCount: 1,
                        taskDefinition: process.env.ECS_TASK_NAME + ':' + data.taskDefinition.revision
                    };
            
                    ecs.updateService(params, function(err, data) {
                        if (err) {
                            console.log(err, err.stack); // an error occurred
                        }
                        else {
                            console.log("Successfully updated service.");
                        } 
                    });
                }
            });
        }
    });
};