Merge pull request #4 from LuKks/master

remoteAddress/Port/Family and setTimeout feature

README.md

@@ -112,6 +112,10 @@ Emitted when the writable side is fully closed.
 
 Emitted when the readable side is fully closed.
 
+#### `connection.on('timeout')`
+
+Emitted when the timeout it's reached. Is only a notification, you should end the connection.
+
 #### `connection.close([callback])`
 
 Closes the connection.
@@ -142,6 +146,22 @@ The callback is called with `callback(err, buffers, lengths)`.
 
 End the writable side of the connection.
 
+#### `connection.setTimeout(millis, [callback])`
+
+Set timeout on the connection. Optionally you can provide an callback. If millis is setted on 0, then it disabled.
+
+#### `connection.remoteAddress`
+
+The IP address of the connection. For example, '172.217.28.163'.
+
+#### `connection.remoteFamily`
+
+The IP family of the connection. Before the event "connect" or "connection" it's a empty string and after it's always "IPv4".
+
+#### `connection.remotePort`
+
+The remote port. For example, 63750.
+
 ## License
 
 MIT

examples/wait.js

@@ -0,0 +1,9 @@
+const turbo = require('../')
+
+const server = turbo.createServer(function (socket) {
+  console.log(socket.remoteFamily, socket.remoteAddress, socket.remotePort)
+
+  socket.close()
+})
+
+server.listen(8080)

lib/connection.js

@@ -14,6 +14,9 @@ class Connection extends events.EventEmitter {
     this.finished = false
     this.ended = false
     this.allowHalfOpen = false
+    this.remoteFamily = ''
+    this.remoteAddress = ''
+    this.remotePort = 0
     this.writable = false
     this.readable = false
 
@@ -22,6 +25,7 @@ class Connection extends events.EventEmitter {
     this._handle = Buffer.alloc(binding.sizeof_turbo_net_tcp_t)
     this._reads = new RequestQueue(8, 0)
     this._writes = new RequestQueue(16, binding.sizeof_uv_write_t)
+    this._timeoutId = false
 
     this._finishing = []
     this._closing = []
@@ -57,6 +61,8 @@ class Connection extends events.EventEmitter {
   }
 
   _onclose () {
+    clearTimeout(this._timeoutId)
+
     if (this._server) unordered.remove(this._server.connections, this)
     this.closed = true
     this._closing = callAll(this._closing, null)
@@ -92,6 +98,10 @@ class Connection extends events.EventEmitter {
       return
     }
 
+    this.remoteFamily = 'IPv4'
+    this.remoteAddress = binding.turbo_net_tcp_remote_address(this._handle)
+    this.remotePort = binding.turbo_net_tcp_remote_port(this._handle)
+
     this.readable = true
     this.writable = true
     if (this._queued) this._unqueue()
@@ -198,6 +208,15 @@ class Connection extends events.EventEmitter {
     binding.turbo_net_tcp_write(this._handle, writing.handle, writing.buffer, len)
   }
 
+  setTimeout (millis, cb) {
+    clearTimeout(this._timeoutId)
+
+    if (millis > 0) {
+      this._timeoutId = setTimeout(this.emit.bind(this), millis, 'timeout')
+      if (cb) this.once('timeout', cb)
+    } else if (cb) this.removeListener('timeout', cb)
+  }
+
   close (cb) {
     if (!cb) cb = noop
     if (this.closed) return process.nextTick(cb)