lukechilds
/
node-sql
mirror of https://github.com/lukechilds/node-sql.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

add basic mysql support

auto-join
Paul Winkler 12 years ago
parent
baa59c9b2e
commit
e55cb2f878
3 changed files with 369 additions and 27 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 338
      lib/dialect/mysql.js
  2. 54
      lib/index.js
  3. 4
      lib/table.js

338
lib/dialect/mysql.js
View File

@ -0,0 +1,338 @@
'use strict';
var util = require('util');
var assert = require('assert');
var From = require(__dirname + '/../node/from');
var Parameter = require(__dirname + '/../node/parameter');
var Mysql = function() {
this.output = [];
this.params = [];
};
Mysql.prototype.getQuery = function(queryNode) {
this.visitQuery(queryNode);
return { text: this.output.join(' '), values: this.params };
};
Mysql.prototype.visit = function(node) {
switch(node.type) {
case 'QUERY': return this.visitQuery(node);
case 'SUBQUERY': return this.visitSubquery(node);
case 'SELECT': return this.visitSelect(node);
case 'INSERT': return this.visitInsert(node);
case 'UPDATE': return this.visitUpdate(node);
case 'DELETE': return this.visitDelete();
case 'CREATE': return this.visitCreate(node);
case 'DROP': return this.visitDrop(node);
case 'ALTER': return this.visitAlter(node);
case 'FROM': return this.visitFrom(node);
case 'WHERE': return this.visitWhere(node);
case 'ORDER BY': return this.visitOrderBy(node);
case 'GROUP BY': return this.visitGroupBy(node);
case 'RETURNING': return this.visitReturning(node);
case 'BINARY': return this.visitBinary(node);
case 'TABLE': return this.visitTable(node);
case 'COLUMN': return this.visitColumn(node);
case 'JOIN': return this.visitJoin(node);
case 'TEXT': return node.text;
case 'UNARY': return this.visitUnary(node);
case 'PARAMETER': return this.visitParameter(node);
case 'DEFAULT': return this.visitDefault(node);
case 'IF EXISTS': return this.visitIfExists();
case 'IF NOT EXISTS': return this.visitIfNotExists();
case 'ADD COLUMN': return this.visitAddColumn(node);
case 'DROP COLUMN': return this.visitDropColumn(node);
case 'RENAME COLUMN': return this.visitRenameColumn(node);
case 'LIMIT':
case 'OFFSET':
return this.visitModifier(node);
default: throw new Error("Unrecognized node type " + node.type);
}
};
Mysql.prototype.quote = function(word) {
return '`' + word + '`';
};
Mysql.prototype.visitSelect = function(select) {
var result = ['SELECT', select.nodes.map(this.visit.bind(this)).join(', ')];
this._selectOrDeleteEndIndex = this.output.length + result.length;
return result;
};
Mysql.prototype.visitInsert = function(insert) {
var self = this;
this._visitedFrom = true;
//don't use table.column for inserts
this._visitedInsert = true;
var paramNodes = insert.getParameters()
.map(function (paramSet) {
return paramSet.map(function (param) {
return self.visit(param);
}).join(', ');
}).map(function (param) {
return '('+param+')';
}).join(', ');
var result = [
'INSERT INTO',
this.visit(this._queryNode.table.toNode()),
'(' + insert.columns.map(this.visit.bind(this)).join(', ') + ')',
'VALUES', paramNodes
];
return result;
};
Mysql.prototype.visitUpdate = function(update) {
//don't auto-generate from clause
this._visitedFrom = true;
var params = [];
/*jshint boss: true */
for(var i = 0, node; node = update.nodes[i]; i++) {
this._visitingUpdateTargetColumn = true;
var target_col = this.visit(node);
this._visitingUpdateTargetColumn = false;
params = params.concat(target_col + ' = ' + this.visit(node.value));
}
var result = [
'UPDATE',
this.visit(this._queryNode.table.toNode()),
'SET',
params.join(', ')
];
return result;
};
Mysql.prototype.visitDelete = function() {
this._selectOrDeleteEndIndex = 1;
return ['DELETE'];
};
Mysql.prototype.visitCreate = function(create) {
this._visitingCreate = true;
//don't auto-generate from clause
this._visitedFrom = true;
var table = this._queryNode.table;
var col_nodes = table.columns.map(function(col) { return col.toNode(); });
var result = ['CREATE TABLE'];
result = result.concat(create.nodes.map(this.visit.bind(this)));
result.push(this.visit(table.toNode()));
result.push('(' + col_nodes.map(this.visit.bind(this)).join(', ') + ')');
this._visitingCreate = false;
return result;
};
Mysql.prototype.visitDrop = function(drop) {
//don't auto-generate from clause
this._visitedFrom = true;
var result = ['DROP TABLE'];
result = result.concat(drop.nodes.map(this.visit.bind(this)));
result.push(this.visit(this._queryNode.table.toNode()));
return result;
};
Mysql.prototype.visitAlter = function(alter) {
this._visitingAlter = true;
//don't auto-generate from clause
this._visitedFrom = true;
var table = this._queryNode.table;
var col_nodes = table.columns.map(function(col) { return col.toNode(); });
var result = [
'ALTER TABLE',
this.visit(table.toNode()),
alter.nodes.map(this.visit.bind(this)).join(', ')
];
this._visitingAlter = false;
return result;
};
Mysql.prototype.visitFrom = function(from) {
this._visitedFrom = true;
var result = [];
result.push('FROM');
for(var i = 0; i < from.nodes.length; i++) {
result = result.concat(this.visit(from.nodes[i]));
}
return result;
};
Mysql.prototype.visitWhere = function(where) {
var result = ['WHERE', where.nodes.map(this.visit.bind(this)).join(', ')];
return result;
};
Mysql.prototype.visitOrderBy = function(orderBy) {
var result = ['ORDER BY', orderBy.nodes.map(this.visit.bind(this)).join(', ')];
return result;
};
Mysql.prototype.visitGroupBy = function(groupBy) {
var result = ['GROUP BY', groupBy.nodes.map(this.visit.bind(this)).join(', ')];
return result;
};
Mysql.prototype.visitBinary = function(binary) {
var self = this;
var result = '(' + this.visit(binary.left) + ' ' + binary.operator + ' ';
if (Array.isArray(binary.right)) {
result += '(' + binary.right.map(function (node) {
return self.visit(node);
}).join(', ') + ')';
}
else {
result += this.visit(binary.right);
}
result += ')';
return result;
};
Mysql.prototype.visitUnary = function(unary) {
return '(' + this.visit(unary.left) + ' ' + unary.operator + ')';
};
Mysql.prototype.visitQuery = function(queryNode) {
this._queryNode = queryNode;
for(var i = 0; i < queryNode.nodes.length; i ++) {
var res = this.visit(queryNode.nodes[i]);
this.output = this.output.concat(res);
}
//implicit 'from'
if(!this._visitedFrom) {
var select = this.output.slice(0, this._selectOrDeleteEndIndex);
var from = this.visitFrom(new From().add(queryNode.table.toNode()));
var rest = this.output.slice(this._selectOrDeleteEndIndex);
this.output = select.concat(from).concat(rest);
}
return this;
};
Mysql.prototype.visitSubquery = function(queryNode) {
var result = [];
for(var i = 0; i < queryNode.nodes.length; i ++) {
var res = this.visit(queryNode.nodes[i]);
result = result.concat(res);
}
//implicit 'from'
if(!this._visitedFrom) {
var select = result.slice(0, this._selectOrDeleteEndIndex);
var from = this.visitFrom(new From().add(queryNode.table.toNode()));
var rest = result.slice(this._selectOrDeleteEndIndex);
result = select.concat(from).concat(rest);
}
result[0] = '('+result[0];
result[result.length-1] = result[result.length-1] + ') ' + queryNode.alias;
return result;
};
Mysql.prototype.visitTable = function(tableNode) {
var table = tableNode.table;
var txt="";
if(table.getSchema()) {
txt = this.quote(table.getSchema());
txt += '.';
}
txt += this.quote(table.getName());
if(table.alias) {
txt += ' AS ' + table.alias;
}
return txt;
};
Mysql.prototype.visitColumn = function(columnNode) {
var table = columnNode.table;
var inSelectClause = !this._selectOrDeleteEndIndex;
var txt = "";
if(inSelectClause) {
if (columnNode.asArray) {
txt += 'array_agg(';
} else if (columnNode.aggCount) {
txt += 'COUNT(';
}
}
if(!this._visitedInsert && !this._visitingUpdateTargetColumn && !this._visitingCreate && !this._visitingAlter) {
if(table.alias) {
txt = table.alias;
} else {
if(table.getSchema()) {
txt = this.quote(table.getSchema());
txt += '.';
}
txt += this.quote(table.getName());
}
txt += '.';
}
if (columnNode.star) {
txt += '*';
} else {
txt += this.quote(columnNode.name);
}
if(inSelectClause && (columnNode.asArray || columnNode.aggCount)) {
txt += ')';
}
if(inSelectClause && columnNode.alias) {
txt += ' as ' + this.quote(columnNode.alias);
}
if(this._visitingCreate || this._visitingAddColumn) {
assert(columnNode.dataType, 'dataType missing for column ' + columnNode.name +
' (CREATE TABLE and ADD COLUMN statements require a dataType)');
txt += ' ' + columnNode.dataType;
}
return txt;
};
Mysql.prototype.visitParameter = function(parameter) {
this.params.push(parameter.value());
return "?";
};
Mysql.prototype.visitDefault = function(parameter) {
var params = this.params;
this.params.push('DEFAULT');
return "?";
};
Mysql.prototype.visitAddColumn = function(addColumn) {
this._visitingAddColumn = true;
var result = ['ADD COLUMN ' + this.visit(addColumn.nodes[0])];
this._visitingAddColumn = false;
return result;
};
Mysql.prototype.visitDropColumn = function(dropColumn) {
return ['DROP COLUMN ' + this.visit(dropColumn.nodes[0])];
};
Mysql.prototype.visitRenameColumn = function(renameColumn) {
return ['RENAME COLUMN ' + this.visit(renameColumn.nodes[0]) + ' TO ' + this.visit(renameColumn.nodes[1])];
};
Mysql.prototype.visitIfExists = function() {
return ['IF EXISTS'];
};
Mysql.prototype.visitIfNotExists = function() {
return ['IF NOT EXISTS'];
};
Mysql.prototype.visitJoin = function(join) {
var result = [];
result = result.concat(this.visit(join.from));
result = result.concat(join.subType + ' JOIN');
result = result.concat(this.visit(join.to));
result = result.concat('ON');
result = result.concat(this.visit(join.on));
return result;
};
Mysql.prototype.visitReturning = function(returning) {
return ['RETURNING', returning.nodes.map(this.visit.bind(this)).join(', ')];
};
Mysql.prototype.visitModifier = function(node) {
return [node.type, node.count];
};
module.exports = Mysql;

