mirror of
https://github.com/SunRed/haste-server.git
synced 2024-11-01 01:30:21 +01:00
Merge pull request #189 from PassTheMayo/master
Added RethinkDB storage option & fixed config to use proper JSON
This commit is contained in:
commit
63c4576633
2 changed files with 62 additions and 1 deletions
22
README.md
22
README.md
|
@ -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>
|
||||
|
|
39
lib/document_stores/rethinkdb.js
Normal file
39
lib/document_stores/rethinkdb.js
Normal 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;
|
Loading…
Reference in a new issue