go thunder/reactive: InvalidateAt/After stores expiration time

samsarahq · Sep 12, 2018 · 00f2096 · 00f2096
1 parent 343d127
commit 00f2096
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/reactive/util.go b/reactive/util.go
@@ -3,13 +3,15 @@ package reactive
 import (
 	"context"
 	"time"
+
+	"github.com/samsarahq/thunder/thunderpb"
 )
 
 func InvalidateAfter(ctx context.Context, d time.Duration) {
 	r := NewResource()
 	timer := time.AfterFunc(d, r.Invalidate)
 	r.Cleanup(func() { timer.Stop() })
-	AddDependency(ctx, r, nil)
+	AddDependency(ctx, r, &thunderpb.ExpirationTime{Time: time.Now().Add(d)})
 }
 
 func InvalidateAt(ctx context.Context, t time.Time) {