node-postgres/lib/buffer-list.js

66 lines
1.4 KiB
JavaScript
Raw Normal View History

2010-10-09 12:00:37 +08:00
BufferList = function() {
this.buffers = [];
};
var p = BufferList.prototype;
2010-10-09 12:00:37 +08:00
p.add = function(buffer, front) {
2010-10-09 12:00:37 +08:00
this.buffers[front ? "unshift" : "push"](buffer);
return this;
};
p.addInt16 = function(val, front) {
2010-10-09 12:00:37 +08:00
return this.add(Buffer([(val >>> 8),(val >>> 0)]),front);
};
p.getByteLength = function(initial) {
2010-10-09 12:00:37 +08:00
return this.buffers.reduce(function(previous, current){
return previous + current.length;
},initial || 0);
};
p.addInt32 = function(val, first) {
2010-10-09 12:00:37 +08:00
return this.add(Buffer([
2010-10-16 06:29:01 +08:00
(val >>> 24 & 0xFF),
(val >>> 16 & 0xFF),
(val >>> 8 & 0xFF),
(val >>> 0 & 0xFF)
2010-10-09 12:00:37 +08:00
]),first);
};
p.addCString = function(val) {
2010-10-09 12:00:37 +08:00
return this.add(Buffer(val + '\0','utf8'));
};
p.addChar = function(char, first) {
return this.add(Buffer(char,'utf8'), first);
2010-10-11 06:40:11 +08:00
};
p.join = function(appendLength, char) {
2010-10-09 12:00:37 +08:00
var length = this.getByteLength();
if(appendLength) {
this.addInt32(length+4, true);
return this.join(false, char);
}
if(char) {
2010-10-11 06:40:11 +08:00
this.addChar(char, true);
2010-10-09 12:00:37 +08:00
length++;
}
var result = Buffer(length);
var index = 0;
this.buffers.forEach(function(buffer) {
buffer.copy(result, index, 0);
index += buffer.length;
});
return result;
};
BufferList.concat = function() {
var total = new BufferList();
for(var i = 0; i < arguments.length; i++) {
total.add(arguments[i]);
}
return total.join();
};
module.exports = BufferList;