From 645b92c152d7c60e9e6023d6c5ae44f9973ffd26 Mon Sep 17 00:00:00 2001
From: David Dollar <david@convox.com>
Date: Wed, 29 Jan 2020 22:56:59 -0500
Subject: [PATCH] router: cleanup unused hosts (#94)

---
 pkg/router/storage_redis.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pkg/router/storage_redis.go b/pkg/router/storage_redis.go
index 73a3432..222e0b7 100644
--- a/pkg/router/storage_redis.go
+++ b/pkg/router/storage_redis.go
@@ -140,5 +140,16 @@ func (s *StorageRedis) TargetRemove(host, target string) error {
 		return err
 	}
 
+	len, err := s.redis.LLen(fmt.Sprintf("router/targets/%s", host)).Result()
+	if err != nil {
+		return err
+	}
+	
+	if len == 0 {
+		if _, err := s.redis.SRem("router/hosts", host).Result(); err != nil {
+			return err
+		}
+	}
+
 	return nil
 }