ORM base & redis adapted

This commit is contained in:
Anatoliy Chakkaev 2011-10-01 19:51:51 +04:00
commit d5341d3126
3 changed files with 491 additions and 0 deletions

345
index.js Normal file
View File

@ -0,0 +1,345 @@
exports.Schema = Schema;
// exports.AbstractClass = AbstractClass;
var slice = Array.prototype.slice;
/**
* Shema - classes factory
* @param name - type of schema adapter (mysql, mongoose, sequelize, redis)
* @param settings - any database-specific settings which we need to
* establish connection (of course it depends on specific adapter)
*/
function Schema(name, settings) {
// just save everything we get
this.name = name;
this.settings = settings;
// create blank models pool
this.models = {};
// and initialize schema using adapter
// this is only one initialization entry point of adapter
// this module should define `adapter` member of `this` (schema)
require('./lib/' + name).initialize(this, function () {
this.connected = true;
this.emit('connected');
}.bind(this));
// we have an adaper now?
if (!this.adapter) {
throw new Error('Adapter is not defined correctly: it should create `adapter` member of schema');
}
};
require('util').inherits(Schema, process.EventEmitter);
function Text() {
}
Schema.Text = Text;
/**
* Define class
* @param className
* @param properties - hash of class properties in format
* {property: Type, property2: Type2, ...}
* or
* {property: {type: Type}, property2: {type: Type2}, ...}
* @param settings - other configuration of class
*/
Schema.prototype.define = function defineClass(className, properties, settings) {
var schema = this;
var args = slice.call(arguments);
if (!className) throw new Error('Class name required');
if (args.length == 1) properties = {}, args.push(properties);
if (args.length == 2) settings = {}, args.push(settings);
standartize(properties, settings);
// every class can receive hash of data as optional param
var newClass = function (data) {
AbstractClass.apply(this, args.concat([data]));
};
hiddenProperty(newClass, 'schema', schema);
hiddenProperty(newClass, 'modelName', className);
hiddenProperty(newClass, 'cache', {});
// setup inheritance
newClass.__proto__ = AbstractClass;
require('util').inherits(newClass, AbstractClass);
// store class in model pool
this.models[className] = newClass;
// pass controll to adapter
this.adapter.define({
model: newClass,
properties: properties,
settings: settings
});
return newClass;
function standartize(properties, settings) {
Object.keys(properties).forEach(function (key) {
var v = properties[key];
if (typeof v === 'function') {
properties[key] = { type: v };
}
});
// TODO: add timestamps fields
// when present in settings: {timestamps: true}
// or {timestamps: {created: 'created_at', updated: false}}
// by default property names: createdAt, updatedAt
}
};
/**
* Abstract class constructor
*/
function AbstractClass(name, properties, settings, data) {
var self = this;
data = data || {};
Object.defineProperty(this, 'id', {
writable: false,
enumerable: true,
configurable: true,
value: data.id
});
Object.keys(properties).forEach(function (attr) {
var _attr = '_' + attr,
attr_was = attr + '_was';
// Hidden property to store currrent value
Object.defineProperty(this, _attr, {
writable: true,
enumerable: false,
configurable: true,
value: isdef(data[attr]) ? data[attr] :
(isdef(this[attr]) ? this[attr] : null)
});
// Public setters and getters
Object.defineProperty(this, attr, {
get: function () {
return this[_attr];
},
set: function (value) {
this[_attr] = value;
},
configurable: true,
enumerable: true
});
// Getter for initial property
Object.defineProperty(this, attr_was, {
writable: true,
value: data[attr],
configurable: true,
enumerable: false
});
}.bind(this));
};
/**
* @param data [optional]
* @param callback(err, obj)
*/
AbstractClass.create = function (data) {
var modelName = this.modelName;
// define callback manually
var callback = arguments[arguments.length - 1];
if (arguments.length == 0 || data === callback) {
data = {};
}
if (typeof callback !== 'function') {
callback = function () {};
}
var obj = null;
if (data instanceof AbstractClass && !data.id) {
obj = data;
data = obj.toObject();
}
this.schema.adapter.create(modelName, data, function (err, id) {
obj = obj || new this(data);
if (id) {
obj.id = id;
this.cache[id] = obj;
}
if (callback) {
callback(err, obj);
}
}.bind(this));
};
AbstractClass.exists = function exists(id, cb) {
this.schema.adapter.exists(this.modelName, id, cb);
};
AbstractClass.find = function find(id, cb) {
this.schema.adapter.find(this.modelName, id, function (err, data) {
var obj = null;
if (data) {
if (this.cache[data.id]) {
obj = this.cache[data.id];
this.call(obj, data);
} else {
obj = new this(data);
this.cache[data.id] = obj;
}
}
cb(err, obj);
}.bind(this));
};
AbstractClass.all = function all(filter, cb) {
if (arguments.length === 1) {
cb = filter;
filter = null;
}
var constr = this;
this.schema.adapter.all(this.modelName, filter, function (err, data) {
var collection = null;
if (data && data.map) {
collection = data.map(function (d) {
var obj = null;
if (constr.cache[d.id]) {
obj = constr.cache[d.id];
constr.call(obj, d);
} else {
obj = new constr(d);
constr.cache[d.id] = obj;
}
return obj;
});
cb(err, collection);
}
});
};
AbstractClass.destroyAll = function destroyAll(cb) {
this.schema.adapter.destroyAll(this.modelName, function (err) {
if (!err) {
Object.keys(this.cache).forEach(function (id) {
delete this.cache[id];
}.bind(this));
}
cb(err);
}.bind(this));
};
AbstractClass.count = function (cb) {
this.schema.adapter.count(this.modelName, cb);
};
AbstractClass.toString = function () {
return '[Model ' + this.modelName + ']';
}
/**
* @param callback(err, obj)
*/
AbstractClass.prototype.save = function (callback) {
var modelName = this.constructor.modelName;
var data = this.toObject();
if (this.id) {
this._adapter().save(modelName, data, function (err) {
if (!err) {
this.constructor.call(this, data);
}
if (callback) {
callback(err, this);
}
}.bind(this));
} else {
this.constructor.create(this, callback);
}
};
AbstractClass.prototype._adapter = function () {
return this.constructor.schema.adapter;
};
AbstractClass.prototype.propertyChanged = function (name) {
return this[name + '_was'] !== this['_' + name];
};
AbstractClass.prototype.toObject = function () {
// blind faith: we only enumerate properties
var data = {};
Object.keys(this).forEach(function (property) {
data[property] = this[property];
}.bind(this));
return data;
};
AbstractClass.prototype.destroy = function (cb) {
this._adapter().destroy(this.constructor.modelName, this.id, function (err) {
delete this.constructor.cache[this.id];
cb(err);
}.bind(this));
};
AbstractClass.prototype.updateAttribute = function (name, value, cb) {
data = {};
data[name] = value;
this.updateAttributes(data, cb);
};
AbstractClass.prototype.updateAttributes = function updateAttributes(data, cb) {
var model = this.constructor.modelName;
this._adapter().updateAttributes(model, this.id, data, function (err) {
if (!err) {
Object.keys(data).forEach(function (key) {
this[key] = data[key];
Object.defineProperty(this, key + '_was', {
writable: false,
configurable: true,
enumerable: false,
value: data[key]
});
}.bind(this));
}
cb(err);
}.bind(this));
};
/**
* Checks is property changed based on current property and initial value
* @param {attr} String - property name
* @return Boolean
*/
AbstractClass.prototype.propertyChanged = function (attr) {
return this['_' + attr] !== this[attr + '_was'];
};
AbstractClass.prototype.reload = function (cb) {
this.constructor.find(this.id, cb);
};
// helper methods
//
function isdef(s) {
var undef;
return s !== undef;
}
function hiddenProperty(where, property, value) {
Object.defineProperty(where, property, {
writable: false,
enumerable: false,
configurable: false,
value: value
});
}

