ButlerBin/lib/document_store.js

48 lines
1.0 KiB
JavaScript

var fs = require("fs");
var crypto = require("crypto");
var FileDocumentStore = function() {
this.basePath = "/data";
this.expire = null;
};
FileDocumentStore.md5 = function(str) {
var md5sum = crypto.createHash("md5");
md5sum.update(str);
return md5sum.digest("hex");
};
FileDocumentStore.prototype.set = function(key, data, callback) {
try {
var _this = this;
fs.mkdir(this.basePath, "700", function() {
var fn = _this.basePath + "/" + FileDocumentStore.md5(key);
fs.writeFile(fn, data, "utf8", function(err) {
if (err) {
callback(false);
}
else {
callback(true);
}
});
});
} catch(err) {
callback(false);
}
};
FileDocumentStore.prototype.get = function(key, callback) {
var _this = this;
var fn = _this.basePath + "/" + FileDocumentStore.md5(key);
fs.readFile(fn, "utf8", function(err, data) {
if (err) {
callback(false);
}
else {
callback(data);
}
});
};
module.exports = FileDocumentStore;