From 178921d6b61dd9f09527245368012aec950627e6 Mon Sep 17 00:00:00 2001
From: Justin Corpron <jncorpron@gmail.com>
Date: Sat, 7 May 2016 19:16:02 -0700
Subject: [PATCH] Notify the HostSelectionPolicy of the partioner

Fixes #724; adds back the call to SetPartitioner on the session's
HostSelectionPolicy. This is mostly needed so that the token
aware policy initializes and updates its token ring representation.
---
 AUTHORS        | 2 +-
 host_source.go | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/AUTHORS b/AUTHORS
index b578ca4f7..9f6e66c99 100644
--- a/AUTHORS
+++ b/AUTHORS
@@ -43,7 +43,7 @@ James Maloney <jamessagan@gmail.com>
 Ashwin Purohit <purohit@gmail.com>
 Dan Kinder <dkinder.is.me@gmail.com>
 Oliver Beattie <oliver@obeattie.com>
-Justin Corpron <justin@retailnext.com>
+Justin Corpron <jncorpron@gmail.com>
 Miles Delahunty <miles.delahunty@gmail.com>
 Zach Badgett <zach.badgett@gmail.com>
 Maciek Sakrejda <maciek@heroku.com>
diff --git a/host_source.go b/host_source.go
index 29ab4d452..0f20e8c86 100644
--- a/host_source.go
+++ b/host_source.go
@@ -391,5 +391,6 @@ func (r *ringDescriber) refreshRing() error {
 	}
 
 	r.session.metadata.setPartitioner(partitioner)
+	r.session.policy.SetPartitioner(partitioner)
 	return nil
 }