Add support for the GetRoutingTable message

Author: bigmontz

bolt-connection/src/rediscovery/routing-table.js

@@ -37,13 +37,21 @@ const MIN_ROUTERS = 1
  * The routing table object used to determine the role of the servers in the driver.
  */
 export default class RoutingTable {
-  constructor ({ database, routers, readers, writers, expirationTime } = {}) {
+  constructor ({
+    database,
+    routers,
+    readers,
+    writers,
+    expirationTime,
+    ttl
+  } = {}) {
     this.database = database
     this.databaseName = database || 'default database'
     this.routers = routers || []
     this.readers = readers || []
     this.writers = writers || []
     this.expirationTime = expirationTime || int(0)
+    this.ttl = ttl
   }
 
   /**
@@ -139,6 +147,7 @@ export function createValidRoutingTable (
   routerAddress,
   rawRoutingTable
 ) {
+  const ttl = rawRoutingTable.ttl
   const expirationTime = calculateExpirationTime(rawRoutingTable, routerAddress)
   const { routers, readers, writers } = parseServers(
     rawRoutingTable,
@@ -153,7 +162,8 @@ export function createValidRoutingTable (
     routers,
     readers,
     writers,
-    expirationTime
+    expirationTime,
+    ttl
   })
 }
 

testkit-backend/src/request-handlers.js

@@ -270,3 +270,25 @@ export function ResolverResolutionCompleted (
   const request = context.getResolverRequest(requestId)
   request.resolve(addresses)
 }
+
+export function GetRoutingTable (context, { driverId, database }, wire) {
+  const serverAddressToString = serverAddress => serverAddress.asHostPort()
+  const driver = context.getDriver(driverId)
+  const routingTable =
+    driver &&
+    driver._getOrCreateConnectionProvider() &&
+    driver._getOrCreateConnectionProvider()._routingTableRegistry &&
+    driver._getOrCreateConnectionProvider()._routingTableRegistry.get(database)
+
+  if (routingTable) {
+    wire.writeResponse('RoutingTable', {
+      database: routingTable.database,
+      ttl: routingTable.ttl,
+      readers: routingTable.readers.map(serverAddressToString),
+      writers: routingTable.writers.map(serverAddressToString),
+      routers: routingTable.routers.map(serverAddressToString)
+    })
+  } else {
+    wire.writeError('Could not find routing table')
+  }
+}

testkit-backend/src/skipped-tests.js

@@ -84,12 +84,6 @@ const skippedTests = [
     ifEndsWith(
       'test_should_fail_when_reading_from_unexpectedly_interrupting_reader_using_session_run'
     )
-  ),
-  skip(
-    'Needs to implement GetRoutingTable tk protocol message',
-    ifStartsWith(
-      'stub.configuration_hints.test_connection_recv_timeout_seconds.TestRoutingConnectionRecvTimeout'
-    )
   )
 ]