e2m/haste-server
6
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed bug in RethinkDB document store and use classes

Browse Source
This commit is contained in:
Jacob Gunther 2018-04-15 23:16:08 -05:00
parent 7f625e22f7
commit dc0f151a7f
1 changed files with 38 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
lib/document_stores/rethinkdb.js
View File

@ -1,39 +1,45 @@
const crypto = require('crypto'); const crypto = require('crypto');
const rethink = require('rethinkdbdash'); const rethink = require('rethinkdbdash');
const winston = require('winston');
var RethinkDBStore = (options) => { class RethinkDBStore {
this.client = rethink({ constructor(options) {
silent: true, this.client = rethink({
host: options.host || '127.0.0.1', silent: true,
port: options.port || 28015, host: options.host || '127.0.0.1',
db: options.db || 'haste', port: options.port || 28015,
user: options.user || 'admin', db: options.db || 'haste',
password: options.password || '' user: options.user || 'admin',
}); password: options.password || ''
}; });
}
RethinkDBStore.md5 = (str) => { set(key, data, callback) {
this.client.table('pastes').insert({ id: RethinkDBStore.md5(key), data: data }).run((error) => {
if (error) {
callback(false);
winston.error('failed to insert to table', error);
return;
}
callback(true);
});
}
get(key, callback) {
this.client.table('pastes').get(RethinkDBStore.md5(key)).run((error, result) => {
if (error || !result) {
callback(false);
winston.error('failed to insert to table', error);
return;
}
callback(result.data);
});
}
}
module.exports = RethinkDBStore;
module.exports.md5 = (str) => {
const md5sum = crypto.createHash('md5'); const md5sum = crypto.createHash('md5');
md5sum.update(str); md5sum.update(str);
return md5sum.digest('hex'); return md5sum.digest('hex');
}; };
RethinkDBStore.prototype.set = (key, data, callback) => {
try {
this.client.table('uploads').insert({ id: RethinkDBStore.md5(key), data: data }).run((error) => {
if (error) return callback(false);
callback(true);
});
} catch (err) {
callback(false);
}
};
RethinkDBStore.prototype.get = (key, callback) => {
this.client.table('uploads').get(RethinkDBStore.md5(key)).run((error, result) => {
if (error || !result) return callback(false);
callback(result.data);
});
};
module.exports = RethinkDBStore;