mirror of
https://github.com/serverless/serverless.git
synced 2026-01-18 14:58:43 +00:00
110 lines
3.6 KiB
JavaScript
110 lines
3.6 KiB
JavaScript
'use strict';
|
|
|
|
/**
|
|
* EventLambdaStream:
|
|
* Deploys a Stream based event sources (dynamoDB & Kinesis).
|
|
*
|
|
* Options:
|
|
* - stage: stage to deploy event to
|
|
* - region: region to deploy event to
|
|
* - path: event path
|
|
*/
|
|
|
|
module.exports = function(SPlugin, serverlessPath) {
|
|
|
|
const path = require('path'),
|
|
SError = require(path.join(serverlessPath, 'Error')),
|
|
BbPromise = require('bluebird');
|
|
let SUtils;
|
|
|
|
|
|
class EventLambdaStream extends SPlugin {
|
|
|
|
constructor(S, config) {
|
|
super(S, config);
|
|
SUtils = S.utils;
|
|
}
|
|
|
|
static getName() {
|
|
return 'serverless.core.' + EventLambdaStream.name;
|
|
}
|
|
|
|
registerActions() {
|
|
|
|
this.S.addAction(this.eventLambdaStream.bind(this), {
|
|
handler: 'eventLambdaStream',
|
|
description: 'Deploy a stream based event source'
|
|
});
|
|
|
|
return BbPromise.resolve();
|
|
}
|
|
|
|
/**
|
|
* Deploy Stream Event
|
|
*/
|
|
|
|
eventLambdaStream(evt) {
|
|
let _this = this;
|
|
_this.evt = evt;
|
|
|
|
if (!_this.evt.options.stage || !_this.evt.options.region || !_this.evt.options.name) {
|
|
return BbPromise.reject(new SError(`Missing stage, region, or event name.`));
|
|
}
|
|
|
|
_this.aws = _this.S.getProvider('aws');
|
|
|
|
let event = _this.S.getProject().getEvent( _this.evt.options.name ),
|
|
populatedEvent = event.toObjectPopulated({stage: _this.evt.options.stage, region: _this.evt.options.region}),
|
|
functionName = event.getFunction().getDeployedName(_this.evt.options),
|
|
regionVars = _this.S.getProject().getRegion(_this.evt.options.stage, _this.evt.options.region).getVariables(),
|
|
eventVar = 'eventID:' + event.name,
|
|
awsAccountId = regionVars.iamRoleArnLambda.split('::')[1].split(':')[0],
|
|
lambdaArn = 'arn:aws:lambda:' + _this.evt.options.region + ':' + awsAccountId + ':function:' + functionName + ':' + _this.evt.options.stage;
|
|
|
|
populatedEvent.config.startingPosition = populatedEvent.config.startingPosition || 'TRIM_HORIZON';
|
|
populatedEvent.config.batchSize = populatedEvent.config.batchSize || 100;
|
|
populatedEvent.config.enabled = populatedEvent.config.enabled ? true : false;
|
|
|
|
let params = {
|
|
FunctionName: lambdaArn,
|
|
BatchSize: populatedEvent.config.batchSize,
|
|
Enabled: populatedEvent.config.enabled
|
|
};
|
|
|
|
// Update or Create
|
|
if (regionVars[eventVar]) {
|
|
params.UUID = regionVars[eventVar];
|
|
|
|
return _this.aws.request('Lambda', 'updateEventSourceMapping', params, _this.evt.options.stage, _this.evt.options.region)
|
|
.then(function (data) {
|
|
|
|
SUtils.sDebug(`updated stream event source ${populatedEvent.config.streamArn} for lambda ${lambdaArn}`);
|
|
|
|
return BbPromise.resolve(data);
|
|
});
|
|
|
|
} else {
|
|
params.EventSourceArn = populatedEvent.config.streamArn;
|
|
params.StartingPosition = populatedEvent.config.startingPosition;
|
|
|
|
return _this.aws.request('Lambda', 'createEventSourceMapping', params, _this.evt.options.stage, _this.evt.options.region)
|
|
.then(function (data) {
|
|
SUtils.sDebug(`Created stream event source ${populatedEvent.config.streamArn} for lambda ${lambdaArn}`);
|
|
|
|
// save UUID
|
|
let newVar = {},
|
|
regionInstance = _this.S.getProject().getRegion(_this.evt.options.stage, _this.evt.options.region);
|
|
newVar[eventVar] = data.UUID;
|
|
regionInstance.addVariables(newVar);
|
|
regionInstance.save();
|
|
|
|
return BbPromise.resolve(data);
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
return( EventLambdaStream );
|
|
};
|