Add option to close passed in client (#66)

Author: jsumners

README.md

@@ -56,10 +56,17 @@ const fastify = require('fastify')()
 const redis = require('redis').createClient({ host: 'localhost', port: 6379 })
 
 fastify.register(require('fastify-redis'), { client: redis })
+```
+
+Note: by default, *fastify-redis* will **not** automatically close the client
+connection when the Fastify server shuts down. To opt-in to this behavior,
+register the client like so:
 
-// ...
-// ...
-// ...
+```js
+fastify.register(require('fastify-redis'), {
+  client: redis,
+  closeClient: true
+})
 ```
 
 ## Registering multiple Redis client instances

index.js

@@ -55,6 +55,9 @@ function fastifyRedis (fastify, options, next) {
       }
 
       fastify.decorate('redis', client)
+      if (options.closeClient === true) {
+        fastify.addHook('onClose', close)
+      }
     }
   }
 

test/test.js

@@ -181,6 +181,37 @@ test('custom client', (t) => {
   })
 })
 
+test('custom client gets closed', (t) => {
+  t.plan(7)
+  const fastify = Fastify()
+  const redis = require('redis').createClient({ host: 'localhost', port: 6379 })
+
+  fastify.register(fastifyRedis, { client: redis, closeClient: true })
+
+  fastify.ready((err) => {
+    t.error(err)
+    t.is(fastify.redis, redis)
+
+    fastify.redis.set('key', 'value', (err) => {
+      t.error(err)
+      fastify.redis.get('key', (err, val) => {
+        t.error(err)
+        t.equal(val, 'value')
+
+        const origQuit = fastify.redis.quit
+        fastify.redis.quit = (cb) => {
+          t.pass('redis client closed')
+          origQuit.call(fastify.redis, cb)
+        }
+
+        fastify.close(function (err) {
+          t.error(err)
+        })
+      })
+    })
+  })
+})
+
 test('custom client inside a namespace', (t) => {
   t.plan(7)
   const fastify = Fastify()