Use local method for md5

This commit is contained in:
Jacob Gunther 2018-04-16 10:52:53 -05:00 committed by GitHub
parent 830dc1bc43
commit cd3bf26dbe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,6 +2,12 @@ const crypto = require('crypto');
const rethink = require('rethinkdbdash'); const rethink = require('rethinkdbdash');
const winston = require('winston'); const winston = require('winston');
const md5 = (str) => {
const md5sum = crypto.createHash('md5');
md5sum.update(str);
return md5sum.digest('hex');
};
class RethinkDBStore { class RethinkDBStore {
constructor(options) { constructor(options) {
this.client = rethink({ this.client = rethink({
@ -15,7 +21,7 @@ class RethinkDBStore {
} }
set(key, data, callback) { set(key, data, callback) {
this.client.table('uploads').insert({ id: RethinkDBStore.md5(key), data: data }).run((error) => { this.client.table('uploads').insert({ id: md5(key), data: data }).run((error) => {
if (error) { if (error) {
callback(false); callback(false);
winston.error('failed to insert to table', error); winston.error('failed to insert to table', error);
@ -26,7 +32,7 @@ class RethinkDBStore {
} }
get(key, callback) { get(key, callback) {
this.client.table('uploads').get(RethinkDBStore.md5(key)).run((error, result) => { this.client.table('uploads').get(md5(key)).run((error, result) => {
if (error || !result) { if (error || !result) {
callback(false); callback(false);
winston.error('failed to insert to table', error); winston.error('failed to insert to table', error);
@ -38,8 +44,3 @@ class RethinkDBStore {
} }
module.exports = RethinkDBStore; module.exports = RethinkDBStore;
module.exports.md5 = (str) => {
const md5sum = crypto.createHash('md5');
md5sum.update(str);
return md5sum.digest('hex');
};