func (app Application) StartWorkers() {
zonedUAAClient := uaa.NewZonedUAAClient(app.env.UAAClientID, app.env.UAAClientSecret, app.env.VerifySSL, UAAPublicKey)
WorkerGenerator{
InstanceIndex: app.env.VCAPApplication.InstanceIndex,
Count: WorkerCount,
}.Work(func(i int) Worker {
cloak, err := conceal.NewCloak(app.env.EncryptionKey)
if err != nil {
panic(err)
}
v1Workflow := postal.NewV1Process(postal.V1ProcessConfig{
DBTrace: app.env.DBLoggingEnabled,
UAAHost: app.env.UAAHost,
Sender: app.env.Sender,
Domain: app.env.Domain,
Packager: postal.NewPackager(app.mother.TemplatesLoader(), cloak),
MailClient: app.mother.MailClient(),
Database: app.mother.Database(),
TokenLoader: uaa.NewTokenLoader(zonedUAAClient),
UserLoader: postal.NewUserLoader(zonedUAAClient),
KindsRepo: app.mother.KindsRepo(),
ReceiptsRepo: app.mother.ReceiptsRepo(),
UnsubscribesRepo: app.mother.UnsubscribesRepo(),
GlobalUnsubscribesRepo: app.mother.GlobalUnsubscribesRepo(),
MessageStatusUpdater: postal.NewMessageStatusUpdater(app.mother.MessagesRepo()),
DeliveryFailureHandler: postal.NewDeliveryFailureHandler(),
})
database := v2models.NewDatabase(app.mother.SQLDatabase(), v2models.Config{})
messageStatusUpdater := postal.NewV2MessageStatusUpdater(v2models.NewMessagesRepository(util.NewClock()))
v2Workflow := postal.NewV2Workflow(app.mother.MailClient(), postal.NewPackager(app.mother.TemplatesLoader(), cloak), postal.NewUserLoader(zonedUAAClient), uaa.NewTokenLoader(zonedUAAClient), messageStatusUpdater, database, app.env.Sender, app.env.Domain, app.env.UAAHost)
worker := postal.NewDeliveryWorker(v1Workflow, v2Workflow, postal.DeliveryWorkerConfig{
ID: i,
UAAHost: app.env.UAAHost,
Logger: app.mother.Logger(),
Queue: app.mother.Queue(),
DBTrace: app.env.DBLoggingEnabled,
Database: database,
StrategyDeterminer: strategy.NewStrategyDeterminer(app.mother.UserStrategy(), app.mother.SpaceStrategy(), app.mother.OrganizationStrategy(), app.mother.EmailStrategy()),
DeliveryFailureHandler: postal.NewDeliveryFailureHandler(),
MessageStatusUpdater: messageStatusUpdater,
})
return &worker
})
}
var _ = Describe("Messages Repository", func() {
var (
repo models.MessagesRepository
conn *db.Connection
clock *mocks.Clock
)
BeforeEach(func() {
database := models.NewDatabase(sqlDB, models.Config{})
helpers.TruncateTables(db.NewDatabase(sqlDB, db.Config{}))
conn = database.Connection().(*db.Connection)
conn.AddTableWithName(models.Message{}, "messages")
clock = mocks.NewClock()
repo = models.NewMessagesRepository(clock)
})
Describe("CountByStatus", func() {
BeforeEach(func() {
err := conn.Insert(&models.Message{
ID: "message-id-123",
CampaignID: "some-campaign-id",
Status: postal.StatusDelivered,
})
Expect(err).NotTo(HaveOccurred())
err = conn.Insert(&models.Message{
ID: "message-id-1234",
CampaignID: "some-campaign-id",
Status: postal.StatusFailed,
func (m *Mother) V2Enqueuer() queue.JobEnqueuer {
return queue.NewJobEnqueuer(m.Queue(), uuid.NewV4, v2models.NewMessagesRepository(util.NewClock()))
}
func NewRouter(mx muxer, config Config) http.Handler {
requestCounter := middleware.NewRequestCounter(mx.GetRouter(), metrics.DefaultLogger)
logging := middleware.NewRequestLogging(config.Logger)
notificationsWriteAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notifications.write")
databaseAllocator := middleware.NewDatabaseAllocator(config.SQLDB, config.DBLoggingEnabled)
warrantConfig := warrant.Config{
Host: config.UAAHost,
SkipVerifySSL: config.SkipVerifySSL,
}
warrantUsersService := warrant.NewUsersService(warrantConfig)
warrantClientsService := warrant.NewClientsService(warrantConfig)
rainmakerConfig := rainmaker.Config{
Host: config.CCHost,
SkipVerifySSL: config.SkipVerifySSL,
}
rainmakerSpacesService := rainmaker.NewSpacesService(rainmakerConfig)
rainmakerOrganizationsService := rainmaker.NewOrganizationsService(rainmakerConfig)
userFinder := uaa.NewUserFinder(config.UAAClientID, config.UAAClientSecret, warrantUsersService, warrantClientsService)
spaceFinder := cf.NewSpaceFinder(config.UAAClientID, config.UAAClientSecret, warrantClientsService, rainmakerSpacesService)
orgFinder := cf.NewOrgFinder(config.UAAClientID, config.UAAClientSecret, warrantClientsService, rainmakerOrganizationsService)
campaignEnqueuer := queue.NewCampaignEnqueuer(config.Queue)
sendersRepository := models.NewSendersRepository(uuid.NewV4)
campaignTypesRepository := models.NewCampaignTypesRepository(uuid.NewV4)
templatesRepository := models.NewTemplatesRepository(uuid.NewV4)
campaignsRepository := models.NewCampaignsRepository(uuid.NewV4)
messagesRepository := models.NewMessagesRepository(util.NewClock())
sendersCollection := collections.NewSendersCollection(sendersRepository, campaignTypesRepository)
templatesCollection := collections.NewTemplatesCollection(templatesRepository)
campaignTypesCollection := collections.NewCampaignTypesCollection(campaignTypesRepository, sendersRepository, templatesRepository)
campaignsCollection := collections.NewCampaignsCollection(campaignEnqueuer, campaignsRepository, campaignTypesRepository, templatesRepository, sendersRepository, userFinder, spaceFinder, orgFinder)
campaignStatusesCollection := collections.NewCampaignStatusesCollection(campaignsRepository, messagesRepository)
info.Routes{
RequestCounter: requestCounter,
RequestLogging: logging,
}.Register(mx)
senders.Routes{
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
SendersCollection: sendersCollection,
}.Register(mx)
campaigntypes.Routes{
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
CampaignTypesCollection: campaignTypesCollection,
}.Register(mx)
templates.Routes{
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
TemplatesCollection: templatesCollection,
}.Register(mx)
campaigns.Routes{
Clock: util.NewClock(),
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
CampaignsCollection: campaignsCollection,
CampaignStatusesCollection: campaignStatusesCollection,
}.Register(mx)
return mx
}