DRAFT: add logic for infinite displays

pkg/base/app_throttle.go

@@ -18,3 +18,10 @@ func NewAppThrottle(expireAt time.Time, ratio float64) *AppThrottle {
 	}
 	return result
 }
+
+// DisplayAppThrottle is a type for displaying data back to the end 
+// user via chatop. Handles infinite TTL and allows ExpiresAt to be "INFINITE"
+type DisplayAppThrottle struct {
+	ExpireAt string
+	Ratio    float64
+}

pkg/group/consensus_service.go

@@ -24,7 +24,7 @@ type ConsensusServiceStatus struct {
 // ConsensusService is a freno-oriented interface for making requests that require consensus.
 type ConsensusService interface {
 	ThrottleApp(appName string, ttlMinutes int64, expireAt time.Time, ratio float64) error
-	ThrottledAppsMap() (result map[string](*base.AppThrottle))
+	ThrottledAppsMap() (result map[string](*base.DisplayAppThrottle))
 	UnthrottleApp(appName string) error
 	RecentAppsMap() (result map[string](*base.RecentApp))
 

pkg/group/mysql.go

@@ -420,7 +420,7 @@ func (backend *MySQLBackend) ThrottleApp(appName string, ttlMinutes int64, expir
 	return err
 }
 
-func (backend *MySQLBackend) ThrottledAppsMap() (result map[string](*base.AppThrottle)) {
+func (backend *MySQLBackend) ThrottledAppsMap() (result map[string](*base.DisplayAppThrottle)) {
 	return backend.throttler.ThrottledAppsMap()
 }
 

pkg/group/store.go

@@ -177,7 +177,7 @@ func (store *Store) SkippedHostsMap() map[string]time.Time {
 	return store.throttler.SkippedHostsMap()
 }
 
-func (store *Store) ThrottledAppsMap() (result map[string](*base.AppThrottle)) {
+func (store *Store) ThrottledAppsMap() (result map[string](*base.DisplayAppThrottle)) {
 	return store.throttler.ThrottledAppsMap()
 }
 

pkg/throttle/throttler.go

@@ -565,11 +565,20 @@ func (throttler *Throttler) IsAppThrottled(appName, storeName string) bool {
 	return false
 }
 
-func (throttler *Throttler) ThrottledAppsMap() (result map[string](*base.AppThrottle)) {
-	result = make(map[string](*base.AppThrottle))
+func (throttler *Throttler) ThrottledAppsMap() (result map[string](*base.DisplayAppThrottle)) {
+	result = make(map[string](*base.DisplayAppThrottle))
 
 	for appName, item := range throttler.throttledApps.Items() {
-		appThrottle := item.Object.(*base.AppThrottle)
+		appThrottle := item.Object.(*base.DisplayAppThrottle)
+		expire, expireErr := time.Parse(appThrottle.ExpireAt, "2024-Jan-01")
+
+		if expireErr != nil {
+			panic(expireErr)
+		}
+
+		if expire.IsZero() {
+			appThrottle.ExpireAt = "INFINITE"
+		} 
 		result[appName] = appThrottle
 	}
 	return result