1
0
Fork 0
mirror of https://github.com/seejohnrun/haste-server.git synced 2024-12-25 11:37:00 +00:00

Merge pull request #189 from PassTheMayo/master

Added RethinkDB storage option & fixed config to use proper JSON
This commit is contained in:
John Crepezzi 2017-12-11 10:46:45 -05:00 committed by GitHub
commit 63c4576633
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 62 additions and 1 deletions

View file

@ -174,6 +174,28 @@ forward on GETs.
All of which are optional except `type` with very logical default values.
### RethinkDB
To use the RethinkDB storage system, you must install the `rethinkdbdash` package via npm
`npm install rethinkdbdash`
Once you've done that, your config section should look like this:
``` json
{
"type": "rethinkdb",
"host": "127.0.0.1",
"port": 28015,
"db": "haste"
}
```
In order for this to work, the database must be pre-created before the script is ran.
Also, you must create an `uploads` table, which will store all the data for uploads.
You can optionally add the `user` and `password` properties to use a user system.
## Author
John Crepezzi <john.crepezzi@gmail.com>
@ -206,4 +228,4 @@ SOFTWARE
* jQuery: MIT/GPL license
* highlight.js: Copyright © 2006, Ivan Sagalaev
* highlightjs-coffeescript: WTFPL - Copyright © 2011, Dmytrii Nagirniak
* highlightjs-coffeescript: WTFPL - Copyright © 2011, Dmytrii Nagirniak

View file

@ -0,0 +1,39 @@
const crypto = require('crypto');
const rethink = require('rethinkdbdash');
var RethinkDBStore = (options) => {
this.client = rethink({
silent: true,
host: options.host || '127.0.0.1',
port: options.port || 28015,
db: options.db || 'haste',
user: options.user || 'admin',
password: options.password || ''
});
};
RethinkDBStore.md5 = (str) => {
const md5sum = crypto.createHash('md5');
md5sum.update(str);
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;