Initial commit

index.js

@@ -0,0 +1,66 @@
+var CopyToQueryStream = require('./copy-to')
+
+module.exports = {
+  to: function(txt) {
+    return new CopyToQueryStream(txt)
+  },
+  from: function (txt) {
+    return new CopyStreamQuery(txt)
+  }
+}
+
+var Transform = require('stream').Transform
+var util = require('util')
+
+var CopyStreamQuery = function(text) {
+  Transform.call(this)
+  this.text = text
+  this._listeners = null
+  this._copyOutResponse = null
+}
+
+util.inherits(CopyStreamQuery, Transform)
+
+CopyStreamQuery.prototype.submit = function(connection) {
+  this.connection = connection
+  connection.query(this.text)
+}
+
+var code = {
+  H: 72, //CopyOutResponse
+  d: 0x64, //CopyData
+  c: 0x63 //CopyDone
+}
+
+var copyDataBuffer = Buffer([code.d])
+CopyStreamQuery.prototype._transform = function(chunk, enc, cb) {
+  this.push(copyDataBuffer)
+  var lenBuffer = Buffer(4)
+  lenBuffer.writeUInt32BE(chunk.length + 4, 0)
+  this.push(lenBuffer)
+  this.push(chunk)
+  cb()
+}
+
+CopyStreamQuery.prototype._flush = function(cb) {
+  var finBuffer = Buffer([code.c, 0, 0, 0, 4])
+  this.push(finBuffer)
+  //never call this callback, do not close underlying stream
+  //cb()
+}
+
+CopyStreamQuery.prototype.handleError = function(e) {
+  this.emit('error', e)
+}
+
+CopyStreamQuery.prototype.streamData = function(connection) {
+  this.pipe(connection.stream)
+}
+
+CopyStreamQuery.prototype.handleCommandComplete = function() {
+  this.unpipe()
+  this.emit('end')
+}
+
+CopyStreamQuery.prototype.handleReadyForQuery = function() {
+}