mirror of
https://github.com/zitadel/zitadel.git
synced 2025-04-24 20:11:33 +00:00
444f682e25
# Which Problems Are Solved The recently introduced notification queue have potential race conditions. # How the Problems Are Solved Current code is refactored to use the queue package, which is safe in regards of concurrency. # Additional Changes - the queue is included in startup - improved code quality of queue # Additional Context - closes https://github.com/zitadel/zitadel/issues/9278
84 lines
2.7 KiB
Go
84 lines
2.7 KiB
Go
package notification
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/zitadel/zitadel/internal/command"
|
|
"github.com/zitadel/zitadel/internal/crypto"
|
|
"github.com/zitadel/zitadel/internal/database"
|
|
"github.com/zitadel/zitadel/internal/eventstore"
|
|
"github.com/zitadel/zitadel/internal/eventstore/handler/v2"
|
|
"github.com/zitadel/zitadel/internal/notification/handlers"
|
|
_ "github.com/zitadel/zitadel/internal/notification/statik"
|
|
"github.com/zitadel/zitadel/internal/query"
|
|
"github.com/zitadel/zitadel/internal/query/projection"
|
|
"github.com/zitadel/zitadel/internal/queue"
|
|
)
|
|
|
|
var (
|
|
projections []*handler.Handler
|
|
worker *handlers.NotificationWorker
|
|
)
|
|
|
|
func Register(
|
|
ctx context.Context,
|
|
userHandlerCustomConfig, quotaHandlerCustomConfig, telemetryHandlerCustomConfig, backChannelLogoutHandlerCustomConfig projection.CustomConfig,
|
|
notificationWorkerConfig handlers.WorkerConfig,
|
|
telemetryCfg handlers.TelemetryPusherConfig,
|
|
externalDomain string,
|
|
externalPort uint16,
|
|
externalSecure bool,
|
|
commands *command.Commands,
|
|
queries *query.Queries,
|
|
es *eventstore.Eventstore,
|
|
otpEmailTmpl, fileSystemPath string,
|
|
userEncryption, smtpEncryption, smsEncryption, keysEncryptionAlg crypto.EncryptionAlgorithm,
|
|
tokenLifetime time.Duration,
|
|
client *database.DB,
|
|
queue *queue.Queue,
|
|
) {
|
|
if !notificationWorkerConfig.LegacyEnabled {
|
|
queue.ShouldStart()
|
|
}
|
|
|
|
q := handlers.NewNotificationQueries(queries, es, externalDomain, externalPort, externalSecure, fileSystemPath, userEncryption, smtpEncryption, smsEncryption)
|
|
c := newChannels(q)
|
|
projections = append(projections, handlers.NewUserNotifier(ctx, projection.ApplyCustomConfig(userHandlerCustomConfig), commands, q, c, otpEmailTmpl, notificationWorkerConfig, queue))
|
|
projections = append(projections, handlers.NewQuotaNotifier(ctx, projection.ApplyCustomConfig(quotaHandlerCustomConfig), commands, q, c))
|
|
projections = append(projections, handlers.NewBackChannelLogoutNotifier(
|
|
ctx,
|
|
projection.ApplyCustomConfig(backChannelLogoutHandlerCustomConfig),
|
|
commands,
|
|
q,
|
|
es,
|
|
keysEncryptionAlg,
|
|
c,
|
|
tokenLifetime,
|
|
))
|
|
if telemetryCfg.Enabled {
|
|
projections = append(projections, handlers.NewTelemetryPusher(ctx, telemetryCfg, projection.ApplyCustomConfig(telemetryHandlerCustomConfig), commands, q, c))
|
|
}
|
|
worker = handlers.NewNotificationWorker(notificationWorkerConfig, commands, q, es, client, c, queue)
|
|
}
|
|
|
|
func Start(ctx context.Context) {
|
|
for _, projection := range projections {
|
|
projection.Start(ctx)
|
|
}
|
|
}
|
|
|
|
func ProjectInstance(ctx context.Context) error {
|
|
for _, projection := range projections {
|
|
_, err := projection.Trigger(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Projections() []*handler.Handler {
|
|
return projections
|
|
}
|