From 4ae44b43a376f217a37fa1ff1a1a7329da507566 Mon Sep 17 00:00:00 2001
From: sacOO7 <sachinshinde7676@gmail.com>
Date: Fri, 9 Aug 2024 08:26:35 +0530
Subject: [PATCH] Replaced old implementation for hostcache with new one

---
 ably/auth_internal_test.go |  4 +-
 ably/export_test.go        |  6 +--
 ably/rest_client.go        | 75 +++-----------------------------------
 3 files changed, 9 insertions(+), 76 deletions(-)

diff --git a/ably/auth_internal_test.go b/ably/auth_internal_test.go
index cfa1505ce..de16aec49 100644
--- a/ably/auth_internal_test.go
+++ b/ably/auth_internal_test.go
@@ -270,8 +270,8 @@ func TestRequestToken(t *testing.T) {
 			auth: &Auth{
 				clientID: "aClientID",
 				client: &REST{
-					successFallbackHost: &fallbackCache{},
-					log:                 logger{l: &stdLogger{mocklogger}},
+					hostCache: &hostCache{},
+					log:       logger{l: &stdLogger{mocklogger}},
 					opts: &clientOptions{authOptions: authOptions{
 						AuthURL: "foo.com",
 						Key:     "abc:def",
diff --git a/ably/export_test.go b/ably/export_test.go
index 7fcbd64e2..d9e73c3b2 100644
--- a/ably/export_test.go
+++ b/ably/export_test.go
@@ -89,12 +89,8 @@ func (a *Auth) SetServerTimeFunc(st func() (time.Time, error)) {
 	a.serverTimeHandler = st
 }
 
-func (c *REST) SetSuccessFallbackHost(duration time.Duration) {
-	c.successFallbackHost = &fallbackCache{duration: duration}
-}
-
 func (c *REST) GetCachedFallbackHost() string {
-	return c.successFallbackHost.get()
+	return c.hostCache.get()
 }
 
 func (c *RealtimeChannel) GetChannelSerial() string {
diff --git a/ably/rest_client.go b/ably/rest_client.go
index 35242850d..ef506d08b 100644
--- a/ably/rest_client.go
+++ b/ably/rest_client.go
@@ -134,9 +134,9 @@ type REST struct {
 	//Channels is a [ably.RESTChannels] object (RSN1).
 	Channels *RESTChannels
 
-	opts                *clientOptions
-	successFallbackHost *fallbackCache
-	log                 logger
+	opts      *clientOptions
+	hostCache *hostCache
+	log       logger
 }
 
 // NewREST construct a RestClient object using an [ably.ClientOption] object to configure
@@ -158,7 +158,7 @@ func NewREST(options ...ClientOption) (*REST, error) {
 		chans:  make(map[string]*RESTChannel),
 		client: c,
 	}
-	c.successFallbackHost = &fallbackCache{
+	c.hostCache = &hostCache{
 		duration: c.opts.fallbackRetryTimeout(),
 	}
 	return c, nil
@@ -634,75 +634,12 @@ func (c *REST) do(ctx context.Context, r *request) (*http.Response, error) {
 	return c.doWithHandle(ctx, r, c.handleResponse)
 }
 
-// fallbackCache caches a successful fallback host for 10 minutes.
-type fallbackCache struct {
-	running  bool
-	host     string
-	duration time.Duration
-	cancel   func()
-	mu       sync.RWMutex
-}
-
-func (f *fallbackCache) get() string {
-	if f.isRunning() {
-		f.mu.RLock()
-		h := f.host
-		f.mu.RUnlock()
-		return h
-	}
-	return ""
-}
-
-func (f *fallbackCache) isRunning() bool {
-	f.mu.RLock()
-	v := f.running
-	f.mu.RUnlock()
-	return v
-}
-
-func (f *fallbackCache) run(host string) {
-	f.mu.Lock()
-	now := time.Now()
-	duration := defaultOptions.FallbackRetryTimeout // spec RSC15f
-	if f.duration != 0 {
-		duration = f.duration
-	}
-	ctx, cancel := context.WithDeadline(context.Background(), now.Add(duration))
-	f.running = true
-	f.host = host
-	f.cancel = cancel
-	f.mu.Unlock()
-	<-ctx.Done()
-	f.mu.Lock()
-	f.running = false
-	f.mu.Unlock()
-}
-
-func (f *fallbackCache) stop() {
-	f.cancel()
-	// we make sure we have stopped
-	for {
-		if !f.isRunning() {
-			return
-		}
-	}
-}
-
-func (f *fallbackCache) put(host string) {
-	if f.get() != host {
-		if f.isRunning() {
-			f.stop()
-		}
-		go f.run(host)
-	}
-}
-
 func (c *REST) doWithHandle(ctx context.Context, r *request, handle func(*http.Response, interface{}) (*http.Response, error)) (*http.Response, error) {
 	req, err := c.newHTTPRequest(ctx, r)
 	if err != nil {
 		return nil, err
 	}
-	if h := c.successFallbackHost.get(); h != "" {
+	if h := c.hostCache.get(); h != "" {
 		req.URL.Host = h // RSC15f
 		c.log.Verbosef("RestClient: setting URL.Host=%q", h)
 	}
@@ -775,7 +712,7 @@ func (c *REST) doWithHandle(ctx context.Context, r *request, handle func(*http.R
 						}
 						return nil, err
 					}
-					c.successFallbackHost.put(h)
+					c.hostCache.put(h)
 					return resp, nil
 				}
 			}