54
lib/index.js
View File

@ -1,26 +1,28 @@
'use strict';
var Table = require(__dirname + '/table');
var sql = {
Table: Table,
define: Table.define,
select: function() {
var Query = require(__dirname + '/node/query');
var query = new Query();
query.select.apply(query, arguments);
return query;
},
setDialect: function(dialect) {
switch(dialect.toLowerCase()) {
case 'postgres':
this.dialect = require(__dirname + '/dialect/postgres');
break;
default:
throw new Error(dialect + ' is unsupported');
}
return this;
}
};
module.exports = sql.setDialect('postgres');
'use strict';
var Table = require(__dirname + '/table');
var sql = {
Table: Table,
define: Table.define,
select: function() {
var Query = require(__dirname + '/node/query');
var query = new Query();
query.select.apply(query, arguments);
return query;
},
setDialect: function(dialect) {
switch(dialect.toLowerCase()) {
case 'postgres':
this.dialect = require(__dirname + '/dialect/postgres');
case 'mysql':
this.dialect = require(__dirname + '/dialect/mysql');
break;
default:
throw new Error(dialect + ' is unsupported');
}
return this;
}
};
module.exports = sql.setDialect('postgres');

4
lib/table.js
View File

@ -47,7 +47,9 @@ Table.prototype.getName = function() {
Table.prototype.star = function() {
var name = this.alias || this._name;
return new TextNode('"'+name+'".*');
var Dialect = require(__dirname + '/../').dialect;
var dialect = new Dialect();
return new TextNode(dialect.quote(name)+'.*');
};
Table.prototype.count = function(alias) {

Write Preview
Loading…
Cancel
Save