index.js
2.7 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
'auth_switch_request auth_switch_response auth_switch_request_more_data binlog_dump register_slave ssl_request handshake handshake_response query resultset_header column_definition text_row binary_row prepare_statement close_statement prepared_statement_header execute change_user'
.split(' ')
.forEach(function(name) {
var ctor = require('./' + name + '.js');
module.exports[ctor.name] = ctor;
// monkey-patch it to include name if debug is on
if (process.env.NODE_DEBUG) {
if (ctor.prototype.toPacket) {
var old = ctor.prototype.toPacket;
ctor.prototype.toPacket = function() {
var p = old.call(this);
p._name = ctor.name;
return p;
};
}
}
});
var Buffer = require('safe-buffer').Buffer;
// simple packets:
var Packet = require('./packet');
module.exports.Packet = Packet;
module.exports.OK = function OK() {};
module.exports.OK.toPacket = function(args, encoding) {
args = args || {};
var affectedRows = args.affectedRows || 0;
var insertId = args.insertId || 0;
var serverStatus = args.serverStatus || 0;
var warningCount = args.warningCount || 0;
var message = args.message || '';
var length = 9 + Packet.lengthCodedNumberLength(affectedRows);
length += Packet.lengthCodedNumberLength(insertId);
var buffer = Buffer.allocUnsafe(length);
var packet = new Packet(0, buffer, 0, length);
packet.offset = 4;
packet.writeInt8(0);
packet.writeLengthCodedNumber(affectedRows);
packet.writeLengthCodedNumber(insertId);
packet.writeInt16(serverStatus);
packet.writeInt16(warningCount);
packet.writeString(message, encoding);
packet._name = 'OK';
return packet;
};
// warnings, statusFlags
module.exports.EOF = function EOF() {};
module.exports.EOF.toPacket = function(warnings, statusFlags) {
if (typeof warnings == 'undefined') {
warnings = 0;
}
if (typeof statusFlags == 'undefined') {
statusFlags = 0;
}
var packet = new Packet(0, Buffer.allocUnsafe(9), 0, 9);
packet.offset = 4;
packet.writeInt8(0xfe);
packet.writeInt16(warnings);
packet.writeInt16(statusFlags);
packet._name = 'EOF';
return packet;
};
module.exports.Error = function Error() {};
module.exports.Error.toPacket = function(args, encoding) {
var length = 13 + Buffer.byteLength(args.message, 'utf8');
var packet = new Packet(0, Buffer.allocUnsafe(length), 0, length);
packet.offset = 4;
packet.writeInt8(0xff);
packet.writeInt16(args.code);
// TODO: sql state parameter
packet.writeString('#_____', encoding);
packet.writeString(args.message, encoding);
packet._name = 'Error';
return packet;
};