node-query/node_modules/mysql2/lib/connection_config.js

121 lines
3.8 KiB
JavaScript

var urlParse = require('url').parse;
var ClientConstants = require('./constants/client');
var Charsets = require('./constants/charsets');
module.exports = ConnectionConfig;
function ConnectionConfig(options) {
if (typeof options === 'string') {
options = ConnectionConfig.parseUrl(options);
}
this.isServer = options.isServer;
this.stream = options.stream;
this.host = options.host || 'localhost';
this.port = options.port || 3306;
this.socketPath = options.socketPath;
this.user = options.user || undefined;
this.password = options.password || undefined;
this.database = options.database;
this.insecureAuth = options.insecureAuth || false;
this.supportBigNumbers = options.supportBigNumbers || false;
this.bigNumberStrings = options.bigNumberStrings || false;
this.debug = options.debug;
this.timezone = options.timezone || 'local';
this.flags = options.flags || '';
this.queryFormat = options.queryFormat;
this.pool = options.pool || undefined;
this.multipleStatements = options.multipleStatements || false;
this.typeCast = (options.typeCast === undefined) ? true : options.typeCast;
this.ssl = options.ssl;
this.compress = options.compress || false;
this.dateStrings = options.dateStrings || false;
if (this.timezone[0] == " ") {
// "+" is a url encoded char for space so it
// gets translated to space when giving a
// connection string..
this.timezone = "+" + this.timezone.substr(1);
}
this.maxPacketSize = 0;
this.charsetNumber = (options.charset)
? ConnectionConfig.getCharsetNumber(options.charset)
: options.charsetNumber||Charsets.UTF8_GENERAL_CI;
this.clientFlags = ConnectionConfig.mergeFlags(ConnectionConfig.getDefaultFlags(options),
options.flags || '');
}
ConnectionConfig.mergeFlags = function(default_flags, user_flags) {
var flags = 0x0, i;
user_flags = (user_flags || '').toUpperCase().split(/\s*,+\s*/);
// add default flags unless "blacklisted"
for (i in default_flags) {
if (user_flags.indexOf("-" + default_flags[i]) >= 0) continue;
flags |= ClientConstants[default_flags[i]] || 0x0;
}
// add user flags unless already already added
for (i in user_flags) {
if (user_flags[i][0] == "-") continue;
if (default_flags.indexOf(user_flags[i]) >= 0) continue;
flags |= ClientConstants[user_flags[i]] || 0x0;
}
return flags;
};
ConnectionConfig.getDefaultFlags = function(options) {
var defaultFlags = [ "LONG_PASSWORD", "FOUND_ROWS", "LONG_FLAG",
"CONNECT_WITH_DB", "ODBC", "LOCAL_FILES",
"IGNORE_SPACE", "PROTOCOL_41", "IGNORE_SIGPIPE",
"TRANSACTIONS", "RESERVED", "SECURE_CONNECTION",
"MULTI_RESULTS" ];
if (options && options.multipleStatements) {
defaultFlags.push("MULTI_STATEMENTS");
}
return defaultFlags;
};
ConnectionConfig.getCharsetNumber = function(charset) {
return Charsets[charset];
};
ConnectionConfig.parseUrl = function(url) {
url = urlParse(url, true);
var options = {
host : url.hostname,
port : url.port,
database : url.pathname.substr(1),
};
if (url.auth) {
var auth = url.auth.split(':');
options.user = auth[0];
options.password = auth[1];
}
if (url.query) {
for (var key in url.query) {
var value = url.query[key];
try {
// Try to parse this as a JSON expression first
options[key] = JSON.parse(value);
} catch (err) {
// Otherwise assume it is a plain string
options[key] = value;
}
}
}
return options;
};