Better metrics insert and service changes

This commit is contained in:
geek 2017-05-12 14:59:37 -05:00 committed by Sérgio Ramos
parent 559df105b6
commit b039fdad8b
2 changed files with 192 additions and 10 deletions

View File

@ -125,10 +125,32 @@ module.exports = class Data {
}); });
} }
getMetrics (deploymentId) { getMetrics (containerId) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
this._db.metrics.query({ deploymentId }, (err, metrics) => { this._db.metrics.get(containerId, (err, result) => {
return err ? reject(err) : resolve(metrics || []); return err ? reject(err) : resolve(result);
});
});
}
insertMetrics (containerId, metrics) {
return new Promise((resolve, reject) => {
this._db.metrics.get(containerId, (err, existing) => {
if (err) {
return reject(err);
}
if (existing) {
this._db.metrics.update(containerId, { metrics: this._db.append(metrics) }, (err) => {
return err ? reject(err) : resolve(existing);
});
return;
}
const entry = { id: containerId, metrics };
this._db.metrics.insert(entry, { merge: true }, (err) => {
return err ? reject(err) : resolve(entry);
});
}); });
}); });
} }
@ -140,6 +162,7 @@ module.exports = class Data {
}); });
}); });
} }
updateService (deploymentId, service) { updateService (deploymentId, service) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
this._db.deployments.get(deploymentId, { filter: 'services' }, (err, deployment) => { this._db.deployments.get(deploymentId, { filter: 'services' }, (err, deployment) => {
@ -147,18 +170,29 @@ module.exports = class Data {
return reject(err); return reject(err);
} }
const services = deployment.services.map((currentService) => { const serviceToUpdate = deployment.services.find((currentService) => {
if (currentService.name === service.name) { return (currentService.name === service.name);
currentService.count = service.count;
}
return currentService;
}); });
this._db.deployments.update(deploymentId, { services }, (err, keys) => { if (!serviceToUpdate) {
deployment.services.push(service);
} else {
serviceToUpdate.count = service.count;
serviceToUpdate.containers = service.containers;
}
this._db.deployments.update(deploymentId, { services: deployment.services }, (err) => {
return err ? reject(err) : resolve(service); return err ? reject(err) : resolve(service);
}); });
}); });
}); });
} }
deploymentChanges (handler) {
return new Promise((resolve, reject) => {
this._db.deployments.changes('*', { reconnect: true, handler }, (err) => {
return err ? reject(err) : resolve();
});
});
}
}; };

View File

@ -55,3 +55,151 @@ describe('getDeployment()', () => {
}); });
}); });
}); });
describe('updateService()', () => {
it('will update the services for an existing deployment', (done) => {
const data = new PortalData(internals.options);
data.connect().then(() => {
const deployment = {
name: 'User Services',
datacenter: 'us-sw-1'
};
const service = {
name: 'consul',
containers: [
{
server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
alias: 'nodejsexample_consul_1',
image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
}
],
count: 1
};
data.createDeployment(deployment).then((deployment) => {
expect(deployment.id).to.exist();
data.updateService(deployment.id, service).then((updatedService) => {
expect(updatedService).to.equal(service);
done();
});
});
});
});
});
describe('deploymentChanges()', () => {
it('will execute the handler when a deployment service changes', (done) => {
const data = new PortalData(internals.options);
data.connect().then(() => {
const deployment = {
name: 'User Services',
datacenter: 'us-sw-1'
};
const service1 = {
name: 'consul',
containers: [
{
server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
alias: 'nodejsexample_consul_1',
image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
}
],
count: 1
};
const service2 = {
name: 'consul',
containers: [
{
server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
alias: 'nodejsexample_consul_1',
image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
},
{
server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
alias: 'nodejsexample_consul_2',
image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
},
{
server_id: '423e7432-b760-11e2-bf6c-002590c3f1a0',
alias: 'nodejsexample_consul_3',
image_id: '91b757b5-bd29-2126-5ff9-ae9235011ff5',
owner_id: '30f62ec2-24a2-6f8e-8fad-d46b04c8a0b9',
id: '81205d4a-92f4-c4d9-da8a-aafd689eeabb'
}
],
count: 3
};
data.createDeployment(deployment).then((deployment) => {
expect(deployment.id).to.exist();
data.updateService(deployment.id, service1).then((updatedService1) => {
expect(updatedService1).to.equal(service1);
let executed = false;
data.deploymentChanges((err, changes) => {
if (executed) {
return;
}
expect(changes.before).to.exist();
expect(changes.after).to.exist();
done();
executed = true;
}).then(() => {
data.updateService(deployment.id, service2).then((updatedService2) => {
expect(updatedService2).to.equal(service2);
});
});
});
});
});
});
});
describe('insertMetrics()', () => {
it('will add new metrics to a service and won\'t overwrite existing ones', (done) => {
const data = new PortalData(internals.options);
data.connect().then(() => {
const containerId = '81205d4a-92f4-c4d9-da8a-aafd689eeabb';
const metrics1 = [
{
timestamp: 1494360995851,
cpu: 1.2,
memory: 23344523,
network: 5024
}
];
const metrics2 = [
{
timestamp: 1495360995851,
cpu: 1.3,
memory: 23344523,
network: 4024
}
];
data.insertMetrics(containerId, metrics1).then((result1) => {
expect(result1.id).to.equal(containerId);
expect(result1.metrics).to.equal(metrics1);
data.insertMetrics(containerId, metrics2).then((result2) => {
expect(result2.id).to.equal(containerId);
data.getMetrics(containerId).then((results) => {
expect(results.metrics.length).to.equal(2);
done();
});
});
});
});
});
});