Run generator

mtojek · mtojek · commit b89f484794b2 · 2024-09-06T14:36:55.000+02:00
diff --git a/cli/server.go b/cli/server.go
@@ -56,6 +56,7 @@ import (
 	"cdr.dev/slog"
 	"cdr.dev/slog/sloggers/sloghuman"
 	"github.com/coder/coder/v2/coderd/entitlements"
+	"github.com/coder/coder/v2/coderd/notifications/reports"
 	"github.com/coder/pretty"
 	"github.com/coder/quartz"
 	"github.com/coder/retry"
@@ -1023,6 +1024,10 @@ func (r *RootCmd) Server(newAPI func(context.Context, *coderd.Options) (*coderd.
 
 				// nolint:gocritic // TODO: create own role.
 				notificationsManager.Run(dbauthz.AsSystemRestricted(ctx))
+
+				// Run report generator to distribute periodic reports.
+				reportGenerator := reports.NewReportGenerator(ctx, logger, options.Database, options.NotificationsEnqueuer, quartz.NewReal())
+				defer reportGenerator.Close()
 			}
 
 			// Wrap the server in middleware that redirects to the access URL if
diff --git a/coderd/notifications/reports/generator.go b/coderd/notifications/reports/generator.go
@@ -10,14 +10,15 @@ import (
 	"github.com/coder/coder/v2/coderd/database"
 	"github.com/coder/coder/v2/coderd/database/dbauthz"
 	"github.com/coder/coder/v2/coderd/database/dbtime"
+	"github.com/coder/coder/v2/coderd/notifications"
 	"github.com/coder/quartz"
 )
 
 const (
 	delay = 5 * time.Minute
 )
 
-func NewReportGenerator(ctx context.Context, logger slog.Logger, db database.Store, clk quartz.Clock) io.Closer {
+func NewReportGenerator(ctx context.Context, logger slog.Logger, db database.Store, notificationsEnqueuer notifications.Enqueuer, clk quartz.Clock) io.Closer {
 	closed := make(chan struct{})
 
 	ctx, cancelFunc := context.WithCancel(ctx)

Original file line number	Diff line number	Diff line change
`@@ -10,14 +10,15 @@ import (`
`10`	`10`	`"github.com/coder/coder/v2/coderd/database"`
`11`	`11`	`"github.com/coder/coder/v2/coderd/database/dbauthz"`
`12`	`12`	`"github.com/coder/coder/v2/coderd/database/dbtime"`
	`13`	`+ "github.com/coder/coder/v2/coderd/notifications"`
`13`	`14`	`"github.com/coder/quartz"`
`14`	`15`	`)`
`15`	`16`
`16`	`17`	`const (`
`17`	`18`	`delay = 5 * time.Minute`
`18`	`19`	`)`
`19`	`20`
`20`		`-func NewReportGenerator(ctx context.Context, logger slog.Logger, db database.Store, clk quartz.Clock) io.Closer {`
	`21`	`+func NewReportGenerator(ctx context.Context, logger slog.Logger, db database.Store, notificationsEnqueuer notifications.Enqueuer, clk quartz.Clock) io.Closer {`
`21`	`22`	`closed := make(chan struct{})`
`22`	`23`
`23`	`24`	`ctx, cancelFunc := context.WithCancel(ctx)`