111 lines
2.9 KiB
JavaScript
111 lines
2.9 KiB
JavaScript
var safeRequire = require('../utils').safeRequire;
|
|
|
|
/**
|
|
* Module dependencies
|
|
*/
|
|
var uuid = require('node-uuid');
|
|
var riak = safeRequire('riak-js');
|
|
|
|
exports.initialize = function initializeSchema(schema, callback) {
|
|
schema.client = riak.getClient({
|
|
host: schema.settings.host || '127.0.0.1',
|
|
port: schema.settings.port || 8091
|
|
});
|
|
schema.adapter = new Riak(schema.client);
|
|
};
|
|
|
|
function Riak(client) {
|
|
this._models = {};
|
|
this.client = client;
|
|
}
|
|
|
|
Riak.prototype.define = function (descr) {
|
|
this._models[descr.model.modelName] = descr;
|
|
};
|
|
|
|
Riak.prototype.save = function (model, data, callback) {
|
|
this.client.save(model, data.id, data, callback);
|
|
};
|
|
|
|
Riak.prototype.create = function (model, data, callback) {
|
|
data.id = uuid();
|
|
this.save(model, data, function (err) {
|
|
if (callback) {
|
|
callback(err, data.id);
|
|
}
|
|
});
|
|
};
|
|
|
|
Riak.prototype.exists = function (model, id, callback) {
|
|
this.client.exists(model, id, function (err, exists, meta) {
|
|
if (callback) {
|
|
callback(err, exists);
|
|
}
|
|
});
|
|
};
|
|
|
|
Riak.prototype.find = function find(model, id, callback) {
|
|
this.client.get(model, id, function (err, data, meta) {
|
|
if (data && data.id) {
|
|
data.id = id;
|
|
} else {
|
|
data = null;
|
|
}
|
|
if (typeof callback === 'function') callback(err, data);
|
|
});
|
|
};
|
|
|
|
Riak.prototype.destroy = function destroy(model, id, callback) {
|
|
this.client.remove(model, id, function (err) {
|
|
callback(err);
|
|
});
|
|
};
|
|
|
|
Riak.prototype.all = function all(model, filter, callback) {
|
|
var opts = {};
|
|
if (filter && filter.where) opts.where = filter.where;
|
|
this.client.getAll(model, function (err, result, meta) {
|
|
if (err) return callback(err, []);
|
|
/// return callback(err, result.map(function (x) { return {id: x}; }));
|
|
result = (result || []).map(function (row) {
|
|
var record = row.data;
|
|
record.id = row.meta.key;
|
|
console.log(record);
|
|
return record;
|
|
});
|
|
|
|
return callback(err, result);
|
|
}.bind(this));
|
|
};
|
|
|
|
Riak.prototype.destroyAll = function destroyAll(model, callback) {
|
|
var self = this;
|
|
this.all(model, {}, function (err, recs) {
|
|
if (err) callback(err);
|
|
|
|
removeOne();
|
|
|
|
function removeOne(error) {
|
|
err = err || error;
|
|
var rec = recs.pop();
|
|
if (!rec) return callback(err && err.statusCode != '404' ? err : null);
|
|
console.log(rec.id);
|
|
self.client.remove(model, rec.id, removeOne);
|
|
}
|
|
|
|
});
|
|
|
|
};
|
|
|
|
Riak.prototype.count = function count(model, callback) {
|
|
this.client.keys(model + ':*', function (err, keys) {
|
|
callback(err, err ? null : keys.length);
|
|
});
|
|
};
|
|
|
|
Riak.prototype.updateAttributes = function updateAttrs(model, id, data, cb) {
|
|
data.id = id;
|
|
this.save(model, data, cb);
|
|
};
|
|
|