1
0
mirror of https://github.com/Unleash/unleash.git synced 2024-10-18 20:09:08 +02:00
unleash.unleash/lib/db/client-strategy-store.js

63 lines
1.5 KiB
JavaScript
Raw Normal View History

2016-11-02 23:17:28 +01:00
'use strict';
const COLUMNS = ['app_name', 'strategies'];
const TABLE = 'client_strategies';
const mapRow = (row) => ({
appName: row.app_name,
strategies: row.strategies,
});
class ClientStrategyStore {
constructor (db) {
this.db = db;
}
updateRow (appName, strategies) {
return this.db(TABLE)
2016-11-02 23:17:28 +01:00
.where('app_name', appName) // eslint-disable-line
.update({
strategies: JSON.stringify(strategies),
updated_at: 'now()', // eslint-disable-line
});
}
insertNewRow (appName, strategies) {
return this.db(TABLE).insert({
2016-11-02 23:17:28 +01:00
app_name: appName, // eslint-disable-line
strategies: JSON.stringify(strategies),
});
}
insert (appName, strategies) {
return this.db(TABLE)
2016-11-02 23:17:28 +01:00
.count('*')
.where('app_name', appName)
.map(row => ({ count: row.count }))
.then(rows => {
2016-11-02 23:24:23 +01:00
if (rows[0].count > 0) {
return this.updateRow(appName, strategies);
2016-11-02 23:24:23 +01:00
} else {
return this.insertNewRow(appName, strategies);
2016-11-02 23:24:23 +01:00
}
2016-11-02 23:17:28 +01:00
});
}
getAll () {
return this.db
2016-11-02 23:17:28 +01:00
.select(COLUMNS)
.from(TABLE)
.map(mapRow);
}
getByAppName (appName) {
return this.db
.select('strategies')
.where('app_name', appName)
.from(TABLE)
.map((row) => row.strategies);
}
2016-11-02 23:17:28 +01:00
};
module.exports = ClientStrategyStore;