128
lib/redis-adapter.js Normal file
View File

@ -0,0 +1,128 @@
/**
* Module dependencies
*/
var redis = require('redis');
exports.initialize = function initializeSchema(schema, callback) {
schema.client = redis.createClient(
schema.settings.port,
schema.settings.host,
schema.settings.options
);
schema.client.auth(schema.settings.password, callback);
schema.adapter = new BridgeToRedis(schema.client);
};
function BridgeToRedis(client) {
this._models = {};
this.client = client;
}
BridgeToRedis.prototype.define = function (descr) {
this._models[descr.model.modelName] = descr;
};
BridgeToRedis.prototype.save = function (model, data, callback) {
this.client.hmset(model + ':' + data.id, data, callback);
};
BridgeToRedis.prototype.create = function (model, data, callback) {
this.client.incr(model + ':id', function (err, id) {
data.id = id;
this.save(model, data, function (err) {
if (callback) {
callback(err, id);
}
});
}.bind(this));
};
BridgeToRedis.prototype.exists = function (model, id, callback) {
this.client.exists(model + ':' + id, function (err, exists) {
if (callback) {
callback(err, exists);
}
});
};
BridgeToRedis.prototype.find = function find(model, id, callback) {
this.client.hgetall(model + ':' + id, function (err, data) {
if (data && data.id) {
data.id = id;
} else {
data = null;
}
callback(err, data);
});
};
BridgeToRedis.prototype.destroy = function destroy(model, id, callback) {
this.client.del(model + ':' + id, function (err) {
callback(err);
});
};
BridgeToRedis.prototype.all = function all(model, filter, callback) {
this.client.keys(model + ':*', function (err, keys) {
if (err) {
return callback(err, []);
}
var query = keys.map(function (key) {
return ['hgetall', key];
});
this.client.multi(query).exec(function (err, replies) {
callback(err, filter ? replies.filter(applyFilter(filter)) : replies);
});
}.bind(this));
};
function applyFilter(filter) {
if (typeof filter === 'function') {
return filter;
}
var keys = Object.keys(filter);
return function (obj) {
var pass = true;
keys.forEach(function (key) {
if (!test(filter[key], obj[key])) {
pass = false;
}
});
return pass;
}
function test(example, value) {
if (typeof value === 'string' && example && example.constructor.name === 'RegExp') {
return value.match(example);
}
// not strict equality
return example == value;
}
}
BridgeToRedis.prototype.destroyAll = function destroyAll(model, callback) {
this.client.keys(model + ':*', function (err, keys) {
if (err) {
return callback(err, []);
}
var query = keys.map(function (key) {
return ['del', key];
});
this.client.multi(query).exec(function (err, replies) {
callback(err);
});
}.bind(this));
};
BridgeToRedis.prototype.count = function count(model, callback) {
this.client.keys(model + ':*', function (err, keys) {
callback(err, err ? null : keys.length);
});
};
BridgeToRedis.prototype.updateAttributes = function updateAttrs(model, id, data, cb) {
this.client.hmset(model + ':' + id, data, cb);
};

18
package.json Normal file
View File

@ -0,0 +1,18 @@
{
"author": "Anatoliy Chakkaev",
"name": "yadm",
"description": "Yet another data mapper",
"version": "0.0.1",
"repository": {
"url": ""
},
"main": "index.js",
"scripts": {
"test": "nodeunit test/*_test.js"
},
"engines": {
"node": "~v0.4.9"
},
"dependencies": {},
"devDependencies": {}
}